关键词: 磁共振图像, 脑肿瘤分类, EfficientNetV2网络, 注意力机制

Abstract: Aiming at the problems of overfitting and low classification accuracy in brain tumor magnetic resonance image classification, we proposed a brain tumor classification method based on an improved EfficientNetV2 network. The method  introduced the coordinate attention mechanism in the EfficientNetV2 network, which simultaneously obtained the feature information of brain tumor from both vertical and horizontal directions and accurately identified the lesion features of brain tumor. It helped the model to locate and identify the lesion area information more comprehensively and accurately, and effectively suppressed the influence of background information on the detection results, so that the model had higher classification accuracy. The problem of low classification accuracy caused by  insufficient acquisition of feature information was solved. In order to further improve the classification accuracy, the Hard-Swish activation function was introduced, which could not only improve the computational speed of the brain tumor classification network model, but also effectively improve the classification accuracy. Meanwhile, the improved model was equipped with Dropout layer and normalization layer, which could better suppress the occurrence of overfitting, accelerate the convergence speed of the model, improve the robustness of the model, and significantly improve the classification accuracy. The experimental results show that the improved model obtains classification accuracy of 98.4% in the validation set, and the effectiveness of the improved model in brain tumor classification task is verified by comparison experiments and ablation experiments.

Key words: magnetic resonance image, brain tumor classification, EfficientNetV2 network, attention mechanism