Squeezed-Eff-Net: Edge-Computed Boost of Tomography Based Brain Tumor Classification leveraging Hybrid Neural Network Architecture
Summary
This paper introduces Squeezed-Eff-Net, a novel hybrid deep learning model combining SqueezeNet v1 and EfficientNet-B0, enhanced with handcrafted radiomic features for highly accurate and computationally efficient brain tumor classification from MRI scans. The model achieved 99.08% accuracy with Test Time Augmentation (TTA) on the Nickparvar Brain Tumor MRI dataset, demonstrating near-clinical reliability for automated diagnosis while maintaining computational efficiency.
Medical Relevance
Accurate and timely diagnosis of brain tumors is crucial for effective treatment planning and patient outcomes. This research provides a highly accurate, automated, and computationally efficient method to classify brain tumors from MRI, reducing reliance on time-consuming and error-prone manual delineation, thereby improving diagnostic efficiency and consistency in clinical settings.
AI Health Application
The AI model is designed for automated MRI-based classification of brain tumors (glioma, meningioma, pituitary tumor, no tumor), aiming to serve as a clinical decision-support system to assist radiologists and clinicians in timely and accurate diagnosis, thereby improving patient care and treatment planning.
Key Points
- A hybrid deep learning architecture, Squeezed-Eff-Net, is proposed for automated MRI-based brain tumor classification, integrating SqueezeNet v1 and EfficientNet-B0.
- The model is enhanced with handcrafted radiomic descriptors, including Histogram of Oriented Gradients (HOG), Local Binary Patterns (LBP), Gabor filters, and Wavelet transforms, to capture greater texture sensitivity.
- Evaluation was conducted on the publicly available Nickparvar Brain Tumor MRI dataset, comprising 7,023 contrast-enhanced T1-weighted axial MRI slices categorized into glioma, meningioma, pituitary tumor, and no tumor.
- The model achieved a high testing accuracy of 98.93%, which improved to 99.08% with the application of Test Time Augmentation (TTA).
- It demonstrates significant computational efficiency, operating with fewer than 2.1 million parameters and less than 1.2 GFLOPs, offering a strong balance between performance and resource consumption.
- The combination of EfficientNet-B0 for hierarchical feature representation and handcrafted features for texture sensitivity contributes to the model's high diagnostic accuracy.
- The proposed model exhibits almost clinical reliability, highlighting its potential for integration into clinical decision-support systems for automated diagnosis.
Methodology
The study proposes a hybrid deep learning model, Squeezed-Eff-Net, which synergistically combines SqueezeNet v1 (for its lightweight nature) and EfficientNet-B0 (for its high performance in feature extraction). This base architecture is augmented by concatenating handcrafted radiomic descriptorsâspecifically Histogram of Oriented Gradients (HOG), Local Binary Patterns (LBP), Gabor filters, and Wavelet transformsâto the network's feature representation. The model was trained and tested on the Nickparvar Brain Tumor MRI dataset, which includes 7,023 T1-weighted axial MRI slices across four diagnostic categories: glioma, meningioma, pituitary tumor, and no tumor. Test Time Augmentation (TTA) was utilized during the evaluation phase to further enhance predictive robustness.
Key Findings
The Squeezed-Eff-Net model achieved a remarkable testing accuracy of 98.93%, which was further boosted to 99.08% with the application of Test Time Augmentation on the Nickparvar Brain Tumor MRI dataset. This performance was attained with high computational efficiency, requiring fewer than 2.1 million parameters and less than 1.2 GFLOPs. The integration of handcrafted features was found to improve sensitivity to texture variations, while the EfficientNet-B0 backbone effectively captured intricate hierarchical features, leading to a robust and generalizeable classification model.
Clinical Impact
This model offers a clinically relevant solution for automated, highly accurate, and efficient brain tumor classification from MRI scans. Its near-clinical reliability suggests its strong potential for integration into clinical decision-support systems, assisting radiologists and neurologists in rapid, consistent, and objective diagnosis. This could lead to earlier and more precise treatment planning, reducing inter-observer variability and ultimately improving patient outcomes in neurological oncology.
Limitations
The abstract indicates that the framework was trained and tested solely on the publicly available Nickparvar Brain Tumor MRI dataset. While substantial, this single-dataset evaluation limits the explicit demonstration of the model's generalization capability across diverse MRI scanners, protocols, or patient populations from different clinical environments.
Future Directions
Not explicitly mentioned in the abstract, but implicit directions would involve validation on broader, multi-institutional datasets, exploration of robustness to various MRI sequences and artifacts, and prospective clinical trials to fully assess its real-world utility and impact.
Medical Domains
Keywords
Abstract
Brain tumors are one of the most common and dangerous neurological diseases which require a timely and correct diagnosis to provide the right treatment procedures. Even with the promotion of magnetic resonance imaging (MRI), the process of tumor delineation is difficult and time-consuming, which is prone to inter-observer error. In order to overcome these limitations, this work proposes a hybrid deep learning model based on SqueezeNet v1 which is a lightweight model, and EfficientNet-B0, which is a high-performing model, and is enhanced with handcrafted radiomic descriptors, including Histogram of Oriented Gradients (HOG), Local Binary Patterns (LBP), Gabor filters and Wavelet transforms. The framework was trained and tested only on publicly available Nickparvar Brain Tumor MRI dataset, which consisted of 7,023 contrast-enhanced T1-weighted axial MRI slices which were categorized into four groups: glioma, meningioma, pituitary tumor, and no tumor. The testing accuracy of the model was 98.93% that reached a level of 99.08% with Test Time Augmentation (TTA) showing great generalization and power. The proposed hybrid network offers a compromise between computation efficiency and diagnostic accuracy compared to current deep learning structures and only has to be trained using fewer than 2.1 million parameters and less than 1.2 GFLOPs. The handcrafted feature addition allowed greater sensitivity in texture and the EfficientNet-B0 backbone represented intricate hierarchical features. The resulting model has almost clinical reliability in automated MRI-based classification of tumors highlighting its possibility of use in clinical decision-support systems.