The integration of Artificial Intelligence (AI) into healthcare represents a transformative sift towards smart medical systems. This paper presents an innovative framework for automated brain tumor classification by integrating multi-head attention mechanisms into pretrained Convolutional Neural Networks (CNNs) to enhance accuracy. We methodically evaluate five state-of-the-art CNN architectures e.g. VGG16, VGG19, MobileNet, Xception, and InceptionV3 enhanced with attention layers to classify MRI brain images into four categories glioma, meningioma, pituitary, and normal. To mitigate dataset imbalance, we apply RandomOverSampler and extensive data augmentation methods. Experimental findings reveal that the attention-enhanced MobileNet model attains superior performance, attaining a classification accuracy of 97%, surpassing conventional transfer learning approaches. To further interpret the model’s decisions, Explainable Artificial Intelligence (XAI) technique is employed to highlight key areas in the MRI images that influenced the classification outcomes, thereby enhancing model transparency. Overall, this work demonstrates performance improvement over other existing methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Brain Tumor Classification in MRI Images Using Multi-head Attention Enhanced Convolutional Neural Networks for Smart Healthcare Systems

  • Abhijit Dey,
  • Md. Abdul Alim Sheikh,
  • Sayok Sanyal,
  • Touhid Anowar Hossain,
  • Kuntal Mitra

摘要

The integration of Artificial Intelligence (AI) into healthcare represents a transformative sift towards smart medical systems. This paper presents an innovative framework for automated brain tumor classification by integrating multi-head attention mechanisms into pretrained Convolutional Neural Networks (CNNs) to enhance accuracy. We methodically evaluate five state-of-the-art CNN architectures e.g. VGG16, VGG19, MobileNet, Xception, and InceptionV3 enhanced with attention layers to classify MRI brain images into four categories glioma, meningioma, pituitary, and normal. To mitigate dataset imbalance, we apply RandomOverSampler and extensive data augmentation methods. Experimental findings reveal that the attention-enhanced MobileNet model attains superior performance, attaining a classification accuracy of 97%, surpassing conventional transfer learning approaches. To further interpret the model’s decisions, Explainable Artificial Intelligence (XAI) technique is employed to highlight key areas in the MRI images that influenced the classification outcomes, thereby enhancing model transparency. Overall, this work demonstrates performance improvement over other existing methods.