Abstract:The manual classification of primary brain tumors through Magnetic Resonance Imaging (MRI) is considered as a critical task during the clinical routines that requires highly qualified neuroradiologists. Deep Learning (DL)-based computer-aided diagnosis tools are established to support the neurosurgeons' opinion during the diagnosis. However, the black-box nature and the lack of transparency and interpretability of such DL-based models make their implementation, especially in critical and sensitive medical applications, very difficult. The explainable artificial intelligence techniques help to gain clinicians' confidence and to provide explanations about the models' predictions. Typical and existing Convolutional Neural Network (CNN)-based architectures could not capture long-range global information and feature from pathology MRI scans. Recently, Vision Transformer (ViT) networks have been introduced to solve the issue of long-range dependency in CNN-based architecture by introducing a self-attention mechanism to analyze images, allowing the network to capture deep long-range reliance between pixels. The purpose of the proposed study is to provide efficient CAD tool for MRI brain tumor classification. At the same, we aim to enhance the neuroradiologists' confidence when using DL in clinical and medical standards. In this paper, we investigated a deep ViT architecture trained from scratch for the multi-classification task of common primary tumors (gliomas, meningiomas, and pituitary brain tumors), using T1-weighted contrast-enhanced MRI sequences. Several XAI techniques have been adopted: Gradient-weighted Class Activation Mapping (Grad-CAM), Local Interpretable Model-agnostic Explanations (LIME), and SHapley Additive exPlanations (SHAP), to visualize the most significant and distinguishing features related to the model prediction results. A publicly available benchmark dataset has been used for the evaluation task. The comparative study confirms the efficiency of ViT architecture compared to the CNN model using the testing dataset. The test accuracy of 83.37% for the Convolutional Neural Network (CNN) and 91.61% for the Vision Transformer (ViT) indicates that the ViT model outperformed the CNN model in the classification task. Based on the experimental results, we could confirm that the proposed ViT model presents a competitive performance outperforming the multi-classification state-of-the-art models using MRI sequences. Further, the proposed models present an exact and correct interpretation. Thus, we could confirm that the proposed CAD could be established during the clinical diagnosis routines.

Construction material classification on imbalanced datasets using Vision Transformer (ViT) architecture

Fish Detection and Classification Based on Improved ViT

Material recognition for construction quality monitoring using deep learning methods

Material Recognition for Automated Progress Monitoring using Deep Learning Methods

HViT: Hybrid vision inspired transformer for the assessment of carotid artery plaque by addressing the cross-modality domain adaptation problem in MRI

DenseViT: A Hybrid CNN-Vision Transformer Model for an Improved Multisensor Lithological Classification

Vision transformers (ViT) and deep convolutional neural network (D-CNN)-based models for MRI brain primary tumors images multi-classification supported by explainable artificial intelligence (XAI)

Implementing vision transformer for classifying 2D biomedical images

ViTAD: Leveraging Modified Vision Transformer for Alzheimer's Disease Multi-Stage Classification from Brain MRI Scans

A deep learning based approach for automated plant disease classification using vision transformer

Identifying Malignant Breast Ultrasound Images Using ViT-Patch

Automated Detection of Defects on Metal Surfaces using Vision Transformers

A novel Skin lesion prediction and classification technique: ViT-GradCAM

Exploring Plain ViT Reconstruction for Multi-class Unsupervised Anomaly Detection

Extended Vision Transformer (ExViT) for Land Use and Land Cover Classification: A Multimodal Deep Learning Framework

A shape-aware enhancement Vision Transformer for building extraction from remote sensing imagery

Transformer-Based Disease Identification for Small-Scale Imbalanced Capsule Endoscopy Dataset

Deep Hierarchical Vision Transformer for Hyperspectral and LiDAR Data Classification

MedViT: A robust vision transformer for generalized medical image classification

Asymmetric Vision Transformers for Multi-Label Classification

Multivariate image processing in minerals engineering with vision transformers