Abstract:Purpose: The objective of this study was to create and validate a machine learning (ML)-based model for predicting the likelihood of lung infections following chemotherapy in patients with lung cancer. Methods: A retrospective study was conducted on a cohort of 502 lung cancer patients undergoing chemotherapy. Data on age, Body Mass Index (BMI), underlying disease, chemotherapy cycle, number of hospitalizations, and various blood test results were collected from medical records. We used the Synthetic Minority Oversampling Technique (SMOTE) to handle unbalanced data. Feature screening was performed using the Boruta algorithm and The Least Absolute Shrinkage and Selection Operator (LASSO). Subsequently, six ML algorithms, namely Logistic Regression (LR), Random Forest (RF), Gaussian Naive Bayes (GNB), Multi-layer Perceptron (MLP), Support Vector Machine (SVM), and K-Nearest Neighbors (KNN) were employed to train and develop an ML model using a 10-fold cross-validation methodology. The model's performance was evaluated through various metrics, including the area under the receiver operating characteristic curve (ROC), accuracy, sensitivity, specificity, F1 score, calibration curve, decision curves, clinical impact curve, and confusion matrix. In addition, model interpretation was performed by the Shapley Additive Explanations (SHAP) analysis to clarify the importance of each feature of the model and its decision basis. Finally, we constructed nomograms to make the predictive model results more readable. Results: The integration of Boruta and LASSO methodologies identified Gender, Smoke, Drink, Chemotherapy cycles, pleural effusion (PE), Neutrophil-lymphocyte count ratio (NLR), Neutrophil-monocyte count ratio (NMR), Lymphocytes (LYM) and Neutrophil (NEUT) as significant predictors. The LR model demonstrated superior performance compared to alternative ML algorithms, achieving an accuracy of 81.80%, a sensitivity of 81.1%, a specificity of 82.5%, an F1 score of 81.6%, and an AUC of 0.888(95%CI(0.863-0.911)). Furthermore, the SHAP method identified Chemotherapy cycles and Smoke as the primary decision factors influencing the ML model's predictions. Finally, this study successfully constructed interactive nomograms and dynamic nomograms. Conclusion: The ML algorithm, combining demographic and clinical factors, accurately predicted post-chemotherapy lung infections in cancer patients. The LR model performed well, potentially improving early detection and treatment in clinical practice.

Anemia Risk Prediction Model for Osteosarcoma Patients Post‐Chemotherapy Using Artificial Intelligence

Identification of a Programmed Cell Death-related Signature for Prediction of Osteosarcoma

Machine Learning-Based Prediction of Lymph Node Metastasis Among Osteosarcoma Patients

A Delta-Radiomics Nomogram Model for Preoperative Prediction of Chemotherapy Response in High-Grade Osteosarcoma

Clinical-radiomics models based on plain X-rays for prediction of lung metastasis in patients with osteosarcoma

A clinical prediction model for lung metastasis risk in osteosarcoma: A multicenter retrospective study

An External-Validated Prediction Model to Predict Lung Metastasis among Osteosarcoma: A Multicenter Analysis Based on Machine Learning

The Construction and Development of a Clinical Prediction Model to Assess Lymph Node Metastases in Osteosarcoma

Machine learning-based individualized survival prediction model for prognosis in osteosarcoma: Data from the SEER database

Interpretable clinical visualization model for prediction of prognosis in osteosarcoma: a large cohort data study

Establishment of a Survival Risk Prediction Model for Adolescent and Adult Osteosarcoma Patients

A deep belief network-based clinical decision system for patients with osteosarcoma

A Binary Survivability Prediction Classification Model towards Understanding of Osteosarcoma Prognosis

A Novel Tool to Predict the Overall Survival of High-Grade Osteosarcoma Patients after Neoadjuvant Chemotherapy: A Large Population-Based Cohort Study

Dynamic Predictive Models With Visualized Machine Learning for Assessing Chondrosarcoma Overall Survival

A Machine Learning-Based Predictive Model for Predicting Lymph Node Metastasis in Patients With Ewing’s Sarcoma

Osteoporosis, fracture and survival: Application of machine learning in breast cancer prediction models

Survival Prediction in High-grade Osteosarcoma Using Radiomics of Diagnostic Computed Tomography

Incidence trends, overall survival, and metastasis prediction using multiple machine learning and deep learning techniques in pediatric and adolescent population with osteosarcoma and Ewing's sarcoma: nomogram and webpage

Construction of Molecular Subtype and Prognosis Prediction Model of Osteosarcoma Based on Aging-Related Genes

Construction of a risk prediction model for lung infection after chemotherapy in lung cancer patients based on the machine learning algorithm