Constructing a Prognostic Model for Subtypes of Colorectal Cancer Based on Machine Learning and Immune Infiltration-Related Genes

Abstract

This study constructed a prognostic model combining machine learning-based immune infiltration-related genes in each CRC subtype. We used publicly accessible gene expression data and clinical information on colorectal cancer patients. Integrated bioinformatics analysis was used for the identification of immune-wise genes. Machine learning algorithms, like LASSO regression and random forest, were utilised to identify the most important genes that may serve as predictors for patient prognosis. Univariate Cox regression, consensus clustering as well as machine learning algorithms were conducted to construct a prognostic risk scoring model. Analysis of functional enrichment, immune infiltration analyses and copy number variations as well as mutational burdens was performed and validated at the single-cell level. A machine learning-based model is designed with good predictive power—an area under the receiver operating characteristic curve (AUC-ROC) of C-index in cross-validation. The model also achieved good calibration and discrimination ability to stratify patients into high- and low-risk groups with a statistically significant difference in OS (p < 0.05). We have integrated multiple types of gene network features into machine learning systems based on the characteristics of integrating networks with Multi-Expense Learning algorithms, and we propose a robust approach for predicting CRC molecular subtype patient survival. This model could potentially steer personalised treatment strategies and ameliorate outcomes in patients. Although validation in other cohorts and clinical situations is necessary, it may be useful.