Generative Adversarial Networks to Improve Fetal Brain Fine-Grained Plane Classification

Sensors (Basel). 2021 Nov 29;21(23):7975. doi: 10.3390/s21237975.

Abstract

Generative adversarial networks (GANs) have been recently applied to medical imaging on different modalities (MRI, CT, X-ray, etc). However there are not many applications on ultrasound modality as a data augmentation technique applied to downstream classification tasks. This study aims to explore and evaluate the generation of synthetic ultrasound fetal brain images via GANs and apply them to improve fetal brain ultrasound plane classification. State of the art GANs stylegan2-ada were applied to fetal brain image generation and GAN-based data augmentation classifiers were compared with baseline classifiers. Our experimental results show that using data generated by both GANs and classical augmentation strategies allows for increasing the accuracy and area under the curve score.

Keywords: deep learning; generative adversarial networks; ultrasound image classification.

MeSH terms

  • Brain / diagnostic imaging
  • Image Processing, Computer-Assisted*
  • Magnetic Resonance Imaging
  • Neural Networks, Computer*
  • Ultrasonography