InstantGroup: Instant Template Generation for Scalable Group of Brain MRI Registration

Abstract

Template generation is a critical step in groupwise image registration, which involves aligning a group of subjects into a common space. While existing methods can generate high-quality template images, they often incur substantial time costs or are limited by fixed group scales. In this paper, we present InstantGroup, an efficient groupwise template generation framework based on variational autoencoder (VAE) models that leverage latent representations' arithmetic properties, enabling scalability to groups of any size. InstantGroup features a Dual VAE backbone with shared-weight twin networks to handle pairs of inputs and incorporates a Displacement Inversion Module (DIM) to maintain template unbiasedness and a Subject-Template Alignment Module (STAM) to improve template quality and registration accuracy. Experiments on 3D brain MRI scans from the OASIS and ADNI datasets reveal that InstantGroup dramatically reduces runtime, generating templates within seconds for various group sizes while maintaining superior performance compared to state-of-the-art baselines on quantitative metrics, including unbiasedness and registration accuracy.

14 pages, 10 figures, and 6 tables. Accepted by IEEE Transactions on Image Processing

References in corpus (12)

• Conditional Generative Adversarial Nets
• VoxelMorph: A Learning Framework for Deformable Medical Image Registration
• Domain Adaptation for Medical Image Analysis: A Survey
• An Unsupervised Learning Model for Deformable Medical Image Registration
• Unsupervised Learning of Probabilistic Diffeomorphic Registration for Images and Surfaces
• Unsupervised Learning for Fast Probabilistic Diffeomorphic Registration
• Unsupervised 3D End-to-End Medical Image Registration with Volume Tweening Network
• Cycle Consistent Adversarial Denoising Network for Multiphase Coronary CT Angiography
• Hetero-Modal Variational Encoder-Decoder for Joint Modality Completion and Segmentation
• Context-encoding Variational Autoencoder for Unsupervised Anomaly Detection
• Retrospective correction of Rigid and Non-Rigid MR motion artifacts using GANs
• Hubless keypoint-based 3D deformable groupwise registration