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 条
  • [31] Speech Emotion Classification with Parallel Architecture of Deep Learning and Multi-Head Attention Transformer
    Nguyen, An Hoang
    Trang, Kien
    Thao, Nguyen Gia Minh
    Vuong, Bao Quoc
    Ton-That, Long
    2023 62ND ANNUAL CONFERENCE OF THE SOCIETY OF INSTRUMENT AND CONTROL ENGINEERS, SICE, 2023, : 1549 - 1554
  • [32] Automated Brain Tumor Segmentation and Classification in MRI Using YOLO-Based Deep Learning
    Almufareh, Maram Fahaad
    Imran, Muhammad
    Khan, Abdullah
    Humayun, Mamoona
    Asim, Muhammad
    IEEE ACCESS, 2024, 12 : 16189 - 16207
  • [33] A Deep Learning Based Effective Model for Brain Tumor Segmentation and Classification Using MRI Images
    Gayathri, T.
    Kumar, Sundeep K.
    JOURNAL OF ADVANCES IN INFORMATION TECHNOLOGY, 2023, 14 (06) : 1280 - 1288
  • [34] A robust MRI-based brain tumor classification via a hybrid deep learning technique
    Shaimaa E. Nassar
    Ibrahim Yasser
    Hanan M. Amer
    Mohamed A. Mohamed
    The Journal of Supercomputing, 2024, 80 : 2403 - 2427
  • [35] A robust MRI-based brain tumor classification via a hybrid deep learning technique
    Nassar, Shaimaa E.
    Yasser, Ibrahim
    Amer, Hanan M.
    Mohamed, Mohamed A.
    JOURNAL OF SUPERCOMPUTING, 2024, 80 (02): : 2403 - 2427
  • [36] CNN-Based Deep Learning Technique for the Brain Tumor Identification and Classification in MRI Images
    Mandle, Anil Kumar
    Sahu, Satya Prakash
    Gupta, Govind P.
    INTERNATIONAL JOURNAL OF SOFTWARE SCIENCE AND COMPUTATIONAL INTELLIGENCE-IJSSCI, 2022, 14 (01):
  • [37] Brain Tumor Classification Based On MRI Image Processing With Alexnet Architecture
    Sy, Nidaan Khofiya
    Fu'adah, Yunendah Nur
    Pratiwi, Nor Kumalasari Caecar
    Naufal, Rizqy Ilmi
    Pratama, Aditya Deta
    2022 IEEE ASIA PACIFIC CONFERENCE ON WIRELESS AND MOBILE (APWIMOB), 2022, : 90 - 95
  • [38] Deep Learning for Brain Tumor Classification
    Paul, Justin S.
    Plassard, Andrew J.
    Landman, Bennett A.
    Fabbri, Daniel
    MEDICAL IMAGING 2017: BIOMEDICAL APPLICATIONS IN MOLECULAR, STRUCTURAL, AND FUNCTIONAL IMAGING, 2017, 10137
  • [39] Hybrid deep spatial and statistical feature fusion for accurate MRI brain tumor classification
    Iqbal, Saeed
    Qureshi, Adnan N.
    Alhussein, Musaed
    Aurangzeb, Khursheed
    Choudhry, Imran Arshad
    Anwar, Muhammad Shahid
    FRONTIERS IN COMPUTATIONAL NEUROSCIENCE, 2024, 18
  • [40] Fusion-Based Machine Learning Architecture for Heart Disease Prediction
    Nadeem, Muhammad Waqas
    Goh, Hock Guan
    Khan, Muhammad Adnan
    Hussain, Muzammil
    Mushtaq, Muhammad Faheem
    Ponnusamy, Vasaki A. P.
    CMC-COMPUTERS MATERIALS & CONTINUA, 2021, 67 (02): : 2481 - 2496