Healthy Counterfactual Generation via Diffusion Inpainting for Mammography Classification
Not on the map. Why?
Abstract (arXiv)
False negatives remain a critical limitation of computer-aided diagnosis (CAD) systems for breast cancer screening due to delayed detection and treatment. To address this issue, we propose a counterfactual data augmentation strategy that generates healthy mammograms by "erasing" lesions from anomalous images, thereby enriching the training distribution. We train a Denoising Diffusion Probabilistic Model on BI-RADS 1 (healthy) mammograms and use a RePaint-based sampling strategy to inpaint realistic normal tissue within annotated lesion bounding boxes. The resulting healthy counterfactuals replace annotated lesion regions with realistic healthy tissue while preserving patient-specific anatomical structure, as supported by similarity metrics between real and generated images. Image realism was further assessed by radiologists and found to be consistent with the original dataset quality. We evaluate counterfactual augmentation across four representative classifier architectures: a convolutional neural network (ConvNeXt), a vision transformer (ViT), a vision-language model pre-trained on mammogram-report pairs (Mammo-CLIP) and a multi-scale attention-based multiple-instance learning framework (FPN-MIL). Experiments conducted on the VinDr-Mammo dataset show improvements in sensitivity across all architectures, particularly at 80% fixed specificity, contributing towards more reliable CAD systems for breast cancer. Code is available at: https://github.com/ines03garcia/diffusion-based-counterfactual-generation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 1 of its 27 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
1 of 27 references.
References · 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2022What is Healthy? Generative Counterfactual Diffusion for Lesion Localization
- 2024Counterfactual MRI Data Augmentation using Conditional Denoising Diffusion Generative Models
- 2025AI Alignment in Medical Imaging: Unveiling Hidden Biases Through Counterfactual Analysis
- 2025Generating healthy counterfactuals with denoising diffusion bridge models
- 2026Generating Chest X-Ray Counterfactuals by Specialising Foundation Image Models
- 2024MaskMedPaint: Masked Medical Image Inpainting with Diffusion Models for Mitigation of Spurious Correlations
- 2023Generating Realistic Counterfactuals for Retinal Fundus and OCT Images using Diffusion Models
- 2026MedDiME: Efficient Latent Diffusion with Adaptive Masking for Medical Counterfactual Generation