Attention transformer mechanism and fusion-based deep learning architecture for MRI brain tumor classification system

被引:23
|
作者
Tabatabaei, Sadafossadat [1 ]
Rezaee, Khosro [2 ]
Zhu, Min [3 ]
机构
[1] Isfahan Univ Technol, Dept Elect & Comp Engn, Esfahan, Iran
[2] Meybod Univ, Dept Biomed Engn, Meybod, Iran
[3] Zhejiang Shuren Univ, Coll Informat Sci & Technol, Hangzhou, Peoples R China
关键词
Brain tumors; MR images; Transformer module; Bi-directional feature fusion; Convolutional neural network; CONVOLUTIONAL NEURAL-NETWORK; IMAGES; MACHINE;
D O I
10.1016/j.bspc.2023.105119
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Most primary brain malignancies are malignant tumors characterized by masses of abnormal tissue that grow uncontrollably. Recently, deep transfer learning (DTL) has been considered in the automated clinical application of magnetic resonance imaging (MRI) for determining brain tumor characteristics. Several previous studies have shown that tumors analyzed in MR images usually have local features, are extensive, and are associated with a high level of uncertainty. Accordingly, we introduce a two-branch parallel model that integrates the Transformer Module (TM) with the Self-Attention Unit (SAU) and Convolutional Neural Networks (CNN) to classify brain tumors in MR images. We also propose a novel approach that combines local and global features derived from CNNs and TMs to improve classification accuracy through a cross-fusion strategy. Hybrid architecture combined with cross-fusion allows parallel systems to be merged between branches, resulting in a pattern that identifies various types of tumors. Additionally, we developed a lightweight and improved CNN architecture (iResNet) that distinguishes tumor features based on MR images. From the 3064 slices in the four-class MR images from the Figshare dataset, 20 % were considered unseen images. The remainder was divided into 60 %, 20 %, and 20 % slices for training, validation, and testing. The overall structure was combined with other similar CNN networks (i.e., DenseNet, VGG, and ResNet), and we found that the accuracy of iVGG, iDensNet, and iResNet is 98.59, 98.94, and 99.30 %, respectively. Using local and global features, we developed an accurate and generalizable model to detect brain tumors in MRI, allowing rapid and accurate diagnosis.
引用
收藏
页数:13
相关论文
共 50 条
  • [21] Classification of Brain MRI Tumor Images Based on Deep Learning PGGAN Augmentation
    Gab Allah, Ahmed M.
    Sarhan, Amany M.
    Elshennawy, Nada M.
    DIAGNOSTICS, 2021, 11 (12)
  • [22] A Deep Learning Architecture for Brain Tumor Segmentation in MRI Images
    Shreyas, V.
    Pankajakshan, Vinod
    2017 IEEE 19TH INTERNATIONAL WORKSHOP ON MULTIMEDIA SIGNAL PROCESSING (MMSP), 2017,
  • [23] A Symmetrical Approach to Brain Tumor Segmentation in MRI Using Deep Learning and Threefold Attention Mechanism
    Rahman, Ziaur
    Zhang, Ruihong
    Bhutto, Jameel Ahmed
    SYMMETRY-BASEL, 2023, 15 (10):
  • [24] Hybrid Window Attention Based Transformer Architecture for Brain Tumor Segmentation
    Peiris, Himashi
    Hayat, Munawar
    Chen, Zhaolin
    Egan, Gary
    Harandi, Mehrtash
    BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2022, PT II, 2023, 14092 : 173 - 182
  • [25] ConvBiFuseNet: a parallel fusion model with routing attention for MRI brain tumor classification
    Liu, Shiguo
    Wei, Dejian
    Zhang, Junzhong
    Ji, Xurui
    Cao, Hui
    JOURNAL OF SUPERCOMPUTING, 2025, 81 (01):
  • [26] Robust brain tumor classification by fusion of deep learning and channel-wise attention mode approach
    Balamurugan, A. G.
    Srinivasan, Saravanan
    Preethi, D.
    Monica, P.
    Mathivanan, Sandeep Kumar
    Shah, Mohd Asif
    BMC MEDICAL IMAGING, 2024, 24 (01):
  • [27] A Fusion-Based Technique With Hybrid Swarm Algorithm and Deep Learning for Biosignal Classification
    Prabhakar, Sunil Kumar
    Rajaguru, Harikumar
    Kim, Chulho
    Won, Dong-Ok
    FRONTIERS IN HUMAN NEUROSCIENCE, 2022, 16
  • [28] A Deep Transfer Learning Based Architecture for Brain Tumor Classification Using MR Images
    Badjie, Bakary
    Ulker, Ezgi Deniz
    INFORMATION TECHNOLOGY AND CONTROL, 2022, 51 (02): : 332 - 344
  • [29] A deep learning approach for brain tumor classification using MRI images*
    Aamir, Muhammad
    Rahman, Ziaur
    Dayo, Zaheer Ahmed
    Abro, Waheed Ahmed
    Uddin, M. Irfan
    Khan, Inayat
    Imran, Ali Shariq
    Ali, Zafar
    Ishfaq, Muhammad
    Guan, Yurong
    Hu, Zhihua
    COMPUTERS & ELECTRICAL ENGINEERING, 2022, 101
  • [30] BrainNet: Optimal Deep Learning Feature Fusion for Brain Tumor Classification
    Zahid, Usman
    Ashraf, Imran
    Khan, Muhammad Attique
    Alhaisoni, Majed
    Yahya, Khawaja M. M.
    Hussein, Hany S. S.
    Alshazly, Hammam
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022