Multi-institutional dose-segmented dosiomic analysis for predicting radiation pneumonitis after lung stereotactic body radiation therapy.

ABSTRACT

PURPOSE: To predict radiation pneumonitis (RP) grade 2 or worse after lung stereotactic body radiation therapy (SBRT) using dose-based radiomic (dosiomic) features. METHODS: This multi-institutional study included 247 early-stage nonsmall cell lung cancer patients who underwent SBRT with a prescribed dose of 48-70 Gy at an isocenter between June 2009 and March 2016. Ten dose-volume indices (DVIs) were used, including the mean lung dose, internal target volume size, and percentage of entire lung excluding the internal target volume receiving greater than x Gy (x = 5, 10, 15, 20, 25, 30, 35, and 40). A total of 6,808 dose-segmented dosiomic features, such as shape, first order, and texture features, were extracted from the dose distribution. Patients were randomly partitioned into two groups model training (70%) and test datasets (30%) over 100 times. Dosiomic features were converted to z-scores (standardized values) with a mean of zero and a standard deviation (SD) of one to put different variables on the same scale. The feature dimension was reduced using the following methods: interfeature correlation based on Spearman's correlation coefficients and feature importance based on a light gradient boosting machine (LightGBM) feature selection function. Three different models were developed using LightGBM as follows (a) a model with ten DVIs (DVI model), (b) a model with the selected dosiomic features (dosiomic model), and (c) a model with ten DVIs and selected dosiomic features (hybrid model). Suitable hyperparameters were determined by searching the largest average area under the curve (AUC) value in the receiver operating characteristic curve (ROC-AUC) via stratified fivefold cross-validation. Each of the final three models with the closest the ROC-AUC value to the average ROC-AUC value was applied to the test datasets. The classification performance was evaluated by calculating the ROC-AUC, AUC in the precision-recall curve (PR-AUC), accuracy, precision, recall, and f1-score. The entire process was repeated 100 times with randomization, and 100 individual models were developed for each of the three models. Then the mean value and SD for the 100 random iterations were calculated for each performance metric. RESULTS: Thirty-seven (15.0%) patients developed RP after SBRT. The ROC-AUC and PR-AUC values in the DVI, dosiomic, and hybrid models were 0.660 ± 0.054 and 0.272 ± 0.052, 0.837 ± 0.054 and 0.510 ± 0.115, and 0.846 ± 0.049 and 0.531 ± 0.116, respectively. For each performance metric, the dosiomic and hybrid models outperformed the DVI models (P < 0.05). Texture-based dosiomic feature was confirmed as an effective indicator for predicting RP. CONCLUSIONS: Our dose-segmented dosiomic approach improved the prediction of the incidence of RP after SBRT.