cs.CVAug 26, 2026

MAMA-FLUX.2: Image-to-Image Synthesis of Post-Contrast Breast DCE-MRI for the MAMA-SYNTH Challenge

Authors: Kamil Kwarciak, Marek Wodzinski

Organizations: Department of Measurement and Electronics, AGH University of Krakow, Krakow, Poland · Sano Centre for Computational Medicine, Krakow, Poland

Abstract

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α=64/64, MHAmax⁡=25\mathrm{MHA}_{\max}=25, λtumor=0.25λ_{\mathrm{tumor}}=0.25, and λstable=0.1λ_{\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.

Figures & tables

Explore similar work

CardsList
  1. Pre- to Post-Contrast Synthesis of Breast DCE-MRI using Latent Bridge Matching

    Aug 7, 2026Sina Amirrajab, Zohaib Salahuddin, Henry C Woodruff +1Magnetic Resonance ImagingMedical Image Synthesis

  2. Lesion-Gated Hybrid Synthesis for Virtual Contrast-Enhanced Breast MRI: A MAMA-SYNTH Challenge Solution

    Sep 18, 2026Shohei YoshimotoMedical Image SynthesisImage Enhancement

  3. Dense Temporal Contrast Synthesis via Conditioned Latent Transport

    Jul 31, 2026Smriti Joshi, Apostolia Tsirikoglou, Daniel M. Lang +15Magnetic Resonance ImagingMedical Image Synthesis