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.

Machine learning-based ensemble approach in prediction of lung cancer predisposition using XRCC1 gene polymorphism

LUADpp: an Effective Prediction Model on Prognosis of Lung Adenocarcinomas Based on Somatic Mutational Features

Machine Learning Study of SNPs in Noncoding Regions to Predict Non-small Cell Lung Cancer Susceptibility

Prediction of Smoking Behavior From Single Nucleotide Polymorphisms With Machine Learning Approaches

Machine learning computational model to predict lung cancer using electronic medical records

Integrative prediction model for radiation pneumonitis incorporating genetic and clinical-pathological factors using machine learning

Implementation of ensemble machine learning algorithms on exome datasets for predicting early diagnosis of cancers

Genome-wide Association Study Based Risk Prediction Model in Predicting Lung Cancer Risk in Chinese

A prediction model based on DNA methylation biomarkers and radiological characteristics for identifying malignant from benign pulmonary nodules

Predicting the risk of lung cancer using machine learning: A large study based on UK Biobank

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

Prediction of Treatment Recommendations Via Ensemble Machine Learning Algorithms for Non-Small Cell Lung Cancer Patients in Personalized Medicine

Machine learning-based proteogenomic data modeling identifies circulating plasma biomarkers for early detection of lung cancer

Machine Learning and Real-World Data to Predict Lung Cancer Risk in Routine Care

A Prediction Model for Lung Cancer Diagnosis that Integrates Genomic and Clinical Features

A large cohort study identifying a novel prognosis prediction model for lung adenocarcinoma through machine learning strategies

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

Machine Learning for Prediction of Non-Small Cell Lung Cancer Based on Inflammatory and Nutritional Indicators in Adults: A Cross-Sectional Study

An Ensemble Learning Approach for Cancer Drug Prediction

Identification of Prognostic Biomarkers for Stage III Non-Small Cell Lung Carcinoma in Female Nonsmokers Using Machine Learning