Radiomics-Clinical AI Model With Probability Weighted Strategy for Prognosis Prediction in Non-small Cell Lung Cancer

In this study, we propose a radiomics-clinical probability weighted model for the prediction of prognosis for NSCLC. The model combines radiomics features extracted from RT planning images with clinical factors such as age, gender, histology, and tumor stage. CT images with radiotherapy structures of 422 NSCLC patients were retrieved from The Cancer Imaging Archive (TCIA). Radiomic features were extracted from gross tumor volume (GTV). Five machine learning algorithms, namely decision trees (DT), random forests (RF), extreme boost (EB), support vector machine (SVM) and generalized linear model (GLM), were optimized by a voted ensemble machine learning (VEML) model. A probabilistic weighted approach is used to incorporate the uncertainty associated with both radiomic and clinical features and to generate a probabilistic risk score for each patient. The performance of the model is evaluated using a receiver operating characteristic (ROC). Radiomic model, clinical factors model and combined radiomic-clinical probability weighted model demonstrated good performance in predicting NSCLC survival with AUC of 0.941, 0.856 and 0.949 respectively. The combined radiomics-clinical probability weighted enhanced model achieved significantly better performance than radiomic model in 1-year survival prediction (chi-square test, p<0.05). The proposed model has the potential to improve NSCLC prognosis and facilitate personalized treatment decisions.