# MultiMorph: On-demand Atlas Construction

arXiv: https://arxiv.org/abs/2504.00247 · HTML: https://mlsanity.com/paper/2504.00247/

## Key facts

- Cited by 9 papers in the map (data through 14 Oct 2026).
- Builds on 32 of its 90 references that are in the map.
- First posted on arXiv 31 Mar 2025.

## Builds on (most cited first, 30 of 32)

- [U-Net: Convolutional Networks for Biomedical Image Segmentation](https://mlsanity.com/paper/1505.04597/) (2015)
- [Spatial Transformer Networks](https://mlsanity.com/paper/1506.02025/) (2015)
- [VoxelMorph: A Learning Framework for Deformable Medical Image Registration](https://mlsanity.com/paper/1809.05231/) (2018)
- [Geometric deep learning on graphs and manifolds using mixture model CNNs](https://mlsanity.com/paper/1611.08402/) (2016)
- [SynthSeg: Segmentation of brain MRI scans of any contrast and resolution without retraining](https://mlsanity.com/paper/2107.09559/) (2021)
- [Unsupervised Learning of Probabilistic Diffeomorphic Registration for Images and Surfaces](https://mlsanity.com/paper/1903.03545/) (2019)
- [SynthMorph: learning contrast-invariant registration without acquired images](https://mlsanity.com/paper/2004.10282/) (2020)
- [SynthStrip: Skull-Stripping for Any Brain Image](https://mlsanity.com/paper/2203.09974/) (2022)
- [Quicksilver: Fast Predictive Image Registration - a Deep Learning Approach](https://mlsanity.com/paper/1703.10908/) (2017)
- [Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning](https://mlsanity.com/paper/2106.02584/) (2021)
- [End-to-End Unsupervised Deformable Image Registration with a Convolutional Neural Network](https://mlsanity.com/paper/1704.06065/) (2017)
- [Learning Conditional Deformable Templates with Convolutional Networks](https://mlsanity.com/paper/1908.02738/) (2019)
- [Learning a Probabilistic Model for Diffeomorphic Registration](https://mlsanity.com/paper/1812.07460/) (2018)
- [Robust machine learning segmentation for large-scale analysis of heterogeneous clinical brain MRI datasets](https://mlsanity.com/paper/2209.02032/) (2022)
- [DenseReg: Fully Convolutional Dense Shape Regression In-the-Wild](https://mlsanity.com/paper/1803.02188/) (2018)
- [Generative Adversarial Registration for Improved Conditional Deformable Templates](https://mlsanity.com/paper/2105.04349/) (2021)
- [Learning General-Purpose Biomedical Volume Representations using Randomized Synthesis](https://mlsanity.com/paper/2411.02372/) (2024)
- [SVoRT: Iterative Transformer for Slice-to-Volume Registration in Fetal Brain MRI](https://mlsanity.com/paper/2206.10802/) (2022)
- [Atlas-ISTN: Joint Segmentation, Registration and Atlas Construction with Image-and-Spatial Transformer Networks](https://mlsanity.com/paper/2012.10533/) (2020)
- [A Robust and Interpretable Deep Learning Framework for Multi-modal Registration via Keypoints](https://mlsanity.com/paper/2304.09941/) (2023)
- [Neuralizer: General Neuroimage Analysis without Re-Training](https://mlsanity.com/paper/2305.02644/) (2023)
- [Non-iterative Coarse-to-fine Transformer Networks for Joint Affine and Deformable Image Registration](https://mlsanity.com/paper/2307.03421/) (2023)
- [Anatomy-aware and acquisition-agnostic joint registration with SynthMorph](https://mlsanity.com/paper/2301.11329/) (2023)
- [Aladdin: Joint Atlas Building and Diffeomorphic Registration Learning with Pairwise Alignment](https://mlsanity.com/paper/2202.03563/) (2022)
- [Partial Volume Segmentation of Brain MRI Scans of any Resolution and Contrast](https://mlsanity.com/paper/2004.10221/) (2020)
- [CINA: Conditional Implicit Neural Atlas for Spatio-Temporal Representation of Fetal Brains](https://mlsanity.com/paper/2403.08550/) (2024)
- [SuperWarp: Supervised Learning and Warping on U-Net for Invariant Subvoxel-Precise Registration](https://mlsanity.com/paper/2205.07399/) (2022)
- [Boosting Skull-Stripping Performance for Pediatric Brain Images](https://mlsanity.com/paper/2402.16634/) (2024)
- [Diff-Def: Diffusion-Generated Deformation Fields for Conditional Atlases](https://mlsanity.com/paper/2403.16776/) (2024)
- [Registration by Regression (RbR): a framework for interpretable and flexible atlas registration](https://mlsanity.com/paper/2404.16781/) (2024)

## Built on by (most cited first, 9 of 9)

- [OmniBrainBench: A Comprehensive Multimodal Benchmark for Brain Imaging Analysis Across Multi-stage Clinical Tasks](https://mlsanity.com/paper/2511.00846/) (2025)
- [Conditional Fetal Brain Atlas Learning for Automatic Tissue Segmentation](https://mlsanity.com/paper/2508.04522/) (2025)
- [Deep infant brain segmentation from multi-contrast MRI](https://mlsanity.com/paper/2512.05114/) (2025)
- [Unified Brain Surface and Volume Registration](https://mlsanity.com/paper/2512.19928/) (2025)
- [Volumetrically Consistent Implicit Atlas Learning via Neural Diffeomorphic Flow for Placenta MRI](https://mlsanity.com/paper/2603.16078/) (2026)
- [Set-Based Groupwise Registration for Variable-Length, Variable-Contrast Cardiac MRI](https://mlsanity.com/paper/2605.10571/) (2026)
- [Deep Image Prototype Learning with Geometric Heat-Kernel Priors](https://mlsanity.com/paper/2606.18658/) (2026)
- [textttDisMorph: learning to disentangle technical distortions from true biological change](https://mlsanity.com/paper/2608.08173/) (2026)
- [Atlases Are Already Inside: Recovering Population Templates from Pretrained Diffusion Models](https://mlsanity.com/paper/2609.30566/) (2026)

## Abstract (arXiv)

We present MultiMorph, a fast and efficient method for constructing anatomical atlases on the fly. Atlases capture the canonical structure of a collection of images and are essential for quantifying anatomical variability across populations. However, current atlas construction methods often require days to weeks of computation, thereby discouraging rapid experimentation. As a result, many scientific studies rely on suboptimal, precomputed atlases from mismatched populations, negatively impacting downstream analyses. MultiMorph addresses these challenges with a feedforward model that rapidly produces high-quality, population-specific atlases in a single forward pass for any 3D brain dataset, without any fine-tuning or optimization. MultiMorph is based on a linear group-interaction layer that aggregates and shares features within the group of input images. Further, by leveraging auxiliary synthetic data, MultiMorph generalizes to new imaging modalities and population groups at test-time. Experimentally, MultiMorph outperforms state-of-the-art optimization-based and learning-based atlas construction methods in both small and large population settings, with a 100-fold reduction in time. This makes MultiMorph an accessible framework for biomedical researchers without machine learning expertise, enabling rapid, high-quality atlas generation for diverse studies.
