Learning Subject-Specific Anatomical Representations via Manifold Expansion: Application to Accelerated Multi-Contrast MRI
Authors: Ruimin Feng, Wanyu Bian, Albert Jang, Zachary Stewart, Fang Liu
Organizations: Athinoula A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital, Charlestown, Massachusetts, United States · Harvard Medical School, Boston, Massachusetts, United States · Department of Radiology, Massachusetts General Hospital, Charlestown, Massachusetts, United States
Clinical MRI routinely acquires multiple contrast-weighted images of the same anatomy for complementary tissue characterization. However, current accelerated MRI methods typically reconstruct each contrast independently, without fully exploiting shared anatomical information. This work aims to learn anatomical representations invariant to contrast-dependent appearance for reconstruction of accelerated multi-contrast MRI. We propose MAX (MAnifold eXpansion), a subject-specific framework that learns anatomical representations from a single fully sampled reference contrast. To address the under-constrained separation of shared anatomy and contrast-dependent components from a single image, MAX expands the multi-contrast manifold using anatomy-preserving intensity augmentations. A disentangled implicit neural representation models augmented samples using shared spatial coordinates for anatomy and spatially invariant coordinates for contrast appearance. The learned anatomical representation is then fixed, with the contrast representation adapted to the undersampled target data, followed by unrolled refinement. Theoretical analyses further provide insight into the disentangled representation learning and explain how the learned anatomical representation improves the target contrast reconstruction. At R = 8 for brain MRI and R = 6 for knee MRI, MAX achieves the highest mean PSNR and SSIM across all tasks, improving PSNR by more than 1 dB over the strongest baseline for both brain contrasts. MAX more faithfully recovers subtle anatomical and pathological structures and remains robust to inter-contrast motion, structural heterogeneity between reference and target contrasts, and measurement noise. Therefore, MAX provides a general strategy for leveraging high-quality reference scans in accelerated MRI and has the potential to be extended to other reference-assisted MRI inverse problems.
Institute of Machine Learning in Biomedical Imaging, Helmholtz Munich, Germany · School of Computation and Information Technology, Technical University of Munich, Germany · Department of Radiology, Boston Children’s Hospital and Harvard Medical School, USA +3
School of Biomedical Engineering & Imaging Sciences, King’s College London, London, UK. · University College London, London, UK. · deepc GMBH, Munich, Germany.
Department of Radiology, Leiden University Medical Center, Leiden, The Netherlands · Philips Innovative Technologies, Hamburg, Germany · Philips Cardiologs, Paris, France +1