A fine-tuned vision transformer based enhanced multi-class brain tumor classification using MRI scan imagery

被引:4
|
作者
Reddy, C. Kishor Kumar [1 ]
Reddy, Pulakurthi Anaghaa [1 ]
Janapati, Himaja [1 ]
Assiri, Basem [2 ]
Shuaib, Mohammed [2 ]
Alam, Shadab [2 ]
Sheneamer, Abdullah [2 ]
机构
[1] Stanley Coll Engn & Technol Women, Dept Comp Sci & Engn, Hyderabad, India
[2] Jazan Univ, Coll Engn & Comp Sci, Dept Comp Sci, Jazan, Saudi Arabia
来源
FRONTIERS IN ONCOLOGY | 2024年 / 14卷
关键词
MRI scans; deep learning models; vision transformers; FTVT; medical image processing; MENINGIOMAS;
D O I
10.3389/fonc.2024.1400341
中图分类号
R73 [肿瘤学];
学科分类号
100214 ;
摘要
Brain tumors occur due to the expansion of abnormal cell tissues and can be malignant (cancerous) or benign (not cancerous). Numerous factors such as the position, size, and progression rate are considered while detecting and diagnosing brain tumors. Detecting brain tumors in their initial phases is vital for diagnosis where MRI (magnetic resonance imaging) scans play an important role. Over the years, deep learning models have been extensively used for medical image processing. The current study primarily investigates the novel Fine-Tuned Vision Transformer models (FTVTs)-FTVT-b16, FTVT-b32, FTVT-l16, FTVT-l32-for brain tumor classification, while also comparing them with other established deep learning models such as ResNet50, MobileNet-V2, and EfficientNet - B0. A dataset with 7,023 images (MRI scans) categorized into four different classes, namely, glioma, meningioma, pituitary, and no tumor are used for classification. Further, the study presents a comparative analysis of these models including their accuracies and other evaluation metrics including recall, precision, and F1-score across each class. The deep learning models ResNet-50, EfficientNet-B0, and MobileNet-V2 obtained an accuracy of 96.5%, 95.1%, and 94.9%, respectively. Among all the FTVT models, FTVT-l16 model achieved a remarkable accuracy of 98.70% whereas other FTVT models FTVT-b16, FTVT-b32, and FTVT-132 achieved an accuracy of 98.09%, 96.87%, 98.62%, respectively, hence proving the efficacy and robustness of FTVT's in medical image processing.
引用
收藏
页数:23
相关论文
共 50 条
  • [31] Multi-class brain tumor classification using residual network and global average pooling
    R Lokesh Kumar
    Jagadeesh Kakarla
    B Venkateswarlu Isunuri
    Munesh Singh
    Multimedia Tools and Applications, 2021, 80 : 13429 - 13438
  • [32] SAR imagery classification using multi-class support vector machines
    Angiulli, G
    Barrile, V
    Cacciola, M
    JOURNAL OF ELECTROMAGNETIC WAVES AND APPLICATIONS, 2005, 19 (14) : 1865 - 1872
  • [33] A novel multi-class brain tumor classification method based on unsupervised PCANet features
    Shahin, Ahmed I.
    Aly, Saleh
    Aly, Walaa
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (15): : 11043 - 11059
  • [34] Brain Tumor Classification Using Fine-Tuned GoogLeNet Features and Machine Learning Algorithms: IoMT Enabled CAD System
    Sekhar, Ardhendu
    Biswas, Soumen
    Hazra, Ranjay
    Sunaniya, Arun Kumar
    Mukherjee, Amrit
    Yang, Lixia
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2022, 26 (03) : 983 - 991
  • [35] Vision Based Nighttime Vehicle Detection Using Adaptive Threshold and Multi-Class Classification
    Sakagawa, Yuta
    Nakajima, Kosuke
    Ohashi, Gosuke
    IEICE TRANSACTIONS ON FUNDAMENTALS OF ELECTRONICS COMMUNICATIONS AND COMPUTER SCIENCES, 2019, E102A (09) : 1235 - 1245
  • [36] Multi-class Motor Imagery Classification using Multi-class SVM with Multi-band Riemannian Tangent Space Mapping
    Shin, Jinhyo
    Chung, Wonzoo
    2023 11TH INTERNATIONAL WINTER CONFERENCE ON BRAIN-COMPUTER INTERFACE, BCI, 2023,
  • [37] Improving Endoscopic Image Analysis: Attention Mechanism Integration in Grid Search Fine-Tuned Transfer Learning Model for Multi-Class Gastrointestinal Disease Classification
    Elmagzoub, Mohamed A.
    Kaur, Swapandeep
    Gupta, Sheifali
    Rajab, Adel
    Rajab, Khairan D.
    Al Reshan, Mana Saleh
    Alshahrani, Hani
    Shaikh, Asadullah
    IEEE ACCESS, 2024, 12 : 80345 - 80358
  • [38] A Deep CNN based Multi-class Classification of Alzheimer's Disease using MRI
    Farooq, Ammarah
    Anwar, Syed Muhammad
    Awais, Muhammad
    Rehman, Saad
    2017 IEEE INTERNATIONAL CONFERENCE ON IMAGING SYSTEMS AND TECHNIQUES (IST), 2017, : 111 - 116
  • [39] FBCSP-based Multi-class Motor Imagery Classification using BP and TDP features
    Abbas, Waseem
    Khan, Nadeem Ahmad
    2018 40TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2018, : 215 - 218
  • [40] An Efficient Multi-Scale Convolutional Neural Network Based Multi-Class Brain MRI Classification for SaMD
    Yazdan, Syed Ali
    Ahmad, Rashid
    Iqbal, Naeem
    Rizwan, Atif
    Khan, Anam Nawaz
    Kim, Do-Hyeun
    TOMOGRAPHY, 2022, 8 (04) : 1905 - 1927