Brain Tumor Segmentation Evaluation and Overall Survival Prediction on the BRATS 2020 Dataset

Authors

  • Annisa Maizano Fahlevi Universitas Kristen Maranatha
  • Riko Saragih Universitas Kristen Maranatha

DOI:

https://doi.org/10.28932/jste.v2i1.13880

Keywords:

Brain, Glioma, MRI, Segmentation

Abstract

Overall survival (OS) assessment in glioma patients is a crucial component of MRI-based medical image analysis, as OS estimation directly influences clinical decision-making and treatment planning. One of the central challenges in developing image-based predictive models lies in the dependency on accurate tumor segmentation. This study aims to construct an MRI-based OS prediction model using the Brain Tumor Segmentation 2020 (BraTS 2020) dataset by incorporating two types of masked images: ground truth masks and automatically generated predicted masks derived from a 3D U-Net segmentation model. OS classification was grouped into three categories (< 10 months, 10–15 months, and > 15 months). The predictive model achieved an accuracy of 0.9792 when using ground truth masks and 0.9583 when using predicted masks. These findings suggest that a fully automated deep-learning–based segmentation pipeline can approximate the performance of manual segmentation and holds strong potential for large-scale clinical applications where manual annotation is impractical.

References

P. Y. Wen and S. Kesari, “Malignant gliomas in adults,” New England Journal of Medicine, vol. 359, no. 5, pp. 492–507, 2008, doi: 10.1056/NEJMra0708126.

B. H. Menze et al., “BRATS 2020: Benchmarking Brain Tumor Segmentation and Survival Prediction,” Proc. Int. MICCAI Conf., 2020. Available: https://zenodo.org/records/3718904

F. Isensee et al., “nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation,” Nat. Methods, vol. 18, pp. 203–211, 2018. Available: https://arxiv.org/abs/1809.10486

S. van Griethuysen et al., “Computational Radiomics System to Decode the Radiographic Phenotype,” Cancer Res., vol. 77, no. 21, pp. 104–113, Nov. 2017. Available: https://pubmed.ncbi.nlm.nih.gov/29092951/

S. Bakas et al., “Identifying the best machine learning algorithms for brain tumor segmentation, progression assessment, and overall survival prediction in the BraTS challenge,” arXiv:1811.02629, 2018. Available: https://arxiv.org/abs/1811.02629

A. Shaheen et al., “Overall Survival Prediction of Glioma Patients With Multiregional Radiomics,” Front. Neurosci., vol. 16, p. 911065, Jul. 2022, doi: 10.3389/fnins.2022.911065. Available: https://pubmed.ncbi.nlm.nih.gov/35873825/

B. Çiçek, A. Abdulkadir, S. S. Lienkamp, T. Brox, and O. Ronneberger, “3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation,” in Proc. MICCAI, 2016, pp. 424–432. Available: https://arxiv.org/abs/1606.06650

D. Nguyen, “3D U-Net on BRATS 2020 Model V2,” Kaggle Notebook, 2020. [Online]. Available: https://kaggle.com/code/duynguyen3213213123/3d-u-net-on-brats-2020-model-v2

L. Breiman, “Random Forests,” Mach. Learn., vol. 45, pp. 5–32, 2001, doi: http://dx.doi.org/10.1023/A:1010933404324

P. Soleimani and N. Farezi, “Utilizing deep learning via the 3D U-net neural network for the delineation of brain stroke lesions in MRI image,”

Scientific Reports, vol. 13, no. 19808, 2023, doi: 10.1038/s41598-023-47107-7. Available: https://www.nature.com/articles/s41598-023-47107-7

Published

2026-03-31

Issue

Section

Articles