返回
EfficientNet and multi-path convolution with multi-head attention network for brain tumor grade classification
DOI:10.1016/j.compeleceng.2023.108700.png)
摘要
En 中文
Grade classification is a challenging task in brain tumor image classification. Contemporary models employ transfer learning technique to attain better performance. The existing models ig-nored the semantic features of a tumor during classification decisions. Moreover, contemporary research requires an optimized model to exhibit better performance on larger datasets. Thus, we propose an EfficientNet and multi-path convolution with a multi-head attention network for the grade classification. We used a pre-trained EfficientNetB4 in the feature extraction phase. Then, a multi-path convolution with multi-head attention network performs a feature enhancement task. Finally, features obtained from the above step are classified using a fully connected double dense network. We utilize TCIA repository datasets to generate a three-class (normal/low-grade/high-grade) classification dataset. Our model achieves 98.35% accuracy and 97.32% Jaccard coefficient. The proposed model achieves superior performance than its competing models in all key metrics. Further, we achieve similar performance on a noisy dataset.
Keyword:
Brain tumor grade classification
Three-class classification
Multi-path convolution network
Multi-head attention
Transfer learning
期刊
C
IF:
4.9
论文数:
6.7K
被引数:
1.3W
机构
引用论文
Trp-dependent auxin biosynthesis in Arabidopsis: involvement of cytochrome P450s CYP79B2 and CYP79B3
A Novel Approach to Improving Brain Image Classification Using Mutual Information-Accelerated Singular Value Decomposition
IEEE ACCESS
IF3.6
Brain tumor segmentation and grading of lower-grade glioma using deep learning in MRI images基于深度学习的MRI图像低级别胶质瘤脑肿瘤分割与分级
Comparison of Feature Selection Methods and Machine Learning Classifiers for Radiomics Analysis in Glioma Grading用于胶质瘤分级的放射组学分析的特征选择方法和机器学习分类器的比较
IEEE ACCESS
IF3.6
Brain tumor classification for MR images using transfer learning and fine-tuning使用迁移学习和微调对MR图像进行脑肿瘤分类

