AnaDiffusion: Anatomically CompositionalLatent Diffusion for Controllable 3D Brain MRI Generation
On the map: Diffusion models → Medical imaging and segmentation
Abstract (arXiv)
3D brain MRI generation has made significant advances in medical imaging, simulation, and controllable anatomical analysis. However, existing generative models typically synthesize 3D volumes monolithically, often overlooking regional anatomical structures and limiting local controllability. To address these limitations, we introduce AnaDiffusion, an anatomically compositional latent diffusion framework that factorizes the generation process into distinct, anatomically meaningful regions, followed by part-to-whole assembly and global refinement. Our approach first trains part diffusion models to capture local structural priors. We then inject an assembled anatomical composite of the parts into the whole-brain latent representation and continue denoising. This mechanism enables the model to resolve global context while preserving the injected anatomy. As a result, AnaDiffusion produces both explicit part assets and a globally coherent volume, thereby enabling controllable part editing without requiring subject-specific dense segmentation maps at inference time while maintaining consistent part-to-whole brain structure. On the subject-disjoint ADNI test split, AnaDiffusion achieves the lowest FID across the whole brain, left and right hemispheres, cerebellar-brainstem complex, and seam regions. It also achieves the best cerebellar and second-best ventricular and brainstem absolute Cohen's d values among the evaluated methods. In localized editing experiments, paired MS-SSIM demonstrates high target transfer and off-target preservation, supporting controllable part replacement with minimal unintended anatomical alterations.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 33 of its 52 references that are in the map. First posted on arXiv 24 Aug 2026.
Builds on
33 of 52 references.
Closest ancestors · 11
Same direction, newest first
- 2026Cortex-Grounded Diffusion Models for Brain Image Generation27 Jan
- 2026Anatomically Guided Latent Diffusion for Brain MRI Progression Modeling21 Jan
- 2025BrainMRDiff: A Diffusion Model for Anatomically Consistent Brain MRI Synthesis6 Apr
- 2025Diffusion Models for conditional MRI generation25 Feb
- 2024Enhancing Spatiotemporal Disease Progression Models via Latent Diffusion and Prior Knowledge6 May
- 2024Anatomically-Controllable Medical Image Generation with Segmentation-Guided Diffusion Models7 Feb
- 2023Adaptive Latent Diffusion Model for 3D Medical Image to Image Translation: Multi-modal Magnetic Resonance Imaging Study1 Nov
- 2023InverseSR: 3D Brain MRI Super-Resolution Using a Latent Diffusion Model23 Aug
- 2023Conditional Diffusion Models for Semantic 3D Brain MRI Synthesis29 May
- 2022Generating Realistic Brain MRIs via a Conditional Diffusion Probabilistic Model15 Dec
- 2022Brain Imaging Generation with Latent Diffusion Models15 Sep
Other references · 22
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
- 2022MONAI: An open-source framework for deep learning in healthcarecited by 438
- 2019Med3D: Transfer Learning for 3D Medical Image Analysiscited by 282
- 2021SynthSeg: Segmentation of brain MRI scans of any contrast and resolution without retrainingcited by 235
- 2022SynthStrip: Skull-Stripping for Any Brain Imagecited by 126
- 2023Generative AI for Medical Imaging: extending the MONAI Frameworkcited by 68
- 2019Generation of 3D Brain MRI Using Auto-Encoding Generative Adversarial Networkscited by 65
- 2020Hierarchical Amortized Training for Memory-efficient High Resolution 3D GANcited by 60
10 more
- 2017Variational Approaches for Auto-Encoding Generative Adversarial Networkscited by 52
- 2022A Survey on Training Challenges in Generative Adversarial Networks for Biomedical Image Analysiscited by 24
- 2025Metrics that matter: Evaluating image quality metrics for medical image generationcited by 12
- 2025Synthesizing Individualized Aging Brains in Health and Disease with Generative Models and Parallel Transportcited by 9
- 2025MRI Super-Resolution with Deep Learning: A Comprehensive Surveycited by 7
- 2024Evaluating the Quality of Brain MRI Generatorscited by 5
- 2025WASABI: A Metric for Evaluating Morphometric Plausibility of Synthetic Brain MRIscited by 5
- 2025IGG: Image Generation Informed by Geodesic Dynamics in Deformation Spacescited by 4
- 2025REOrdering Patches Improves Vision Modelscited by 3
- 2025Flatten Wisely: How Patch Order Shapes Mamba-Powered Vision for MRI Segmentationcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Anatomica: Localized Control over Geometric and Topological Properties for Anatomical Diffusion Models
- 20243D MedDiffusion: A 3D Medical Latent Diffusion Model for Controllable and High-quality Medical Image Generation
- 2025Deep generative priors for 3D brain analysis
- 2024Physics-Informed Latent Diffusion for Multimodal Brain MRI Synthesis
- 2023Make-A-Volume: Leveraging Latent Diffusion Models for Cross-Modality 3D Brain MRI Synthesis
- 2025CA-Diff: Collaborative Anatomy Diffusion for Brain Tissue Segmentation
- 2025RealDeal: Enhancing Realism and Details in Brain Image Generation via Image-to-Image Diffusion Models
- 2024X-Diffusion: Generating Detailed 3D MRI Volumes From a Single Image Using Cross-Sectional Diffusion Models