Abstract:Non-small cell lung cancer (NSCLC) represents the foremost cause of cancer-related mortality, underscoring the need for precise prognostic tools to enhance clinical decision-making. Existing models often fall short in providing reliable survival predictions. This study aims to develop a robust machine learning ensemble model to refine survival probability estimations for NSCLC patients and support clinical decisions. Following the application of inclusion and exclusion criteria, we utilized the SEER cohort to identify 93,828 NSCLC patients diagnosed between 2004 and 2019. The dataset was partitioned into training (70%) and testing (30%) subsets. Spearman correlation and univariate Cox regression analyses were conducted to identify significant variables influencing overall survival. A soft voting ensemble model was constructed, integrating MLP, XGBoost, and Random Forest algorithms. Model performance was assessed using Area Under the Curve (AUC) and decision curve analysis (DCA) in both the training and internal validation datasets. An external validation cohort comprising 325 NSCLC patients from our hospital (2015-2023) was used to evaluate generalizability. Eleven critical variables were identified: sex, age, histologic type, tumor grade, tumor size, N stage, M stage, and metastases (bone, lung, brain, liver). The soft voting ensemble model yielded AUCs of 0.819, 0.818, and 0.819 for predicting 1-year, 3-year, and 5-year survival in the internal test set, respectively, demonstrating substantial predictive accuracy. External validation produced AUCs of 0.809, 0.793, and 0.800. DCA affirmed the model’s clinical relevance. SHapley Additive exPlanations (SHAP) analysis highlighted tumor size, N stage, and M stage as the most influential factors. The soft voting ensemble model offers a highly accurate prognostic tool for NSCLC survival prediction, significantly enhancing clinical prognosis. SHAP feature importance analysis further supports the development of personalized treatment strategies. This study was supported by the National Natural Science Foundation of China (82172670, 81972858, 82202951 and 81773245), the Technology Innovation and Application Development Project of Chongqing (2023DBXM002 and CSTB2022TIAD-KPX0176 and CSTB2022NSCQ-MSX1356 and 2022yjgA06) and the Cultivation Program for Clinical Research Talents of Army Medical University (2018XLC1010 and 2019XQN10).

Optimizing Skin Cancer Survival Prediction with Ensemble Techniques

Multimodal Survival Ensemble Network: Integrating Genomic and Histopathological Insights for Enhanced Cancer Prognosis.

Evaluation of Machine Learning Algorithms for the Prognosis of Breast Cancer from the Surveillance, Epidemiology, and End Results Database

Enhancing skin lesion classification with advanced deep learning ensemble models: a path towards accurate medical diagnostics

Multiclass Skin Cancer Classification Using Ensemble of Fine-Tuned Deep Learning Models

Skin cancer detection using ensemble of machine learning and deep learning techniques

Identification and Validation of Three Risk Models in Skin Melanoma Based on Bioinformatics Analysis

Combining State-of-the-Art Pre-Trained Deep Learning Models: A Noble Approach for Skin Cancer Detection Using Max Voting Ensemble

Survival Prediction in Non-Small Cell Lung Cancer Patients: Performance Evaluation of Machine Learning Ensemble Models (Preprint)

A multimodal ensemble approach for clear cell renal cell carcinoma treatment outcome prediction

An ensemble of machine learning and anti-learning methods for predicting tumour patient survival rates

An Ensemble Approach for Patient Prognosis of Head and Neck Tumor Using Multimodal Data

Enhancing Breast Cancer Survival Prognosis Through Omic and Non-Omic Data Integration

Ensemble Classification Model With CFS-IGWO-Based Feature Selection for Cancer Detection Using Microarray Data

A Deep-Ensemble-Learning-Based Approach for Skin Cancer Diagnosis

A Novel Approach for Predicting the Survival of Colorectal Cancer Patients Using Machine Learning Techniques and Advanced Parameter Optimization Methods

Survival analysis in breast cancer: evaluating ensemble learning techniques for prediction

Current status of artificial intelligence methods for skin cancer survival analysis: a scoping review

Predicting Breast Cancer Relapse from Histopathological Images with Ensemble Machine Learning Models

An Ensembled Framework for Human Breast Cancer Survivability Prediction Using Deep Learning

Deep-CR MTLR: a Multi-Modal Approach for Cancer Survival Prediction with Competing Risks