Abstract
Established prognostic tests based on limited numbers of transcripts can identify high-risk breast cancer patients yet are approved only for individuals presenting with specific clinical features or disease characteristics. Deep learning algorithms could hold potential for stratifying patient cohorts based on full transcriptome data, yet the development of robust classifiers is hampered by the number of variables in omics datasets typically far exceeding the number of patients. To overcome this hurdle, we propose a classifier based on a data augmentation pipeline consisting of a Wasserstein generative adversarial network (GAN) with gradient penalty and an embedded auxiliary classifier to obtain a trained GAN discriminator (T-GAN-D). Applied to 1244 patients of the METABRIC breast cancer cohort, this classifier outperformed established breast cancer biomarkers in separating low- from high-risk patients (disease specific death, progression or relapse within 10 years from initial diagnosis). Importantly, the T-GAN-D also performed across independent, merged transcriptome datasets (METABRIC and TCGA-BRCA cohorts), and merging data improved overall patient stratification. In conclusion, GAN-based data augmentation therefore allowed generating a robust classifier capable of stratifying low- vs high-risk patients based on full transcriptome data and across independent and heterogeneous breast cancer cohorts.
Competing Interest Statement
The authors have declared no competing interest.
Funding Statement
MR and CG receive funding by the Deutsche Forschungsgemeinschaft (DFG, German Research Foundation) under Germany's Excellence Strategy - EXC 2075 - 390740016 and acknowledge the support by the Stuttgart Center for Simulation Science (SimTech).
Author Declarations
I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Yes
The details of the IRB/oversight body that provided approval or exemption for the research described are given below:
The study used only openly available human data that were originally available through cBioportal.org, the R package MetaGxBreast (Gendoo et al. 2019, DOI: 10.1038/s41598-019-45165-4) and the following publications: Xia et al. 2019 (DOI: 10.1038/s41467-019-13588-2); Rueda et al. 2019 (DOI: 10.1038/s41586-019-1007-8); Liu et al. 2018 (DOI: 10.1016/j.cell.2018.02.052).
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Yes
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Yes
I have followed all appropriate research reporting guidelines and uploaded the relevant EQUATOR Network research reporting checklist(s) and other pertinent material as supplementary files, if applicable.
Yes
Data Availability
Transcriptome data (median Z-scores), OS, DSS and associated clinical records of the METABRIC cohort were downloaded from cbioportal.org. Locoregional and distant recurrence records and ROR-P scores of the MB cohort were retrieved from Rueda et al. 2019 (DOI: 10.1038/s41586-019-1007-8) and Xia et al. 2019 (DOI: 10.1038/s41467-019-13588-2) respectively. Clinical records, OS, DSS and PFI of the TCGA-BRCA cohort were integrated from cBioportal.org and Liu et al. 2018 (DOI: 10.1016/j.cell.2018.02.052). Curated METABRIC and BRCA-TCGA expression data were downloaded using the MetaGxBreast R package (Gendoo et al. 2019, DOI: 10.1038/s41598-019-45165-4). The input files and scripts used to generate the results are available in the Zenodo repository (DOI: 10.5281/zenodo.7151831).