# Conditional Infilling GANs for Data Augmentation in Mammogram Classification

arXiv: https://arxiv.org/abs/1807.08093 · HTML: https://mlsanity.com/paper/1807.08093/

## Key facts

- Cited by 19 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 21 Jul 2018.

## Built on by (most cited first, 19 of 19)

- [Recent advances and clinical applications of deep learning in medical image analysis](https://mlsanity.com/paper/2105.13381/) (2021)
- [GAN-Supervised Dense Visual Alignment](https://mlsanity.com/paper/2112.05143/) (2021)
- [Ensembling with Deep Generative Views](https://mlsanity.com/paper/2104.14551/) (2021)
- [Data Augmentation in High Dimensional Low Sample Size Setting Using a Geometry-Based Variational Autoencoder](https://mlsanity.com/paper/2105.00026/) (2021)
- [GAN-Based Data Augmentation and Anonymization for Skin-Lesion Analysis: A Critical Review](https://mlsanity.com/paper/2104.10603/) (2021)
- [A Survey on Training Challenges in Generative Adversarial Networks for Biomedical Image Analysis](https://mlsanity.com/paper/2201.07646/) (2022)
- [CHORUS: Learning Canonicalized 3D Human-Object Spatial Relations from Unbounded Synthesized Images](https://mlsanity.com/paper/2308.12288/) (2023)
- [Free-form tumor synthesis in computed tomography images via richer generative adversarial network](https://mlsanity.com/paper/2104.09701/) (2021)
- [Data synthesis and adversarial networks: A review and meta-analysis in cancer imaging](https://mlsanity.com/paper/2107.09543/) (2021)
- [Analyzing Tumors by Synthesis](https://mlsanity.com/paper/2409.06035/) (2024)
- [Distilling and Transferring Knowledge via cGAN-generated Samples for Image Classification and Regression](https://mlsanity.com/paper/2104.03164/) (2021)
- [METGAN: Generative Tumour Inpainting and Modality Synthesis in Light Sheet Microscopy](https://mlsanity.com/paper/2104.10993/) (2021)
- [Trustworthy image-to-image translation: evaluating uncertainty calibration in unpaired training scenarios](https://mlsanity.com/paper/2501.17570/) (2025)
- [Overcoming limited battery data challenges: A coupled neural network approach](https://mlsanity.com/paper/2111.15348/) (2021)
- [CCS-GAN: COVID-19 CT-scan classification with very few positive training images](https://mlsanity.com/paper/2110.01605/) (2021)
- [Ebola Optimization Search Algorithm (EOSA): A new metaheuristic algorithm based on the propagation model of Ebola virus disease](https://mlsanity.com/paper/2106.01416/) (2021)
- [Computer Vision-based Characterization of Large-scale Jet Flames using a Synthetic Infrared Image Generation Approach](https://mlsanity.com/paper/2206.02110/) (2022)
- [The State of Applying Artificial Intelligence to Tissue Imaging for Cancer Research and Early Detection](https://mlsanity.com/paper/2306.16989/) (2023)
- [Generation of synthetic data using breast cancer dataset and classification with resnet18](https://mlsanity.com/paper/2405.16286/) (2024)

## Abstract (arXiv)

Deep learning approaches to breast cancer detection in mammograms have recently shown promising results. However, such models are constrained by the limited size of publicly available mammography datasets, in large part due to privacy concerns and the high cost of generating expert annotations. Limited dataset size is further exacerbated by substantial class imbalance since "normal" images dramatically outnumber those with findings. Given the rapid progress of generative models in synthesizing realistic images, and the known effectiveness of simple data augmentation techniques (e.g. horizontal flipping), we ask if it is possible to synthetically augment mammogram datasets using generative adversarial networks (GANs). We train a class-conditional GAN to perform contextual in-filling, which we then use to synthesize lesions onto healthy screening mammograms. First, we show that GANs are capable of generating high-resolution synthetic mammogram patches. Next, we experimentally evaluate using the augmented dataset to improve breast cancer classification performance. We observe that a ResNet-50 classifier trained with GAN-augmented training data produces a higher AUROC compared to the same model trained only on traditionally augmented data, demonstrating the potential of our approach.