Explainable machine learning for predicting aspiration pneumonia after radiotherapy in oral cavity cancer
Yu-Fan Wu1, Jhen-Bin Lin2, Yi-Shing Leu3
1Department of Information Management, Fu-Jen Catholic University, New Taipei City, Taiwan.
Background And Purpose:
Aspiration pneumonia is a severe late event that occurs following radiotherapy for oral cavity cancer (OCC). This study aimed to develop explainable machine learning models to predict aspiration pneumonia.
Materials And Methods:
This study included 880 patients with OCC (derivation cohort, 561 patients; external validation cohort, 319 patients) who underwent surgery and adjuvant radiotherapy between 2010 and 2020. Random Forest (RF), eXtreme Gradient Boosting (XGBoost), and Categorical Boosting (CatBoost) models were trained to predict aspiration pneumonia using clinical and dosimetric features. Model performance was evaluated using the area under the curve (AUC). The SHapley Additive exPlanations (SHAP) method was applied to rank feature importance and explain the model.
Results:
The RF model achieved the highest AUC compared with the XGBoost and CatBoost models in the external validation set (0.966, 0.962, and 0.955, respectively). An explainable RF model identified the mean doses to the superior pharyngeal constrictor muscle (PCM), middle PCM, and supraglottic larynx as the three most important features for predicting aspiration pneumonia in the training and validation sets. The dose-toxicity relationship between the mean dose to the swallowing structures and aspiration pneumonia was nonlinear. The threshold mean dose to the swallowing structures for lowering the risk of aspiration pneumonia was determined. The SHAP force plot provided an individualized interpretation of the model predictions for each patient.
Conclusion:
An explainable machine learning model can assist clinicians in predicting aspiration pneumonia after radiotherapy for OCC. Individualized predictions may help tailor radiotherapy to reduce the risk of aspiration pneumonia.
