Radiation pneumonitis prediction after stereotactic body radiation therapy based on 3D dose distribution: dosiomics and/or deep learning-based radiomics features

Abstract

Abstract Background This study was designed to establish radiation pneumonitis (RP) prediction models using dosiomics and/or deep learning-based radiomics (DLR) features based on 3D dose distribution. Methods A total of 140 patients with non-small cell lung cancer who received stereotactic body radiation therapy (SBRT) were retrospectively included in this study. These patients were randomly divided into the training (n = 112) and test (n = 28) sets. Besides, 107 dosiomics features were extracted by Pyradiomics, and 1316 DLR features were extracted by ResNet50. Feature visualization was performed based on Spearman’s correlation coefficients, and feature selection was performed based on the least absolute shrinkage and selection operator. Three different models were constructed based on random forest, including (1) a dosiomics model (a model constructed based on dosiomics features), (2) a DLR model (a model constructed based on DLR features), and (3) a hybrid model (a model constructed based on dosiomics and DLR features). Subsequently, the performance of these three models was compared with receiver operating characteristic curves. Finally, these dosiomics and DLR features were analyzed with Spearman’s correlation coefficients. Results In the training set, the area under the curve (AUC) of the dosiomics, DLR, and hybrid models was 0.9986, 0.9992, and 0.9993, respectively; the accuracy of these three models was 0.9643, 0.9464, and 0.9642, respectively. In the test set, the AUC of these three models was 0.8462, 0.8750, and 0.9000, respectively; the accuracy of these three models was 0.8214, 0.7857, and 0.8571, respectively. The hybrid model based on dosiomics and DLR features outperformed other two models. Correlation analysis between dosiomics features and DLR features showed weak correlations. The dosiomics features that correlated DLR features with the Spearman’s rho |ρ| ≥ 0.8 were all first-order features. Conclusion The hybrid features based on dosiomics and DLR features from 3D dose distribution could improve the performance of RP prediction after SBRT.

Keywords

• Radiation pneumonitis prediction
• 3D dose distribution
• Dosiomics
• Deep learning-based radiomics
• Random forest