Machine Learning to Build and Validate a Model for Radiation Pneumonitis Prediction in Patients with Non–Small Cell Lung Cancer

Abstract

Purpose: Radiation pneumonitis is an important adverse event in patients with non–small cell lung cancer (NSCLC) receiving thoracic radiotherapy. However, the risk of radiation pneumonitis grade ≥ 2 (RP2) has not been well predicted. This study hypothesized that inflammatory cytokines or the dynamic changes during radiotherapy can improve predictive accuracy for RP2.

Experimental Design: Levels of 30 inflammatory cytokines and clinical information in patients with stages I–III NSCLC treated with radiotherapy were from our prospective studies. Statistical analysis was used to select predictive cytokine candidates and clinical covariates for adjustment. Machine learning algorithm was used to develop the generalized linear model for predicting risk RP2.

Results: A total of 131 patients were eligible and 17 (13.0%) developed RP2. IL8 and CCL2 had significantly (Bonferroni) lower expression levels in patients with RP2 than without RP2. But none of the changes in cytokine levels during radiotherapy was significantly associated with RP2. The final predictive GLM model for RP2 was established, including IL8 and CCL2 at baseline level and two clinical variables. Nomogram was constructed based on the GLM model. The model's predicting ability was validated in the completely independent test set (AUC = 0.863, accuracy = 80.0%, sensitivity = 100%, specificity = 76.5%).

Conclusions: By machine learning, this study has developed and validated a comprehensive model integrating inflammatory cytokines with clinical variables to predict RP2 before radiotherapy that provides an opportunity to guide clinicians.