Synthesizing Accurate and Realistic T1-weighted Contrast-Enhanced MR Images using Posterior-Mean Rectified Flow
On the map: Diffusion models → Medical imaging and segmentation
Abstract (arXiv)
Contrast-enhanced (CE) T1-weighted MRI is central to neuro-oncologic diagnosis but requires gadolinium-based agents, which add cost and scan time, raise environmental concerns, and may pose risks to patients. In this work, we propose a two-stage Posterior-Mean Rectified Flow (PMRF) pipeline for synthesizing volumetric CE brain MRI from non-contrast inputs. First, a patch-based 3D U-Net predicts the voxel-wise posterior mean (minimizing MSE). Then, this initial estimate is refined by a time-conditioned 3D rectified flow to incorporate realistic textures without compromising structural fidelity. We train this model on a multi-institutional collection of paired pre- and post-contrast T1w volumes (BraTS 2023-2025). On a held-out test set of 360 diverse volumes, our best refined outputs achieve an axial FID of 12.46 and KID of 0.007 (~ 68.7% lower FID than the posterior mean) while maintaining low volumetric MSE of 0.057 (~ 27% higher than the posterior mean). Qualitative comparisons confirm that our method restores lesion margins and vascular details realistically, effectively navigating the perception-distortion trade-off for clinical deployment.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 6 of its 16 references that are in the map. First posted on arXiv 18 Aug 2025.
Builds on
6 of 16 references.
References · 6
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2018Demystifying MMD GANscited by 1,268
- 2017The Perception-Distortion Tradeoffcited by 582
- 2021MedPerf: Open Benchmarking Platform for Medical Artificial Intelligence using Federated Evaluationcited by 88
- 2024Posterior-Mean Rectified Flow: Towards Minimum MSE Photo-Realistic Image Restorationcited by 32
- 2019Introducing Hann windows for reducing edge-effects in patch-based image segmentationcited by 16
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025An Efficient 3D Latent Diffusion Model for T1-contrast Enhanced MRI Generation
- 2026MAMA-FLUX.2: Image-to-Image Synthesis of Post-Contrast Breast DCE-MRI for the MAMA-SYNTH Challenge
- 2025Flow Matching for Conditional MRI-CT and CBCT-CT Image Synthesis
- 2026VoxStruct3D: Structure-Leading Flow Matching for Voxel-Space 3D MRI Synthesis
- 2026MPFlow: Multi-modal Posterior-Guided Flow Matching for Zero-Shot MRI Reconstruction
- 2024Conditional Generative Models for Contrast-Enhanced Synthesis of T1w and T1 Maps in Brain MRI
- 2026DRIFT: Difficulty-aware Rectified Flows for Through-plane MRI Super-Resolution
- 2025MAISI-v2: Accelerated 3D High-Resolution Medical Image Synthesis with Rectified Flow and Region-specific Contrastive Loss