MosaicMRI: A Diverse Dataset and Benchmark for Raw Musculoskeletal MRI
Organizations: Department of Computer Science, University of Southern California · USC Center on AI Foundations for the Sciences · Department of Electrical and Computer Engineering, University of Southern California · Department of Radiology and Imaging Sciences, University of Utah
Abstract
Deep learning underpins a wide range of applications in MRI, including reconstruction, artifact removal, and segmentation. However, progress has been driven largely by public datasets focused on brain and knee imaging, shaping how models are trained and evaluated. As a result, careful studies of the reliability of these models across diverse anatomical settings remain limited. In this work, we introduce MosaicMRI, a large and diverse collection of fully sampled raw musculoskeletal (MSK) MR measurements designed for training and evaluating machine-learning--based methods. MosaicMRI is the largest open-source raw MSK MRI dataset to date, comprising 3,150 volumes and 89,830 slices, including a 3,096-volume 1.5T core dataset and a 54-volume 0.55T low-field set. The dataset offers substantial diversity in volume orientation (e.g., axial, sagittal), imaging contrasts (e.g., PD, T1, T2), anatomies (e.g., spine, knee, hip, ankle, and others), numbers of acquisition coils, and field strengths. Using accelerated reconstruction as a testbed, we study scaling, robustness, and data selection under realistic MSK distribution shifts. Across E2E-VarNet, U-Net, and ViT baselines, mixed-anatomy training improves over anatomy-specific training, showing that the benefit of anatomical diversity is not architecture-specific. Zero-shot experiments on the 0.55~T low-field set highlight that using more training data improves average reconstruction under field-strength shift. Controlled subset experiments further suggest that these gains come from transferable cross-anatomy structure, rather than anatomy or contrast matching alone.
Figures & tables
| Dataset | Anatomy | View | Contrast | Vendor | Magnet | Coils | # Scans / # Subjects |
| fastMRI brain [ 33 ] | brain | axial | T1, T1POST, T2, FLAIR | Siemens | 1.5T, 3T | 4-20 | 6.4k / 6.4k |
| OCMR [ 7 ] | heart | various | SSFP | Siemens | 0.5T - 3T | 15-38 | 165 / 165 |
| AHEAD [ 3 ] | brain | various | MP2RAGE-ME | Philips | 7T | 32 | 105 / 105 |
| M4Raw [ 23 ] | brain | axial | T1, T2, FLAIR | XGY | 0.3T | 4 | 1.3k / 183 |
| CMRxRecon2023 [ 32 ] | heart | various | SSFP-Balanced | Siemens | 3T | 10 | 300 / 300 |
| fastMRI prostate [ 29 ] | prostate | axial | T2, DWI | Siemens | 3T | 10-30 | 312 / 312 |
| Dataset split | # scans | # patients | # slices |
|---|---|---|---|
| train | 2,166 | 598 | 61,903 |
| val | 465 | 128 | 13,265 |
| test | 465 | 128 | 13,248 |
| 0.55T low-field | 54 | 8 | 1,414 |
| accel. | accel. | |||
| Method | PSNR (dB) | SSIM | PSNR (dB) | SSIM |
| ESPIRiT | ||||
| E2E-VarNet | ||||
| U-Net | – | – | ||
| ViT | – | – | ||
| LBM (1 step) | – | – | ||
| Model | 10% | 20% | 50% | 100% |
|---|---|---|---|---|
| U-Net | 35.78 / 0.898 | 36.36 / 0.903 | 36.99 / 0.909 | 37.44 / 0.913 |
| ViT | 32.73 / 0.853 | 32.04 / 0.837 | 35.35 / 0.890 | 36.61 / 0.904 |
| Model | Test anatomy | Target anatomy only | Full MosaicMRI |
|---|---|---|---|
| U-Net | Knee | 37.35 / 0.894 | 37.94 / 0.898 |
| U-Net | Shoulder | 42.05 / 0.952 | 42.41 / 0.954 |
| ViT | Knee | 34.01 / 0.850 | 37.47 / 0.892 |
| ViT | Shoulder | 35.79 / 0.912 | 41.66 / 0.950 |
| Model trained on | |||
| Test set | fastMRI | MosaicMRI | MosaicMRI (knee) |
| fastMRI | 36.90 / 0.888 | 35.89 / 0.876 | 33.13 / 0.832 |
| MosaicMRI | 34.76 / 0.886 | 38.57 / 0.924 | 35.81 / 0.889 |
| MosaicMRI (knee) | 33.04 / 0.849 | 38.09 / 0.891 | 37.95 / 0.890 |
| Broad/full training | Matched slice budget | |||||
| Test anatomy (contrast) | Anatomy-only | Contrast-specific | MosaicMRI | Slice-matched random | Anatomy+contrast | DreamSim |
| Knee (PD-FS) | 34.41 / 0.802 | 34.54 / 0.803 | 34.61 / 0.806 | 32.95 / 0.781 | 33.40 / 0.787 | 33.75 / 0.792 |
| Training slices | 5,864 | 8,661 | 61,903 | 541 | 541 | 541 |
| Shoulder (PD-FS) | 44.64 / 0.974 | 44.39 / 0.973 | 44.63 / 0.974 | 42.96 / 0.968 | 42.81 / 0.968 | 43.63 / 0.971 |
| Training slices | 8,534 | 8,661 | 61,903 | 1,744 | 1,744 | 1,744 |
| Test set | |||||
|---|---|---|---|---|---|
| Model | Training data | Knee | Shoulder | Spine | Average |
| MosaicMRI (10%) | 29.285 / 0.753 | 28.794 / 0.676 | 28.385 / 0.708 | 28.821 / 0.712 | |
| E2E-VarNet | MosaicMRI (50%) | 29.836 / 0.773 | 29.028 / 0.688 | 28.443 / 0.720 | 29.102 / 0.727 |
| MosaicMRI | 30.184 / 0.788 | 29.382 / 0.711 | 29.202 / 0.757 | 29.590 / 0.752 | |
| MosaicMRI (10%) | 29.759 / 0.805 | 28.449 / 0.712 | 29.205 / 0.777 | 29.093 / 0.762 | |
| U-Net | MosaicMRI (50%) | 29.855 / 0.798 | 28.858 / 0.715 | 29.564 / 0.780 | 29.396 / 0.762 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Metric | spine | knee | shoulder | hip | ankle | foot | wrist/hand | elbow | pelvis | tib/fib |
|---|---|---|---|---|---|---|---|---|---|---|
| FOV x (mm) | 238.28 | 159.42 | 162.69 | 281.82 | 165.86 | 155.96 | 109.46 | 167.38 | 305.26 | 257.37 |
| FOV y (mm) | 237.46 | 156.31 | 162.43 | 278.46 | 166.59 | 150.62 | 99.13 | 166.36 | 294.00 | 243.40 |
| Res x (mm) | 0.7016 | 0.3674 | 0.3076 | 0.7881 | 0.4483 | 0.4364 | 0.3185 | 0.5200 | 0.8628 | 0.7128 |
| Res y (mm) | 0.7073 | 0.4138 | 0.4254 | 0.8983 | 0.4483 | 0.4527 | 0.3319 | 0.5620 | 0.9485 | 0.8700 |