GAN-based one dimensional medical data augmentation

Ye Zhang, Zhixiang Wang, Zhen Zhang, Junzhuo Liu, Ying Feng, Leonard Wee, Andre Dekker, Qiaosong Chen*, Alberto Traverso*

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

Abstract

With the continuous development of human life and society, the medical field is constantly improving. However, modern medicine still faces many limitations, including challenging and previously unsolvable problems. In these cases, artificial intelligence (AI) can provide solutions. The research and application of generative adversarial networks (GAN) are a clear example. While most researchers focus on image augmentation, there are few one-dimensional data augmentation examples. The radiomics feature extracted from RT and CT images is one-dimensional data. As far as we know, we are the first to apply the WGAN-GP algorithm to generate radiomics data in the medical field. In this paper, we input a portion of the original real data samples into the model. The model learns the distribution of the input data samples and generates synthetic data samples with similar distribution to the original real data, which can solve the problem of obtaining annotated medical data samples. We have conducted experiments on the public dataset Heart Disease Cleveland and the private dataset. Compared with the traditional method of Synthetic Minority Oversampling Technique (SMOTE) and common GAN for data augmentation, our method has significantly improved the AUC and SEN values under different data proportions. At the same time, our method has also shown varying levels of improvement in ACC and SPE values. This demonstrates that our method is effective and feasible.
Original languageEnglish
Pages (from-to)10481-10491
Number of pages11
JournalSoft Computing
Volume27
Issue number15
Early online date1 May 2023
DOIs
Publication statusPublished - 1 Aug 2023

Keywords

  • Generative adversarial networks
  • SMOTE
  • Medical data augmentation
  • Deep learning
  • Artificial intelligence

Cite this