用少量数据训练模型,把乳腺癌MRI的未增强图转为增强图。
MAMA-FLUX.2: Image-to-Image Synthesis of Post-Contrast Breast DCE-MRI for the MAMA-SYNTH Challenge

- 基于FLUX.2框架,用空间条件控制生成过程。
- 在肿瘤区域提升准确率,整体图像质量达到竞赛最优。
- 适合医学影像生成、放射科医生及算法开发者参考。
动态对比增强乳腺MRI是癌症诊断与监测的核心手段,但需使用钆基对比剂。本文针对MAMA-SYNTH挑战,解决从术前到术后乳腺MRI的图像合成问题。提出MAMA-FLUX.2,一种基于FLUX.2-Klein-4B的条件潜变量流匹配方法。以术前图像作为空间条件,模型预测目标术后潜变量对应的流场。为高效适配预训练模型,采用LoRA微调,并引入区域训练目标:全局流匹配、肿瘤区域监督与稳定前景正则化。进一步研究了LoRA秩、强度窗宽及区域损失权重对轴向切片的影响,优先关注临床相关的肿瘤指标。消融实验表明,适度的肿瘤区域与稳定前景加权可优化图像保真度与肿瘤区域准确性之间的平衡。最终模型在LoRA秩/α=64/64、MHA_max=25、λ_tumor=0.25、λ_stable=0.1时表现最佳。结果表明,通过参数高效微调与任务感知区域损失,小型预训练修正流变换器可有效用于对比增强MRI合成。
原文摘要 · Abstract (English)
Dynamic contrast-enhanced breast MRI is central to cancer diagnosis and monitoring, but requires gadolinium-based contrast agents. In this work, we address pre-to-post contrast breast MRI synthesis for the MAMA-SYNTH challenge. We propose MAMA-FLUX.2, a conditional latent flow-matching approach based on FLUX.2-Klein-4B. The pre-contrast image is encoded as spatial conditioning, while the model predicts the flow field associated with the post-contrast target latent. To adapt the pretrained model efficiently, we use LoRA fine-tuning and introduce a regional training objective combining global flow matching, tumor-region supervision, and stable foreground regularization. We further investigate LoRA rank, intensity windowing, and regional loss weights on axial slices, prioritizing clinically relevant tumor-focused metrics. Our ablation study shows that moderate tumor and stable-foreground weighting improves the trade-off between image fidelity and tumor-region accuracy. The final model achieves the best overall balance with LoRA rank/$α=64/64$, $\mathrm{MHA}_{\max}=25$, $λ_{\mathrm{tumor}}=0.25$, and $λ_{\mathrm{stable}}=0.1$. These results demonstrate that compact pretrained rectified-flow transformers can be adapted for contrast-enhanced MRI synthesis using parameter-efficient fine-tuning and task-aware regional losses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。