Back To Projects

Evaluating the Efficacy of the 3D U-Net Architecture For Glioblastoma Multiforme Tumor Segmentation

This research evaluates the performance of the 3D U-Net model for automated glioblastoma tumor segmentation from MRIs, achieving 98.6% accuracy and significantly faster processing times than human oncologists, crucial for effective radiation therapy.


Glioblastoma is the deadliest form of brain cancer which begins as a congregation of cancerous cells within the brain but then progresses into invading and destroying healthy brain tissue [9]. Radiation therapy, the most popular treatment option, is where neuro-oncologists apply intense radiation energy beams directly on the tumor region to kill the cancerous cells. However, for radiation therapy to be effective, the segmented magnetic resonance image (MRI) in which the oncologists base where to apply the radiation must be segmented nearly 100% accurately, or else the energy beams will mistakenly damage healthy brain tissue. Additionally, radiation therapy will be futile if the MRI segmentation is not complete in time for the therapy appointment. Our research problem is evaluating the performance of notable segmentation models for accurate and immediate 3D segmentation of glioblastoma multiforme brain tumors from MRIs while focusing on one particular architecture, the 3D U-Net. Each of the models was given 850 MRIs from the BraTS 2021 dataset, which is an annual competition hosted by the University of Pennsylvania [8]. Our 3D U-Net model achieved a testing accuracy of 0.986) accuracy and took approximately 55 seconds to predict segmentations of the tumor region. The results reveal that the 3D U-Net model is capable of automating glioblastoma tumor segmentation in significantly fewer hours than a human oncologist would take, all while maintaining similar or higher accuracy, where even a minor difference can mean the distinction between life and death.

Explore More!

Published Paper
Arnav J.
Erick Siavichay
Computer Science Alum from Stanford, Software Engineering and Data Science at Tagg

Related Projects

Brain Tumor Classification

Early classification and diagnosis of Brain Tumors are essential for providing the right treatment to a patient. It is crucial to get treatment as soon as possible because the survival rate for someone with an untreated brain tumor can range from as low as 3 months to as high as 5 years. In this project, we classified brain tumor images into 4 categories: glioma, meningioma, pituitary, and no tumor. With the use of baseline and deep learning models, the deep learning models demonstrated a significantly higher performance due to their ability to analyze images. The model with the highest accuracy was the MobileNet, a pre-trained transfer learning model trained on 5,608 images. This model yielded a validation accuracy of 98.24%. Using metrics including Kappa cohen score, precision, and recall, we validated the machine learning model's performance. We deployed the MobileNet model to a web app using Streamlit, where users submit MRI images and receive diagnoses of tumor class. We found that the model performed very well while utilizing the web app, indicating that it is safe to be used. However, since we only have 4 classes and there are over 150 total types of brain tumors, it could easily get a diagnosis wrong if it is not in one of these 4 classes.
Rohan S.
Mentored by Sriram Hathwar
Melanoma detection through AI models

This project explores the use of AI models to detect skin cancer from images of skin lesions, achieving 80-90% accuracy in predicting malignancy and revealing key physical attributes that could aid early detection.
Aditya J.
Mentored by Alaisha Alexander
Optimizing Predictive Models for Healthcare: A Comparison of Personal and General Datasets in Coronary Artery Disease Risk Assessment

This research investigates how the choice of training population, personalized versus more general datasets, impacts the performance and generalizability of machine learning models for predicting Coronary Artery Disease risk. By comparing models trained on datasets of varying specificity using data from the UC Irvine Machine Learning Repository, the study explores trade offs between accuracy and broad applicability to improve strategies for early CAD risk assessment and prevention.
Isaac G.
Mentored by Raphi Kang