3D Deep Learning for Brain Tumor Segmentation and Survival Prediction: A Comprehensive Multi-Modal Analysis Using the...

Three-dimensional deep learning offers promise for automated accurate brain tumor segmentation and survival prediction but requires robust validation across multiple MRI modalities to be effectively implemented in clinical practice.

This study presents a comprehensive 3D deep learning framework using 369 cases from the BraTS2020 dataset. A 3D U-Net architecture was developed for tumor segmentation utilizing combined imaging data and optimized for computational efficiency and memory. The final 3D U-Net model segmentations were used to build machine learning 6-month and 12-month survival classifiers. Segmentation models were evaluated using multiple metrics, including the Dice Similarity Coefficient, Hausdorff Distance, and Cohen's d. The classification models were evaluated using AUC-ROC and balanced accuracy.

Segmentation achieved a modest, but promising, performance across 30 epochs and with 295 training patients, achieving the best mean validation Dice = 0.8388 and a final-epoch mean Dice of 0.8263. Survival classification with a hybrid clinical and imaging logistic regression showed promising results, with 12-month prediction achieving AUC = 0.746 and 69% accuracy. The top contributing features for the 12-month prediction classifier were extent of resection, T1 contrast-enhanced tumor median, and FLAIR tumor median.

This comprehensive framework demonstrates that a multi-modal approach provides meaningful performance gains, while segmentation-derived features show a promising ability to enable survival prediction.