OpenMASC: An Open-Source Pipeline for Cross-Trajectory Metal-Aware Sampling and Correction in Accelerated MRI
Authors: Zhengyi Lu, Ming Lu, Chongyu Qu, Junchao Zhu, Junlin Guo, Marilyn Lionts, Yanfan Zhu, Yuechen Yang, +6 more
Organizations: Vanderbilt University, Nashville, TN, USA · Vanderbilt University Medical Center, Nashville, TN, USA · Oak Ridge National Laboratory, Oak Ridge, TN, USA
Metal implants corrupt MRI measurements throughout k-space, yet existing accelerated MRI methods assume clean data and most metal artifact reduction approaches assume fully sampled acquisitions. No public dataset provides paired k-space and images with and without metal for the same anatomy, and no framework jointly addresses artifact-aware acquisition and reconstruction across sampling trajectories. We present OpenMASC, an open-source pipeline covering the full workflow from data generation to deployment. A physics-based data generation module converts public CT volumes into paired clean and metal-corrupted MRI data in both Cartesian and radial formats. MA-VarNet, an unrolled reconstruction network with a per-cascade DC Rectifier, corrects artifacts that data-consistency steps reintroduce from corrupted measurements. A reinforcement learning agent actively selects k-space readouts and co-trains with the reconstruction network through a decoupled three-stage procedure. The framework is trajectory-agnostic except for the data-consistency operator, supporting both Cartesian and radial acquisition without architectural changes. Experiments on two datasets at 4× and 8× acceleration demonstrate consistent improvements over conventional and learned baselines on both trajectories.
Figures & tables
Figure 1: Overview of the OpenMASC pipeline. The user selects a trajectory type (Cartesian or radial) and an acceleration factor ( 4× or 8× ); the toolbox handles the rest. (A) The Data Generation module converts existing CT volumes into paired clean and metal-corrupted MRI data via physics-based simulation, in either Cartesian or radial format. (B) An active sampling agent and MA-VarNet (metal-aware reconstruction) are jointly trained on the generated data. (C) At deployment, the agent iteratively selects k -space readouts while MA-VarNet reconstructs, repeating until the acquisition budget is exhausted to produce an artifact-reduced image.
Figure 2: Data Generation pipeline. Stage 1: CT volumes from the autoPET dataset are automatically segmented into multi-tissue phantoms. The user interactively selects the implant site and positions the virtual prosthesis, producing paired digital phantoms (with and without metal) and six voxel-wise MR property maps. Stage 2: An open-source MRI simulator generates paired Cartesian k -space and reconstructed images via a TSE sequence. Stage 3: Radial k -space and images are derived from the same paired data via NUFFT using MRI-NUFFT.
Method
Simulation 4 ×
Simulation 8 ×
fastMRI 4 ×
fastMRI 8 ×
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
Rand (VD) + E2E-VarNet
0.756 ± 0.015
25.83 ± 1.75
0.747 ± 0.016
25.46 ± 1.69
0.749 ± 0.029
28.72 ± 1.33
0.679 ± 0.035
27.77 ± 1.29
Equi + E2E-VarNet
0.746 ± 0.017
25.45 ± 1.72
0.716 ± 0.022
24.07 ± 1.57
0.727 ± 0.027
28.14 ± 1.28
0.651 ± 0.032
27.00 ± 1.37
Rand (VD) + FI-VarNet
0.759 ± 0.014
26.01 ± 1.71
0.752 ± 0.015
25.67 ± 1.65
0.751 ± 0.029
28.89 ± 1.27
0.683 ± 0.035
27.97 ± 1.24
Equi + FI-VarNet
0.750 ± 0.015
25.64 ± 1.68
0.721 ± 0.021
24.31 ± 1.55
0.731 ± 0.027
28.26 ± 1.27
0.656 ± 0.033
27.10 ± 1.35
SS-DDQN + FI-VarNet
0.760 ± 0.014
26.01 ± 1.70
0.755 ± 0.014
25.81 ± 1.65
0.750 ± 0.028
28.70 ± 1.23
0.685 ± 0.036
28.08 ± 1.23
Table 1: Cartesian k -space results at 4× and 8× acceleration. Each method combines a sampling strategy with a reconstruction network. The simulation dataset is generated from autoPET via the Data Generation module, and fastMRI + augment uses a different implant model for cross-dataset evaluation. Best results in bold.
Figure 3: Cartesian k -space results shown as violin plots. Top row: SSIM; bottom row: PSNR. Left column: simulation dataset; right column: fastMRI + augment. Each method shows paired distributions for 4× (blue) and 8× (orange) acceleration. OpenMASC (red, rightmost) achieves the highest median and narrowest spread across all conditions.
Figure 4: Qualitative Cartesian results on the simulation dataset at 4× acceleration. Rows from top to bottom: reconstructed images with region of interest highlighted by red dashed boxes, error maps, sampling masks, and action step visualizations colored by acquisition order from Step 0 to Step 42. OpenMASC produces reconstructions closest to the ground truth with the darkest error maps.
Method
Simulation 4 ×
Simulation 8 ×
fastMRI 4 ×
fastMRI 8 ×
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
SSIM ↑
PSNR ↑
Ran + E2E-VarNet
0.651 ± 0.027
21.73 ± 1.27
0.599 ± 0.032
19.86 ± 1.29
0.687 ± 0.038
24.23 ± 2.02
0.622 ± 0.042
22.95 ± 1.98
GA + E2E-VarNet
0.663 ± 0.025
22.12 ± 1.27
0.619 ± 0.030
20.91 ± 1.21
0.698 ± 0.035
25.15 ± 1.90
0.638 ± 0.039
24.02 ± 1.88
Equi + E2E-VarNet
0.667 ± 0.027
22.35 ± 1.30
0.623 ± 0.031
21.00 ± 1.31
0.681 ± 0.035
24.84 ± 1.94
0.633 ± 0.038
23.97 ± 1.91
Rand + FI-VarNet
0.663 ± 0.020
22.63 ± 1.28
0.610 ± 0.027
20.44 ± 1.30
0.711 ± 0.034
25.34 ± 2.11
0.645 ± 0.040
23.70 ± 2.04
GA + FI-VarNet
0.680 ± 0.019
23.26 ± 1.26
0.636 ± 0.024
21.71 ± 1.22
0.724 ± 0.031
26.42 ± 1.77
0.663 ± 0.037
25.09 ± 1.93
Table 2: Radial k -space results at 4× and 8× acceleration. Each method combines a sampling strategy with a reconstruction network. The simulation dataset is generated from autoPET via the Data Generation module, and fastMRI + augment uses a different implant model for cross-dataset evaluation. Best results in bold.
Figure 5: Radial k -space results shown as violin plots. Top row: SSIM; bottom row: PSNR. Left column: simulation dataset; right column: fastMRI + augment. Each method shows paired distributions for 4× (blue) and 8× (orange) acceleration. OpenMASC (red, rightmost) achieves the highest median and narrowest spread across all conditions.
Figure 6: Qualitative radial results on the simulation dataset at 4× acceleration. Rows from top to bottom: reconstructed images with region of interest (red dashed box), error maps, radial sampling masks, and action step visualizations colored by acquisition order. OpenMASC recovers sharp structural detail near the implant that baselines distort.
Method
SSIM ↑
PSNR ↑
MSE ( ×10−3 ) ↓
MAE ( ×10−2 ) ↓
w/o Active Acq. (Rand † )
0.776 ± 0.013
27.53 ± 1.53
1.879 ± 0.691
2.312 ± 0.218
w/o Active Acq. (Equi)
0.768 ± 0.013
27.09 ± 1.52
2.081 ± 0.754
2.429 ± 0.255
w/o DCR
0.710 ± 0.022
21.45 ± 1.89
7.841 ± 3.365
3.716 ± 0.416
w/o VarNet Pretrain
0.770 ± 0.013
27.11 ± 1.48
2.059 ± 0.724
2.417 ± 0.235
w/o Decoupled Training
0.776 ± 0.013
27.47 ± 1.51
1.901 ± 0.674
2.318 ± 0.219
w/o PPO-DCR Co-train
0.776 ± 0.013
27.55 ± 1.52
1.873 ± 0.687
2.303 ± 0.219
Table 3: Ablation study on Cartesian simulation dataset at 4× acceleration. Each row removes one component from the full model. All paired t -tests vs. the full model yield p<0.001 across n=720 slices. † Low-frequency biased random sampling. Best results in bold.
Department of Radiology & Biomedical Imaging, Yale School of Medicine, New Haven, CT 06510 USA · Department of Biomedical Engineering, Rensselaer Polytechnic Institute, Troy, NY 12180 USA