Abstract:Accurate prediction of cancer stage is important in that it enables more appropriate treatment for patients with cancer. Many measures or methods have been proposed for more accurate prediction of cancer stage, but recently, machine learning, especially deep learning-based methods have been receiving increasing attention, mostly owing to their good prediction accuracy in many applications. Machine learning methods can be applied to high throughput DNA mutation or RNA expression data to predict cancer stage. However, because the number of genes or markers generally exceeds 10,000, a considerable number of data samples is required to guarantee high prediction accuracy. To solve this problem of a small number of clinical samples, we used a Generative Adversarial Networks (GANs) to augment the samples. Because GANs are not effective with whole genes, we first selected significant genes using DNA mutation data and random forest feature ranking. Next, RNA expression data for selected genes were expanded using GANs. We compared the classification accuracies using original dataset and expanded datasets generated by proposed and existing methods, using random forest, Deep Neural Networks (DNNs), and 1-Dimensional Convolutional Neural Networks (1DCNN). When using the 1DCNN, the F1 score of GAN5 (a 5-fold increase in data) was improved by 39% in relation to the original data. Moreover, the results using only 30% of the data were better than those using all of the data. Our attempt is the first to use GAN for augmentation using numeric data for both DNA and RNA. The augmented datasets obtained using the proposed method demonstrated significantly increased classification accuracy for most cases. By using GAN and 1DCNN in the prediction of cancer stage, we confirmed that good results can be obtained even with small amounts of samples, and it is expected that a great deal of the cost and time required to obtain clinical samples will be reduced. The proposed sample augmentation method could also be applied for other purposes, such as prognostic prediction or cancer classification.

phylaGAN: data augmentation through conditional GANs and autoencoders for improving disease prediction accuracy using microbiome data

DeepMicroGen: a generative adversarial network-based method for longitudinal microbiome data imputation

MB-GAN: Microbiome Simulation via Generative Adversarial Network

Increasing prediction accuracy of pathogenic staging by sample augmentation with a GAN

Ensemble Data Augmentation for Imbalanced Fault Diagnosis.

Generative Adversarial Networks for Data Augmentation

How Good Are Synthetic Medical Images? An Empirical Study with Lung Ultrasound

A Semi-Supervised Generative Adversarial Network for Prediction of Genetic Disease Outcomes

DeepPhylo: Phylogeny‐Aware Microbial Embeddings Enhanced Predictive Accuracy in Human Microbiome Data Analysis

Gut metagenome-derived image augmentation and deep learning improve prediction accuracy of metabolic disease classification

Cross-Modality Synthetic Data Augmentation using GANs: Enhancing Brain MRI and Chest X-ray Classification

GAN-based generative modelling for dermatological applications -- comparative study

Conditional Infilling GANs for Data Augmentation in Mammogram Classification

GAN Augmentation: Augmenting Training Data using Generative Adversarial Networks

GAN Inversion for Data Augmentation to Improve Colonoscopy Lesion Classification

Enhanced Conditional GAN for High-Quality Synthetic Tabular Data Generation in Mobile-Based Cardiovascular Healthcare

Multimodal deep learning applied to classify healthy and disease states of human microbiome

A Comparative Analysis of the Novel Conditional Deep Convolutional Neural Network Model, Using Conditional Deep Convolutional Generative Adversarial Network-Generated Synthetic and Augmented Brain Tumor Datasets for Image Classification

A Critical Assessment of Generative Models for Synthetic Data Augmentation on Limited Pneumonia X-ray Data

Enhancing Alzheimer's disease classification through split federated learning and GANs for imbalanced datasets

MedGAN: An adaptive GAN approach for medical image generation