3D-StyleGAN: A Style-Based Generative Adversarial Network for Generative Modeling of Three-Dimensional Medical Images

Sungmin Hong; Razvan Marinescu; Adrian V. Dalca; Anna K. Bonkhoff; Martin Bretzner; Natalia S. Rost; Polina Golland

Home/Archive/arXiv

arXiv · 2107.09700

3D-StyleGAN: A Style-Based Generative Adversarial Network for Generative Modeling of Three-Dimensional Medical Images

Sungmin Hong·Razvan Marinescu·Adrian V. Dalca·Anna K. Bonkhoff·Martin Bretzner·Natalia S. Rost·Polina Golland

View original source ↗Read source document ↗Export metadata ↓

Abstract

Image synthesis via Generative Adversarial Networks (GANs) of three-dimensional (3D) medical images has great potential that can be extended to many medical applications, such as, image enhancement and disease progression modeling. However, current GAN technologies for 3D medical image synthesis need to be significantly improved to be readily adapted to real-world medical problems. In this paper, we extend the state-of-the-art StyleGAN2 model, which natively works with two-dimensional images, to enable 3D image synthesis. In addition to the image synthesis, we investigate the controllability and interpretability of the 3D-StyleGAN via style vectors inherited form the original StyleGAN2 that are highly suitable for medical applications: (i) the latent space projection and reconstruction of unseen real images, and (ii) style mixing. We demonstrate the 3D-StyleGAN's performance and feasibility with ~12,000 three-dimensional full brain MR T1 images, although it can be applied to any 3D volumetric images. Furthermore, we explore different configurations of hyperparameters to investigate potential improvement of the image synthesis with larger networks. The codes and pre-trained networks are available online: https://github.com/sh4174/3DStyleGAN.

Explore related subjects

eess.IV cs.CV cs.LG

Keep this discovery

Explore connections, maps & timelines

BibTeX RIS

Sungmin Hong, Razvan Marinescu, Adrian V. Dalca, Anna K. Bonkhoff, Martin Bretzner, Natalia S. Rost, Polina Golland. 2021-07-20. 3D-StyleGAN: A Style-Based Generative Adversarial Network for Generative Modeling of Three-Dimensional Medical Images. https://arxiv.org/abs/2107.09700

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

VideoPulse: Neonatal heart rate and peripheral capillary oxygen saturation (SpO2) estimation from contact free video

Remote photoplethysmography (rPPG) enables contact free monitoring of vital signs and is especially valuable for neonates, since conventional methods often require sustained skin contact with adhesive probes that can irritate fragile skin and increase infection control burden. We present VideoPulse, a neonatal dataset and an end to end pipeline that estimates neonatal heart rate and peripheral capillary oxygen saturation (SpO2) from facial video. VideoPulse contains 157 recordings totaling 2.6 hours from 52 neonates with diverse face orientations. Our pipeline performs face alignment and artifact aware supervision using denoised pulse oximeter signals, then applies 3D CNN backbones for heart rate and SpO2 regression with label distribution smoothing and weighted regression for SpO2. Predictions are produced in 2 second windows. On the NBHR neonatal dataset, we obtain heart rate MAE 2.97 bpm using 2 second windows (2.80 bpm at 6 second windows) and SpO2 MAE 1.69 percent. Under cross dataset evaluation, the NBHR trained heart rate model attains 5.34 bpm MAE on VideoPulse, and fine tuning an NBHR pretrained SpO2 model on VideoPulse yields MAE 1.68 percent. These results indicate that short unaligned neonatal video segments can support accurate heart rate and SpO2 estimation, enabling low cost non invasive monitoring in neonatal intensive care.

eess.IV↗

arXiv2026

Synthetic Fingerprints for Children Under Four: Generation and Biometric Evaluation

Fingerprint recognition in children under four is of interest for longitudinal identity applications, but research in this age range is constrained by the limited availability and sensitivity of real fingerprint data. Synthetic data may provide a useful complementary resource if generated samples are carefully evaluated for biometric quality, similarity to the real training data, and identity diversity. This paper presents an evaluation and selection protocol for synthetic fingerprints generated from a small pediatric dataset. The protocol was applied to 32,000 candidates produced by an age-conditioned generator fine-tuned with 205 fingerprints from nine children. Candidates were evaluated using NFIQ 2, NBIS minutiae extrac-tion and matching, fingerprint-pattern classification, similarity to the complete real reference set, and pairwise similarity among retained synthetic samples. After candidate filtering and a final symmetric pairwise verification, 1,985 synthetic fingerprints remained. The youngest age group continued to produce retained samples within the sampling budget, whereas the oldest original age group produced 24 retained prints. A data-derived two-group age representation improved pattern-distribution agreement for the younger group. Repeated renderings of fixed synthetic iden-tities produced minutiae-based mated scores comparable to the real mated scores, although a DINOv2 texture representation showed substantially greater within-generator similarity. The results indicate that synthetic pediatric fingerprints can support controlled research use, but their evaluation should include both real-to-synthetic similarity and synthetic-to-synthetic diversity, and conclusions should remain specific to the matchers and measurements used.

eess.IV↗

arXiv2026

Adaptive Color Grading

Independent control of tonescale regions (e.g., shadows, highlights) is essential for painters, photographers and cinematographers to bring 2D images to life. In image manipulation software this is most directly addressed by color grading modules, which use intensity thresholds to segment distinct illumination regions for local manipulation. In this work we develop an open source color grading tool and use it to annotate a large dataset of video frames with tonescale region thresholds. Using these thresholds we conduct modeling experiments with strategies based on both practitioners' conventional wisdom and machine learning. Results show that K-nearest neighbors is an effective prediction strategy, outperforming state-of-the-art end-to-end methods for image enhancement. This outcome demonstrates the benefit of focusing on a compact set of core parameters when modeling creative stylization processes. Our adaptive color grading interface and data are available at https://github.com/SamsungLabs/adaptive-color-grading.

eess.IV↗