Development and Validation of an Artificial Intelligence Prediction Model and a Survival Risk Stratification for Lung Metastasis in Colorectal Cancer From Highly Imbalanced Data

Background:

To assist clinicians with diagnosis and optimal treatment decision-making, we attempted to develop and validate an artificial intelligence prediction model for lung metastasis (LM) in colorectal cancer (CRC) patients.

Method:

The clinicopathological characteristics of 46037 CRC patients from the Surveillance, Epidemiology, and End Results (SEER) database and 2779 CRC patients from a multi-center external validation set were collected retrospectively. After feature selection by univariate and multivariate analyses, six machine learning (ML) models, including logistic regression, K-nearest neighbor, support vector machine, decision tree, random forest, and balanced random forest (BRF), were developed and validated for the LM prediction. The optimization model with best performance was compared to the clinical predictor. In addition, stratified LM patients by risk score were utilized for survival analysis.