Brain Tumor Detection Using 3D-UNet Segmentation Features and Hybrid Machine Learning Model

Abstract

Machine learning has significantly improved disease diagnosis, enhancing the efficiency and accuracy of the healthcare system. One critical area where it proves beneficial is diagnosing brain tumors, a life-threatening disease, where early and accurate predictions can save lives. This study focuses on deploying a machine learning-based approach for brain tumor detection, utilizing Magnetic Resonance Imaging (MRI) features. We train the proposed model using 3D-UNet and 2D-UNet segmentation features extracted from MRI, encompassing shape, statistics, gray level size zone matrix, gray level dependence matrix, gray level co-occurrence matrix, and gray level run length matrix values. To improve performance, we propose a hybrid model that combines the strengths of two machine learning models, K-nearest neighbor (KNN) and gradient boosting classifier (GBC), using soft voting criteria. We combine them because, in cases where KNN exhibits poor performance for certain data points, GBC demonstrates significant performance, and vice versa, where GBC shows poor results, KNN performs significantly better. With 2D-UNet segmentation features, the model achieves a 64% accuracy. By training it on 3D-UNet segmentation features, we achieve a significant accuracy of 71% which surpasses existing state-of-the-art models that utilize 3D-UNet segmentation features.

Keywords

• Brain tumor detection
• MRI features
• machine learning
• ensemble learning
• segmentation