论文
MAMA-FLUX.2:用于 MAMA-SYNTH 挑战的对比后乳腺 DCE-MRI 的图像到图像合成
MAMA-FLUX.2: Image-to-Image Synthesis of Post-Contrast Breast DCE-MRI for the MAMA-SYNTH Challenge
摘要
动态对比增强乳腺 MRI 对于癌症诊断和监测至关重要,但需要基于钆的对比剂。在这项工作中,我们解决了 MAMA-SYNTH 挑战的前后对比乳腺 MRI 合成问题。我们提出了 MAMA-FLUX.2,一种基于 FLUX.2-Klein-4B 的条件潜在流匹配方法。对比前图像被编码为空间条件,而模型预测与对比后目标潜在相关的流场。为了有效地适应预训练模型,我们使用 LoRA 微调 并引入了结合全局流匹配、肿瘤区域监督和稳定前景正则化的区域训练目标。我们进一步研究 LoRA 等级、强度窗口和轴向切片上的区域损失权重,优先考虑临床相关的以肿瘤为中心的指标。我们的消融研究表明,适度的肿瘤和稳定的前景加权可以改善图像保真度和肿瘤区域准确性之间的权衡。最终模型达到了最佳的整体平衡,LoRA 等级/$α=64/64$,$\mathrm{MHA}_{\max}=25$,$λ_{\mathrm{tumor}}=0.25$,$λ_{\mathrm{stable}}=0.1$。这些结果表明,紧凑的预训练整流流 Transformer 可以适用于使用参数高效的 微调 和任务感知区域损失来进行对比增强 MRI 合成。