Options
Trans-EffNet: A Hybrid Model for Brain Tumor Detection Using EfficientNet and Transformer Encoder
Journal
2024 IEEE Smart World Congress (SWC)
Date Issued
2024-12-02
Author(s)
Abdul Haseeb Nizamani
Zhigang Chen
Ahsan Ahmed Nizamani
Mughair Aslam Bhatti
Hauwei Ma
DOI
10.1109/SWC62898.2024.00260
Abstract
Accurate classification of brain tumors from MRI is critical for effective diagnosis and treatment. In this study, we introduce Trans-EffNet, a hybrid model combining pre-trained EfficientNet architectures with a transformer encoder to enhance brain tumor classification accuracy. By leveraging EfficientNet's deep CNN capabilities for localized feature extraction and the transformer encoder for capturing global contextual relationships, our model improves the identification of intricate tumor characteristics. Fine-tuned with ImageNet-derived weights and utilizing extensive data augmentation, Trans-EffNet was validated on both multi-class and binary datasets. Trans-EffNetB1 achieved 99.49 % accuracy on the multi-class dataset, while Trans-EffNetB2 recorded 99.83 % accuracy on the binary dataset, with perfect precision, recall, and F1-Score. These results underscore Trans-EffNet's robustness and potential as a significant advancement in brain tumor detection and classification.
File(s)
No Thumbnail Available
Name
Waiting for Repository Version.pdf
Size
37.66 KB
Format
Adobe PDF
Checksum
(MD5):70439f9ac5a8bde2f366653765cefe3c