Identifying Primary Brain Tumors and Lung Cancer Brain Metastases by Training XGBoost Models Based on Radiomics Features from Brain MRI Data
摘要
This study aimed to enhance the differential diagnosis of primary malignant brain tumors and lung cancer brain metastases using a radiomics approach. Leveraging MRI-derived imaging features and machine learning, this study sought to improve diagnostic accuracy and robustness while addressing the limitations of subjective radiological assessments.
MethodsRadiomics features were extracted from magnetic resonance imaging (MRI) scans of patients with primary malignant brain tumors and lung cancer brain metastases. An extreme gradient boosting (XGBoost) classification model was trained and evaluated using metrics such as precision, recall, F1-score, receiver operating characteristic (ROC) curve, area under the curve (AUC), and confusion matrix. The robustness of the model was validated through 5-fold cross-validation.
ResultsThe XGBoost model demonstrated a robust performance in classifying primary brain tumors and lung cancer brain metastases, with ROC-AUC values ranging from 0.93 to 0.98. Confusion matrices revealed true positive rates of 84.62–97.00% and true negative rates of 82.71–95.65%, with low fold-specific variability. Cross-validation metrics showed median accuracy, precision, recall, and F1-Scores above 0.90, with narrow interquartile ranges.
ConclusionThe XGBoost model exhibited high accuracy, precision, and discriminative ability in classifying primary malignant brain tumors and lung cancer brain metastases, with consistent performance across cross-validation folds.