Normal is All You Need: A Symmetry-Informed Inverse Learning Foundation Model for Neuroimaging Diagnostics
Wang, S.; Ayubcha, C.; Hua, Y.; Beam, A.
Show abstract
BackgroundDeveloping generalizable neuroimaging models is often hindered by limited labeled data which has led to an increased interest in unsupervised inverse learning. Existing approaches often neglect geometric principles and struggle with diverse pathologies. We propose a symmetry-informed inverse learning foundation model to address these shortcomings for robust and efficient anomaly detection in brain MRI. MethodsOur framework employs a reconstruction-to-embedding pipeline, trained exclusively on healthy brain MRI slices. A 2D U-Net uses a novel, symmetry-aware masking strategy to reconstruct a disorder-free slice. Difference maps are embedded into a 1024-dimensional latent space via a Beta-VAE. Anomaly scoring is performed using Mahalanobis distance. We evaluated generalization by fine-tuning on external lesion datasets, BraTS Africa (SSA), and the ADNI-derived Alzheimers disease cohort (Alz). ResultsOn the source metastasis (Mets) dataset, the framework achieved high performance (AB1+MSE: 99.28% accuracy, 99.79% sensitivity). Generalization to the external lesion dataset (SSA) was robust, with the Symmetry ROC configuration achieving 91.93% accuracy. Transfer to the Alzheimers dataset (Alz) was more challenging, achieving a peak accuracy of 70.54% with a high false-positive rate, suggesting difficulty in separating subtle, diffuse changes. ConclusionThe symmetry-informed inverse learning framework establishes a robust foundation model for neuroimaging, showing strong performance for focal lesions and successful generalization under domain shift. Limitations in diffuse neurodegeneration underscore the necessity for richer representations and multimodal integration to improve future foundation models. Summary StatementA symmetry-informed inverse learning framework trained on normal brain MRI achieved high accuracy for detecting focal lesions and demonstrated strong generalization across external datasets under domain shift. Key Points[bullet] A symmetry-informed disorder-free reconstruction framework trained only on normal brain MRI achieved 99.28% accuracy and 99.79% sensitivity for metastasis detection on the BrainMetShare dataset, demonstrating non-inferior performance compared with all but one strategy while offering improved computational efficiency. [bullet]The model generalized effectively to an external tumor dataset (BraTS SSA), achieving up to 91.93% accuracy using receiver operating characteristic-optimized thresholding with minimal fine-tuning. [bullet]Embedding-based anomaly detection using Mahalanobis distance enabled consistent separation between normal and abnormal slices, supporting robust and interpretable anomaly detection across datasets.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.