Predicting local control of brain metastases after stereotactic radiotherapy with clinical, radiomics and deep learning features.

Abstract

Background and purpose: Timely identification of local failure after stereotactic radiotherapy for brain metastases allows for treatment modifications, potentially improving outcomes. While previous studies showed that adding radiomics or Deep Learning (DL) features to clinical features increased Local Control (LC) prediction accuracy, their combined potential to predict LC remains unexplored. We examined whether a model using a combination of radiomics, DL and clinical features achieves better accuracy than models using only a subset of these features.

Materials and methods: We collected pre-treatment brain MRIs (TR/TE: 25/1.86 ms, FOV: 210 × 210 × 150, flip angle: 30°, transverse slice orientation, voxel size: 0.82 × 0.82 × 1.5 mm) and clinical data for 129 patients at the Gamma Knife Center of the Elisabeth-TweeSteden Hospital. Radiomics features were extracted using the Python radiomics feature extractor and DL features were obtained using a 3D ResNet model. A Random Forest machine learning algorithm was employed to train four models using: (1) clinical features only; (2) clinical and radiomics features; (3) clinical and DL features; and (4) clinical, radiomics, and DL features. The average accuracy and other metrics were derived using K-fold cross validation.

Results: The prediction model utilizing only clinical variables provided an Area Under the receiver operating characteristic Curve (AUC) of 0.85 and an accuracy of 75.0%. Adding radiomics features increased the AUC to 0.86 and accuracy to 79.33%, while adding DL features resulted in an AUC of 0.82 and accuracy of 78.0%. The best performance came from combining clinical, radiomics, and DL features, achieving an AUC of 0.88 and accuracy of 81.66%. This model's prediction improvement was statistically significant compared to models trained with clinical features alone or with the combination of clinical and DL features. However, the improvement was not statistically significant when compared to the model trained with clinical and radiomics features.

Conclusion: Integrating radiomics and DL features with clinical characteristics improves prediction of local control after stereotactic radiotherapy for brain metastases. Models incorporating radiomics features consistently outperformed those utilizing clinical features alone or clinical and DL features. The increased prediction accuracy of our integrated model demonstrates the potential for early outcome prediction, enabling timely treatment modifications to improve patient management.