Comparative Performance of Autoencoders and Traditional Machine Learning Algorithms in Clinical Data Analysis for Predicting Post-Staged GKRS Tumor Dynamics

Abstract

Introduction: Accurate prediction of tumor dynamics following Gamma Knife radiosurgery (GKRS) is critical for optimizing treatment strategies for patients with brain metastases (BMs). Traditional machine learning (ML) algorithms have been widely used for this purpose; however, recent advancements in deep learning, such as autoencoders, offer the potential to enhance predictive accuracy. This study aims to evaluate the efficacy of autoencoders compared to traditional ML models in predicting tumor progression or regression after GKRS. Objectives: The primary objective of this study is to assess whether integrating autoencoder-derived features into traditional ML models can improve their performance in predicting tumor dynamics three months post-GKRS in patients with brain metastases. Methods: This retrospective analysis utilized clinical data from 77 patients treated at the “Prof. Dr. Nicolae Oblu” Emergency Clinic Hospital-Iasi. Twelve variables, including socio-demographic, clinical, treatment, and radiosurgery-related factors, were considered. Tumor progression or regression within three months post-GKRS was the primary outcome, with 71 cases of regression and 6 cases of progression. Traditional ML models, such as Logistic Regression, Support Vector Machine (SVM), K-Nearest Neighbors (KNN), Extra Trees, Random Forest, and XGBoost, were trained and evaluated. The study further explored the impact of incorporating features derived from autoencoders, particularly focusing on the effect of compression in the bottleneck layer on model performance. Results: Traditional ML models achieved accuracy rates ranging from 0.91 (KNN) to 1.00 (Extra Trees). Integrating autoencoder-derived features generally enhanced model performance. Logistic Regression saw an accuracy increase from 0.91 to 0.94, and SVM improved from 0.85 to 0.96. XGBoost maintained consistent performance with an accuracy of 0.94 and an AUC of 0.98, regardless of the feature set used. These results demonstrate that hybrid models combining deep learning and traditional ML techniques can improve predictive accuracy. Conclusion: The study highlights the potential of hybrid models incorporating autoencoder-derived features to enhance the predictive accuracy and robustness of traditional ML models in forecasting tumor dynamics post-GKRS. These advancements could significantly contribute to personalized medicine, enabling more precise and individualized treatment planning based on refined predictive insights, ultimately improving patient outcomes.