A4-MR: Abnormality-aware Anatomical Alignment for MR
Organizations: GE Healthcare, Seattle, USA · GE Healthcare, Budapest, Hungary · Amazon, USA
Abstract
Recent advances in large-scale foundation models have enabled strong generalization across a wide range of clinical and research tasks in magnetic resonance imaging (MRI). However, existing vision-language MRI models remain limited in anatomy-specific localization and pathology-aware representation, particularly in real-world settings. We introduce A4-MR, a 3D vision-language foundation model trained on approximately 50,000 MRI studies comprising 491,000 series, roughly half of which are paired with radiology reports. A4-MR is designed to: (1) learn anatomy localization through large-scale pretraining with noisy labels; (2) align image and text representations at the anatomical level; and (3) learn distinct representations of healthy and abnormal anatomy. Across a diverse set of downstream tasks, A4-MR outperforms existing foundation models, including a +1.7 AUROC improvement over MedSigLIP for abnormality detection. Using a frozen encoder, A4-MR also surpasses an end-to-end task-specific MedNeXt model for knee segmentation by +1.4 Dice and performs comparably to the best task-specific model for abdominal organ segmentation on AMOS, within 0.4 Dice. Finally, without anatomical prompts, A4-MR localizes 3D anatomy comparably to MedSAM2 provided with ground-truth bounding boxes. Together, these results demonstrate that anatomy-level multimodal alignment and abnormality-aware representation learning can enable a single MRI foundation model to support localization, segmentation, and pathology-related tasks.
Figures & tables
| Entire Report | Organ Descriptions | |||||||
|---|---|---|---|---|---|---|---|---|
| Model | @1 | @3 | @5 | mAP | @1 | @3 | @5 | mAP |
| A 4 -MR (ours) | 93.27 | 86.52 | 84.13 | 90.13 | 85.42 | 74.51 | 68.74 | 77.58 |
| Decipher-MR | 90.90 | 79.60 | 71.90 | 77.80 | 86.60 | 72.30 | 62.80 | 66.70 |
| BiomedCLIP | 81.90 | 69.80 | 64.40 | 70.90 | 66.40 | 48.80 | 41.00 | 51.70 |
| MedImageInsight | 78.00 | 65.10 | 58.10 | 71.70 | 73.00 | 57.30 | 50.50 | 60.70 |
| No Abnormality | 88.25 | 82.49 | 78.87 | 84.38 | 81.47 | 68.71 | 60.76 | 70.66 |
| Entire Report | Organ Descriptions | |||||||
|---|---|---|---|---|---|---|---|---|
| Model | @1 | @3 | @5 | mAP | @1 | @3 | @5 | mAP |
| A 4 -MR (ours) | 96.49 | 94.60 | 92.70 | 88.32 | 86.93 | 82.58 | 79.53 | 78.42 |
| Decipher-MR | 94.60 | 91.40 | 89.70 | 78.10 | 84.50 | 78.80 | 75.10 | 73.80 |
| BiomedCLIP | 83.50 | 72.40 | 68.10 | 75.50 | 66.00 | 52.70 | 47.40 | 62.40 |
| MedImageInsight | 91.90 | 81.10 | 74.90 | 77.20 | 73.90 | 63.70 | 56.90 | 66.80 |
| No Abnormality | 97.30 | 91.89 | 87.84 | 81.16 | 79.92 | 71.28 | 65.65 | 72.24 |
| Abnormality detection (AUROC) | ||||
|---|---|---|---|---|
| Model | 10% | 25% | 50% | 100% |
| A 4 -MR (ours) | 94.7 6.8 | 97.2 2.1 | 98.8 0.7 | 98.9 0.3 |
| Decipher-MR | 86.8 4.8 | 89.7 3.5 | 92.4 2.3 | 94.5 0.6 |
| BiomedParse | 80.0 8.1 | 87.2 6.0 | 89.6 2.7 | 90.7 0.3 |
| Vista3D CT-MR | 78.9 8.1 | 87.8 3.7 | 92.2 1.7 | 93.2 0.7 |
| MRI-CORE | 69.4 5.4 | 73.0 4.7 | 71.9 3.7 | 72.5 0.7 |
| CLS | Mean organ emb | Random PE | Abnormality loss | Liver emb | Pred crop CLS | GT crop CLS | Abnormality detection (AUROC) | ||||
| Model | 10% | 25% | 50% | 100% | |||||||
| A 4 -MR (ours) | ✓ | ✓ | 94.7 6.8 | 97.2 2.1 | 98.8 0.7 | 98.9 0.3 | |||||
| Mean organ | ✓ | ✓ | 89.9 5.7 | 94.6 3.2 | 97.9 1.3 | 99.4 0.2 | |||||
| Random PE | ✓ | ✓ | 80.5 9.7 | 92.6 4.8 | 96.5 1.1 | 97.4 0.7 | |||||
| Liver | ✓ | ✓ | 87.7 5.9 | 94.0 3.1 | 97.0 2.0 | 96.9 0.5 | |||||
| Liver crop CLS | ✓ | ✓ | 95.5 3.7 | 96.8 2.4 | 99.1 0.6 | 99.7 0.2 | |||||
| Model | AMOS | Dataset A (pelvis) | CirrMRI600+ |
|---|---|---|---|
| A 4 -MR (ours) | 85.3 | 87.6 | 96.5 |
| Decipher-MR | 85.7 | 84.3 | 96.3 |
| nnUNet3D | 85.6 | 88.7 | 84.8 |
| Model | Femoral Cartilage | Tibial Cartilage | Patellar Cartilage | Meniscus | Average |
|---|---|---|---|---|---|
| A 4 -MR (ours) | 88.2 | 87.2 | 89.0 | 85.0 | 87.3 |
| MedNeXt | 86.6 | 85.0 | 87.4 | 84.4 | 85.9 |
| VNet | 86.4 | 85.0 | 88.0 | 84.1 | 85.9 |
| nnUNet3D | 79.9 | 79.2 | 86.8 | 79.7 | 81.4 |
| Model | Bounding-box IoU |
|---|---|
| A 4 -MR (zero-shot) | 66.3 |
| No Abnormality (zero-shot) | 65.0 |
| MedSAM2 (1 box) | 65.5 |
| MedSAM2 (3 slices) | 74.8 |
| nnUNet3D (AMOS-trained) | 82.4 |
| Lesion detection | Lesion attributes | Prostate | |||||
|---|---|---|---|---|---|---|---|
| Model | Precision | Recall | F1 | Size agreement | Location ROUGE-F1 | PI-RADS agreement | Volume error (cc) |
| Qwen2.5-VL-3B | 45.00 | 55.33 | 47.60 | 52.05 | 31.34 | 66.94 | 1174.48 |
| Qwen2.5-VL-32B | 26.00 | 26.00 | 26.00 | 60.00 | 33.33 | 75.00 | 20.00 |
| Decipher-MR + Qwen2.5-3B | 53.00 | 49.02 | 49.44 | 64.88 | 27.88 | 83.78 | 20.32 |
| A 4 -MR + Qwen2.5-3B | 53.33 | 49.13 | 49.64 | 63.94 | 24.74 | 80.74 | 16.43 |
| # | Anatomy | # | Anatomy | # | Anatomy |
|---|---|---|---|---|---|
| 1 | spleen | 43 | gluteus_medius_right | 85 | pallidum_right |
| 2 | kidney_right | 44 | gluteus_minimus_left | 86 | thalamus_left |
| 3 | kidney_left | 45 | gluteus_minimus_right | 87 | thalamus_right |
| 4 | gallbladder | 46 | autochthon_left | 88 | hypothalamus_left |
| 5 | liver | 47 | autochthon_right | 89 | hypothalamus_right |
| 6 | stomach | 48 | iliopsoas_left | 90 | ventricle_3rd |
| Decoder | Encoder (pre-training crop size) | Curriculum learning | Initialization | Overall Dice (%) |
|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | No | coldstart | 86.00 |
| segresnet37 | A 4 -MR (384) | Yes | coldstart | 86.20 |
| segresnet37 | A 4 -MR (384) | Yes | warmstart | 86.67 |
| segresnet37 | vit-b + DinoV3 (256) | Yes | coldstart | 86.37 |
| segresnet37 | vit-L + DinoV3 (256) | Yes | coldstart | 86.82 |
| Decoder | Encoder | Curriculum | Initialization | Urethra | Penile bulb | Seminal vesicles |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | No | Coldstart | 57.00 | 80.00 | 80.10 |
| segresnet37 | A 4 -MR (384) | Yes | Coldstart | 56.02 | 86.55 | 79.50 |
| segresnet37 | A 4 -MR (384) | Yes | Warmstart | 57.52 | 86.68 | 81.15 |
| segresnet37 | ViT-B + DINOv3 (256) | Yes | Coldstart | 56.94 | 87.11 | 79.13 |
| segresnet37 | ViT-L + DINOv3 (256) | Yes | Coldstart | 58.30 | 87.24 | 78.97 |
| Decoder | Encoder | Curriculum | Initialization | Prostate | Rectum | Bladder |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | No | Coldstart | 90.00 | 91.20 | 96.00 |
| segresnet37 | A 4 -MR (384) | Yes | Coldstart | 89.80 | 91.63 | 96.06 |
| segresnet37 | A 4 -MR (384) | Yes | Warmstart | 89.70 | 91.55 | 96.25 |
| segresnet37 | ViT-B + DINOv3 (256) | Yes | Coldstart | 89.47 | 91.14 | 95.75 |
| segresnet37 | ViT-L + DINOv3 (256) | Yes | Coldstart | 89.52 | 91.90 | 95.56 |
| Decoder | Encoder | Curriculum | Initialization | Bowel bag | Femoral head (L) | Femoral head (R) |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | No | Coldstart | 84.60 | 90.90 | 85.60 |
| segresnet37 | A 4 -MR (384) | Yes | Coldstart | 84.17 | 90.09 | 85.31 |
| segresnet37 | A 4 -MR (384) | Yes | Warmstart | 84.83 | 91.26 | 85.27 |
| segresnet37 | ViT-B + DINOv3 (256) | Yes | Coldstart | 82.81 | 92.76 | 85.37 |
| segresnet37 | ViT-L + DINOv3 (256) | Yes | Coldstart | 83.50 | 94.09 | 85.24 |
| Decoder | Encoder | Curriculum | Initialization | Pelvis body | Hydrogel |
|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | No | Coldstart | 96.80 | 93.70 |
| segresnet37 | A 4 -MR (384) | Yes | Coldstart | 95.56 | 93.47 |
| segresnet37 | A 4 -MR (384) | Yes | Warmstart | 94.72 | 94.41 |
| segresnet37 | ViT-B + DINOv3 (256) | Yes | Coldstart | 96.16 | 93.40 |
| segresnet37 | ViT-L + DINOv3 (256) | Yes | Coldstart | 96.36 | 94.30 |
| Decoder | Encoder | LR | Initialization | Overall Dice (%) |
|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Coldstart | 83.90 | |
| segresnet37 | ViT-B + DINOv3 (256) | Coldstart | 83.86 | |
| segresnet37 | ViT-L + DINOv3 (256) | Coldstart | 84.24 | |
| segresnet37 † | Decipher-MR (256) | Coldstart | 83.76 | |
| segresnet37 † | A 4 -MR (384) | Coldstart | 83.75 | |
| segresnet37 | A 4 -MR (384) | Warmstart | 84.19 |
| Decoder | Encoder | LR | Init. | Spleen | R. Kidney | L. Kidney |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Cold | 95.37 | 95.10 | 95.58 | |
| segresnet37 | ViT-B + DINOv3 (256) | Cold | 95.51 | 94.49 | 95.42 | |
| segresnet37 | ViT-L + DINOv3 (256) | Cold | 95.93 | 95.42 | 95.68 | |
| segresnet37 † | Decipher-MR (256) | Cold | 95.86 | 95.52 | 95.49 | |
| segresnet37 † | A 4 -MR (384) | Cold | 95.87 | 95.38 | 95.55 | |
| segresnet37 | A 4 -MR (384) | Warm | 95.68 | 95.19 | 95.65 |
| Decoder | Encoder | LR | Init. | Gall Bladder | Esophagus | Liver |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Cold | 75.84 | 77.80 | 97.04 | |
| segresnet37 | ViT-B + DINOv3 (256) | Cold | 75.48 | 77.83 | 97.07 | |
| segresnet37 | ViT-L + DINOv3 (256) | Cold | 79.39 | 76.23 | 96.99 | |
| segresnet37 † | Decipher-MR (256) | Cold | 75.63 | 77.17 | 96.76 | |
| segresnet37 † | A 4 -MR (384) | Cold | 73.97 | 77.59 | 97.06 | |
| segresnet37 | A 4 -MR (384) | Warm | 78.44 | 77.04 | 97.11 |
| Decoder | Encoder | LR | Init. | Stomach | Aorta | Postcava |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Cold | 89.31 | 91.83 | 89.40 | |
| segresnet37 | ViT-B + DINOv3 (256) | Cold | 89.64 | 92.31 | 89.42 | |
| segresnet37 | ViT-L + DINOv3 (256) | Cold | 88.03 | 92.04 | 89.29 | |
| segresnet37 † | Decipher-MR (256) | Cold | 89.93 | 92.16 | 89.42 | |
| segresnet37 † | A 4 -MR (384) | Cold | 89.98 | 91.93 | 89.24 | |
| segresnet37 | A 4 -MR (384) | Warm | 90.56 | 91.94 | 89.83 |
| Decoder | Encoder | LR | Init. | Pancreas | R. Adrenal | L. Adrenal |
|---|---|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Cold | 86.17 | 60.97 | 65.80 | |
| segresnet37 | ViT-B + DINOv3 (256) | Cold | 86.72 | 60.85 | 62.94 | |
| segresnet37 | ViT-L + DINOv3 (256) | Cold | 86.12 | 62.17 | 65.62 | |
| segresnet37 † | Decipher-MR (256) | Cold | 85.41 | 58.97 | 65.09 | |
| segresnet37 † | A 4 -MR (384) | Cold | 86.31 | 60.21 | 63.51 | |
| segresnet37 | A 4 -MR (384) | Warm | 85.87 | 60.62 | 65.25 |
| Decoder | Encoder | LR | Init. | Duodenum |
|---|---|---|---|---|
| segresnet37 | A 4 -MR (384) | Cold | 70.51 | |
| segresnet37 | ViT-B + DINOv3 (256) | Cold | 72.45 | |
| segresnet37 | ViT-L + DINOv3 (256) | Cold | 72.27 | |
| segresnet37 † | Decipher-MR (256) | Cold | 71.48 | |
| segresnet37 † | A 4 -MR (384) | Cold | 72.17 | |
| segresnet37 | A 4 -MR (384) | Warm | 71.30 |
| Dice (%) | mHD (mm) | |||
| Structure | No bottleneck | Bottleneck | No bottleneck | Bottleneck |
| Femoral cartilage | 78.8 9.3 | 78.3 9.4 | 0.519 0.682 | 0.540 0.733 |
| Tibial cartilage | 78.1 9.8 | 77.7 9.6 | 0.581 0.767 | 0.589 0.798 |
| Patellar cartilage | 80.0 11.1 | 80.0 10.8 | 0.586 1.601 | 0.592 1.516 |
| Menisci | 77.8 12.9 | 77.7 12.2 | 0.916 2.131 | 0.849 1.925 |
| All | 78.7 10.9 | 78.4 10.6 | 0.651 1.436 | 0.642 1.345 |
| Model | All pairs ( ) | After exclusion ( ) | |
|---|---|---|---|
| A 4 -MR (zero-shot) | 65.1 | 66.3 | +1.2 |
| No Abnormality | 63.8 | 65.0 | +1.2 |
| MedSAM2 (1 box) | 64.6 | 65.5 | +0.9 |
| MedSAM2 (3 slices) | 74.0 | 74.8 | +0.8 |
| nnUNet3D (AMOS-trained) | 81.0 | 82.4 | +1.4 |
| Organ | A 4 -MR (ours) | No Abnormality | MedSAM2 (1 box) | MedSAM2 (3 slices) | nnUNet3D (AMOS) | |
|---|---|---|---|---|---|---|
| Spleen | 20 | 74.5 | 72.4 | 92.3 | 94.3 | 94.8 |
| Kidney R | 20 | 76.4 | 74.8 | 96.0 | 96.1 | 93.7 |
| Kidney L | 18 | 75.8 | 75.4 | 96.1 | 96.6 | 98.3 |
| Gallbladder | 18 | 52.0 | 52.0 | 70.1 | 75.4 | 68.8 |
| Esophagus | 20 | 61.2 | 59.8 | 38.4 | 54.0 | 79.3 |
| Liver | 20 | 81.1 | 79.2 | 90.1 | 90.1 | 97.3 |
| PICAI-Crop | PICAI | |||||
| Model | 10% | 25% | 100% | 10% | 25% | 100% |
| A 4 -MR (256) | ||||||
| A 4 -MR (384) | ||||||
| A 4 -MR (1st stage) | ||||||
| PICAI-Crop | PICAI | |||||
| Model | 10% | 25% | 100% | 10% | 25% | 100% |
| A 4 -MR (256) | ||||||
| A 4 -MR (384) | ||||||
| A 4 -MR (1st stage) | ||||||