mlsanityData to 14 Oct 2026

Atlases Are Already Inside: Recovering Population Templates from Pretrained Diffusion Models

arXiv 2609.30566 ↗First posted 24 Sep 2026cs.CVcited by 0

On the map: Diffusion models → Medical imaging and segmentation

Abstract (arXiv)

We present a new inference-time sampler for diffusion models that gives a pretrained model a capability it was never trained for: constructing the atlas of the population it synthesizes. The sampler converges from every random seed to the population's central anatomy, which we call the intrinsic atlas. The advantage is threefold. (1) It requires no retraining. A diffusion model that has already learned a coherent population, including the released ones, yields its atlas in a single inference pass without involving deformable registration. (2) It applies to multiple domains, such as brain MRI, chest X-ray, faces, and 3D shapes. (3) It extends to subpopulations. One age-conditioned model gives an atlas at any age in its training range, and the resulting family reproduces the CSF expansion of healthy aging. Evaluated as a registration target, the intrinsic atlas is best or second-best on every dataset against classical and learned templates, and the most central template on held-out brain MRI cohorts. Atlas construction can be reframed as a byproduct of generative modeling: a diffusion model is a learned representation of population structure, and the atlas is what it already contains.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 24 of its 47 references that are in the map. First posted on arXiv 24 Sep 2026.

Builds on

24 of 47 references.

Closest ancestors · 2

Same direction, newest first

  1. 2022Medical Diffusion: Denoising Diffusion Probabilistic Models for 3D Medical Image Generation7 Nov
  2. 2021DiffuseMorph: Unsupervised Deformable Image Registration Using Diffusion Model9 Dec

Other references · 22

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2020Denoising Diffusion Implicit Modelscited by 9,781
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  4. 2014Deep Learning Face Attributes in the Wildcited by 3,693
  5. 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,937
  6. 2018VoxelMorph: A Learning Framework for Deformable Medical Image Registrationcited by 662
  7. 2023Emergent Correspondence from Image Diffusioncited by 446
  8. 2023An Edit Friendly DDPM Noise Space: Inversion and Manipulationscited by 244
  9. 2023Diffusion Hyperfeatures: Searching Through Time and Space for Semantic Correspondencecited by 214
  10. 2021Learn2Reg: comprehensive multi-task medical image registration challenge, dataset and evaluation in the era of deep learningcited by 125
  11. 2020Deformed Implicit Field: Modeling 3D Shapes with Learned Dense Correspondencecited by 119
  12. 2020Deep Implicit Templates for 3D Shape Representationcited by 86
10 more
  1. 2019DeepAtlas: Joint Semi-Supervised Learning of Image Registration and Segmentationcited by 66
  2. 2019Learning Conditional Deformable Templates with Convolutional Networkscited by 65
  3. 2021GAN-Supervised Dense Visual Alignmentcited by 62
  4. 2020KeypointNet: A Large-scale 3D Keypoint Dataset Aggregated from Numerous Human Annotationscited by 51
  5. 2021Generative Adversarial Registration for Improved Conditional Deformable Templatescited by 31
  6. 2023Neural Congealing: Aligning Images to a Joint Semantic Atlascited by 31
  7. 2022Aladdin: Joint Atlas Building and Diffeomorphic Registration Learning with Pairwise Alignmentcited by 18
  8. 2025MultiMorph: On-demand Atlas Constructioncited by 9
  9. 2023Continual atlas-based segmentation of prostate MRIcited by 5
  10. 2025AtlasMorph: Learning conditional deformable templates for brain MRIcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Diff-Def: Diffusion-Generated Deformation Fields for Conditional Atlases
  2. 2025Deep generative priors for 3D brain analysis
  3. 2024TADM: Temporally-Aware Diffusion Model for Neurodegenerative Progression on Brain MRI
  4. 20264DLoG: Generative Modeling of Neurodegenerative Brain Anatomy with 4D Longitudinal Diffusion Model
  5. 2026A continually expandable foundation model for brain MRI
  6. 2025Identity Preserving Latent Diffusion for Brain Aging Modeling
  7. 2025Integrating Anatomical Priors into a Causal Diffusion Model
  8. 2022Brain Imaging Generation with Latent Diffusion Models