Synthetic Lung Ultrasound Data Generation Using Autoencoder With Generative Adversarial Network.

Journal: IEEE transactions on ultrasonics, ferroelectrics, and frequency control
PMID:

Abstract

Class imbalance is a significant challenge in medical image analysis, particularly in lung ultrasound (LUS), where severe patterns are often underrepresented. Traditional oversampling techniques, which simply duplicate original data, have limited effectiveness in addressing this issue. To overcome these limitations, this study introduces a novel supervised autoencoder generative adversarial network (SA-GAN) for data augmentation, leveraging advanced generative artificial intelligence (AI) to create high-quality synthetic samples for minority classes. In addition, the traditional data augmentation technique is used for comparison. The SA-GAN incorporates an autoencoder to develop a conditional latent space, effectively addressing weight clipping issues and ensuring higher quality synthetic data. The generated samples are evaluated using similarity metrics and expert analysis to validate their utility. Furthermore, state-of-the-art neural networks are used for multiclass classification, and their performance is compared when trained with GAN-based augmentation versus traditional data augmentation techniques. These contributions enhance the robustness and reliability of AI models in mitigating class imbalance in LUS analysis.

Authors

  • Noreen Fatima
    Department of Information Engineering and Computer Science, University of Trento, Trento, Italy.
  • Federico Mento
  • Sajjad Afrakhteh
    Department of Information Engineering and Computer Science, University of Trento, Trento, Italy.
  • Tiziano Perrone
    Department of Internal Medicine, IRCCS San Matteo, 27100, Pavia, Italy.
  • Andrea Smargiassi
  • Riccardo Inchingolo
  • Libertario Demi