Reverse Imaging: Any-Sequence Generalization for Cardiac MRI Segmentation.

Abstract

Pretrained segmentation models for cardiac magnetic resonance imaging (MRI) often fail to generalize across imaging sequences due to substantial contrast variations. These variations arise from different imaging protocols, yet fundamentally, all contrasts are governed by the same underlying tissue properties, primarily captured by three components: the magnetization strength (M0), T1, and T2. Building on this insight, we introduce Reverse Imaging, a physics-driven framework for data augmentation and domain generalization in cardiac MRI. Our method infers tissue properties from observed MR images with annotation by solving an ill-posed nonlinear inverse problem, regularized by a generative prior. The prior is learned from the multiparametric saturation-recovery single shot acquisition (mSASHA) dataset for joint cardiac T1 and T2 mapping. In inference, we characterize imaging sequences as weak, moderate, or strong observations according to the physical information they provide and the degree of ill-posedness. This motivates an iterative prior-learning strategy that uses moderate T1-mapping observations to alleviate mSASHA data scarcity via pseudo tissue-property estimates. We further integrate MRI physics into posterior inference by expressing the sequence model as a likelihood term guiding the reverse diffusion process. For widely used but weak cine observations, we develop a sequence-specific ControlNet to improve efficiency and spatial consistency. Extensive experiments on eight unseen cardiac MRI sequences with markedly different contrast mechanisms show that Reverse Imaging yields plausible tissue-property estimates, supports synthesis of diverse yet physically consistent contrasts, and improves segmentation robustness under severe cross-sequence shifts.