Precision Oncology: An Ensembled Machine Learning Approach to Identify a Candidate mRNA Panel for Stratification of Patients with Breast Cancer

The rise of machine learning (ML) has recently buttressed the efforts for big data-driven precision oncology. This study used ensemble ML for precision oncology in breast cancer, which is one of the most common malignancies worldwide with marked heterogeneity of the underlying molecular mechanisms. We analyzed clinical and RNA-seq data from The Cancer Genome Atlas (TCGA) (844 patients with breast cancer and 113 healthy individuals) and the Molecular Taxonomy of Breast Cancer International Consortium (METABRIC) (1784 patients with breast cancer and 202 healthy individuals). We evaluated six algorithms in the context of ensemble modeling and identified a candidate mRNA diagnostic panel that can differentiate patients from healthy controls, and stratify breast cancer into molecular subtypes. The ensemble model included 50 mRNAs and displayed 82.55% accuracy, 79.22% specificity, and 84.55% sensitivity in stratifying patients into molecular subtypes in TCGA cohort. Its performance was markedly higher, however, in distinguishing the basal, LumB, and Her2+ breast cancer subtypes from healthy individuals. In overall survival analysis, the mRNA panel showed a hazard ratio of 2.25 (p = 5 x 10(-7)) for breast cancer and was significantly associated with molecular pathways related to carcinogenesis. In conclusion, an ensemble ML approach, including 50 mRNAs, was able to stratify patients with different breast cancer subtypes and differentiate them from healthy individuals. Future prospective studies in large samples with deep phenotyping can help advance the ensemble ML approaches in breast cancer. Advanced ML methods such as ensemble learning are timely additions to the precision oncology research toolbox.

Automated summary

This study utilized a machine learning approach to precision oncology research in breast cancer and identified a candidate mRNA panel that can diagnose and stratify breast cancer. The ensemble model showed high accuracy and sensitivity in stratifying patients and differentiating breast cancer subtypes from healthy individuals.