AnaDiffusion: Anatomically Compositional Latent Diffusion for Controllable 3D Brain MRI Generation
cs.CV
Submitted: 2026-08-24
Updated: 2026-10-12
Code: https://github.com/phai-lab/AnaDiffusion
Project page: https://anadiffusion.github.io
License: http://creativecommons.org/licenses/by/4.0/
The gist: 3D brain MRI generation has made significant advances in medical imaging, simulation, and controllable anatomical analysis.
Abstract
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.
Sources
- BrainMRDiff: A Diffusion Model for Anatomically Consistent Brain MRI Synthesis
- Cortex-Grounded Diffusion Models for Brain Image Generation
- MONAI: An open-source framework for deep learning in healthcare
- Med3D: Transfer Learning for 3D Medical Image Analysis
- Metrics that matter: Evaluating image quality metrics for medical image generation
- Flatten Wisely: How Patch Order Shapes Mamba-Powered Vision for MRI Segmentation
- MRI Super-Resolution with Deep Learning: A Comprehensive Survey
- Generative AI for Medical Imaging: extending the MONAI Framework
- Variational Approaches for Auto-Encoding Generative Adversarial Networks
- Denoising Diffusion Implicit Models
- Anatomically Guided Latent Diffusion for Brain MRI Progression Modeling
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models