Maintenance Notice

Due to necessary scheduled maintenance, the JMIR Publications website will be unavailable from Wednesday, July 01, 2020 at 8:00 PM to 10:00 PM EST. We apologize in advance for any inconvenience this may cause you.

Who will be affected?

Accepted for/Published in: JMIR Medical Informatics

Date Submitted: Feb 10, 2025
Open Peer Review Period: Feb 10, 2025 - Feb 19, 2025
Date Accepted: Nov 18, 2025
(closed for review but you can still tweet)

The final, peer-reviewed published version of this preprint can be found here:

A Machine Learning Approach to Predicting Mortality Risk in Chemotherapy-Treated Lung Cancer: Machine Learning Model Development and Validation

Zou J, Huang J, Lu K, Lin A, Xie C, Zhang J, Rao B, Li Z, Xie D, Lu L, Luo F, Yang L, Qiu F, Zhang X, Deng Y, Lu J

A Machine Learning Approach to Predicting Mortality Risk in Chemotherapy-Treated Lung Cancer: Machine Learning Model Development and Validation

JMIR Med Inform 2025;13:e72424

DOI: 10.2196/72424

PMID: 41411039

PMCID: 12757710

A Machine Learning Approach to Predicting Mortality Risk in Chemotherapy-Treated Lung Cancer: Machine Learning Model Development and Validation

  • Jianjun Zou; 
  • Jinyi Huang; 
  • Katie Lu; 
  • Ao Lin; 
  • Chen Xie; 
  • Jinrong Zhang; 
  • Boqi Rao; 
  • Zhi Li; 
  • Dongming Xie; 
  • Ling Lu; 
  • Feng Luo; 
  • Lei Yang; 
  • Fuman Qiu; 
  • Xin Zhang; 
  • Yibin Deng; 
  • Jiachun Lu

ABSTRACT

Background:

Backgroung: Accurately predicting the survival outcomes of lung cancer patients receiving chemotherapy remains challenging.

Objective:

Objective:

Aiming to improve patient management, this study constructed a multivariate model based on machine learning methods to assess the all-cause mortality risk in these patients.

Methods:

Methods:

This study retrospectively recruited 1278 post-chemotherapy lung cancer patients from Guangzhou Chest Hospital between 2007 and 2019. Candidate features such as demographic characteristics, environmental exposures, clinical information, and patient-reported symptoms were collected by questionnaires and the electronic medical record system. The survival status and deceased date were investigated twice a year. On the training set, a total of 84 predicted models were built with five machine-learning algorithms individually or binately. On the testing set, concordance indexes (C-indexes) were calculated to select the optimal model, and the model performance was internally validated through receiver operating characteristic curve (ROC), calibration curves, and decision curve analysis (DCA). Furthermore, Shapley additive explanation (SHAP) and Restricted cubic spline (RCS) were utilized to feature attribution analysis.

Results:

Results:

The optimal model ultimately retained 21 prognosis-association features, including age, gender, BMI, smoking status, environmental smoke, MDASI-LC total score trajectories, CD56, TNM stage, histology, and pre-chemotherapy blood biomarkers (CYFRA21-1, CA12-5, CA19-9, WBC, D-dimer, CEA, HGB, CRP, NSE, PLT). On the testing set, the model acquired a C-index of 0.702 (95% CI: 0.652-0.753). The decision curves demonstrated positive clinical benefit when the risk thresholds were 0.40-0.69, 0.62-0.99, and 0.72-0.99 for 1-, 3-, and 5-year mortality predictions. The calibration curves showed that the predicted mortality probabilities fluctuated around the observed probabilities, and the Brier scores for 1-, 3-, and 5-year predictions were 0.20, 0.18, and 0.11 for 1-, 3-, and 5-year mortality predictions. The AUC of the model was 0.740, 0.777, and 0.915 for 1-, 3-, and 5-year mortality predictions. Interpretability feature attribution analysis revealed that the significant features could predict all-cause mortality risk in chemotherapy-treated lung cancer patients.

Conclusions:

Conclusions:

Our machine learning models exhibited excellent discrimination, calibration, and clinical benefit in predicting the mortality risk of chemotherapy-treated lung cancer patients, which could help clinicians in personalized prognostic management. Clinical Trial: None


 Citation

Please cite as:

Zou J, Huang J, Lu K, Lin A, Xie C, Zhang J, Rao B, Li Z, Xie D, Lu L, Luo F, Yang L, Qiu F, Zhang X, Deng Y, Lu J

A Machine Learning Approach to Predicting Mortality Risk in Chemotherapy-Treated Lung Cancer: Machine Learning Model Development and Validation

JMIR Med Inform 2025;13:e72424

DOI: 10.2196/72424

PMID: 41411039

PMCID: 12757710

Download PDF


Request queued. Please wait while the file is being generated. It may take some time.

© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.