A Linear Time Shrinking-SL(t)-ViT Approach for Brain Tumor Identification and Categorization

Abstract

Automated brain tumor detection and classification systems have gained popularity in recent years because traditional diagnosis procedures are time-consuming and costly in nature. Deep learning(DL) methods, specifically pre-trained convolutional neural networks (CNNs), have shown promising results in accurately and rapidly classifying brain tumors. However, the lack of diverse magnetic resonance imaging (MRI) datasets has hindered the ability of DL algorithms to generalize effectively. To address this issue, this paper proposes a DL model using generative adversarial networks (GANs) in conjunction with data augmentation strategies and a structural similarity loss function employed for generating annotated images. A novel DL model inspired from Vision Transformer, Shrinking Linear Time Vision Transformer (SL(t)-ViT) network model is proposed for disease data classification. The model underwent extensive evaluation across multiple datasets, employing standard performance metrics to assess its efficacy in brain tumor identification. The proposed model achieved remarkable testing accuracies of 0.995, 0.996, 0.9954, 0.998, and 0.997 for binary classification tasks across multiple datasets, and 0.986, 0.982, 0.985, and 0.993 for multi-class classification tasks. These results underscore the superior performance of our proposed model, showcasing its capability to outperform state-of-the-art techniques. Specifically, it demonstrated a substantial margin of improvement, ranging from 1-2% for binary classification and 9-10% for multi-class classification, solidifying its position as a leading approach in brain tumor identification. Results demonstrate the efficacy of the proposed model, outperforming other models. Overall, this study highlights the potential of SL(t)-ViT and GANs in improving the accuracy, resource consumption, and efficiency of brain tumor diagnosis.