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 条
  • [1] Efficient Deep Learning Fusion-Based Approach for Brain Tumor Diagnosis
    Gairola, Ajay Krishan
    Kumar, Vidit
    Singh, Gagan Deep
    Bajaj, Mohit
    Rathore, Rajkumar Singh
    Mahmoud, Mohamed H.
    El-Shafai, Walid
    TRAITEMENT DU SIGNAL, 2024, 41 (05) : 2573 - 2584
  • [2] A deep fusion-based vision transformer for breast cancer classification
    Fiaz, Ahsan
    Raza, Basit
    Faheem, Muhammad
    Raza, Aadil
    HEALTHCARE TECHNOLOGY LETTERS, 2024, 11 (06) : 471 - 484
  • [3] A deep fusion-based vision transformer for breast cancer classification
    Fiaz, Ahsan
    Raza, Basit
    Faheem, Muhammad
    Raza, Aadil
    HEALTHCARE TECHNOLOGY LETTERS, 2024,
  • [4] Deep learning and spark architecture based intelligent brain tumor MRI image severity classification
    Abirami, S.
    Venkatesan, G. K. D. Prasanna
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 76
  • [5] SwinGALE: fusion of swin transformer and attention mechanism for GAN-augmented liver tumor classification with enhanced deep learning
    Sumash Chandra Bandaru
    G. Bharathi Mohan
    R. Prasanna Kumar
    Ali Altalbe
    International Journal of Information Technology, 2024, 16 (8) : 5351 - 5369
  • [6] Attention Multisource Fusion-Based Deep Few-Shot Learning for Hyperspectral Image Classification
    Liang, Xuejian
    Zhang, Ye
    Zhang, Junping
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 8773 - 8788
  • [7] Enhancing brain tumor MRI classification with an ensemble of deep learning models and transformer integration
    Benzorgat, Nawal
    Xia, Kewen
    Benzorgat, Mustapha Noure Eddine
    PEERJ COMPUTER SCIENCE, 2024, 10 : 1 - 27
  • [8] Multi-objective Archimedes Optimization Algorithm with Fusion-based Deep Learning model for brain tumor diagnosis and classification
    B. Devanathan
    M. Kamarasan
    Multimedia Tools and Applications, 2023, 82 : 16985 - 17007
  • [9] Deep Learning Feature Fusion-Based Retina Image Classification
    Zhang Tianfu
    Zhong Shuncong
    Lian Chaoming
    Zhou Ning
    Xie Maosong
    LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (24)
  • [10] Fusion-Based Deep Learning Model for Hyperspectral Images Classification
    Kriti
    Haq, Mohd Anul
    Garg, Urvashi
    Khan, Mohd Abdul Rahim
    Rajinikanth, V
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 72 (01): : 939 - 957