Boosting the Prediction of Brain Tumor Using Two Stage BiGait Architecture

Abstract

Deep learning (DL) techniques hold immense promise for revolutionizing medical diagnostics, including brain tumor detection. Detecting malignancies in the brain is fraught with challenges that carry critical implications for accurate and timely diagnosis. Early detection is paramount, and our paper presents a novel approach to address this issue. We introduce the BiGait ensemble transfer learning (TL) architecture tailored for brain tumor classification tasks. Our proposed methodology is a two-stage process that harnesses the power of TL. In the initial phase, we employ two established models, ResNet50 and EfficientNetB1, and fine-tune them using the Kaggle multiclass dataset. This includes the incorporation of supplementary feature layers. We also leverage a stack ensemble solution to combine these feature layers, enhancing the model's overall predictive capabilities. Our approach is designed to shift the input distribution positively, effectively mitigating overfitting and reducing hardware overhead. To validate the effectiveness of our method, we conducted extensive experiments and evaluation. The dataset utilized in this work, publicly available on Kaggle contains 3,264 brain MRIs categorized into four groups: normal (500 images), meningioma (937 images), glioma (926 images), and pituitary tumor (901 images). The experimental results demonstrate that our model surpasses traditional single-model approaches, achieving an impressive accuracy rate of 96% and exhibiting superior generalization capabilities. This advancement in brain tumor classification holds great promise for early diagnosis and improved patient outcomes.

Topics & Concepts