Adversarial Data Augmentation on Breast MRI Segmentation

Abstract

The scarcity of balanced and annotated datasets has been a recurring problem in medical image analysis. Several researchers have tried to fill this gap employing dataset synthesis with adversarial networks (GANs). Breast magnetic resonance imaging (MRI) provides complex, texture-rich medical images, with the same annotation shortage issues, for which, to the best of our knowledge, no previous work tried synthesizing data. Within this context, our work addresses the problem of synthesizing breast MRI images from corresponding annotations and evaluate the impact of this data augmentation strategy on a semantic segmentation task. We explored variations of image-to-image translation using conditional GANs, namely fitting the generator’s architecture with residual blocks and experimenting with cycle consistency approaches. We studied the impact of these changes on visual verisimilarity and how an U-Net segmentation model is affected by the usage of synthetic data. We achieved sufficiently realistic-looking breast MRI images and maintained a stable segmentation score even when completely replacing the dataset with the synthetic set. Our results were promising, especially when concerning to Pix2PixHD and Residual CycleGAN architectures.

Special Issue: Generative Models in Artificial Intelligence and Their Applications

“…This result suggests the usage of the proposed method to tackle arts algorithmically. The fifth contribution, "Adversarial Data Augmentation on Breast MRI Segmentation" [5], is an application in the area of medical image analysis. The authors considered the problem of synthesizing breast MRI images from corresponding annotations and evaluates the impact of this data augmentation strategy on a semantic segmentation task.…”