A fine-tuned transformer model for brain tumor detection and classification

B. Srinivas,B. Anilkumar,NLakshmi devi,VBKL Aruna
DOI: https://doi.org/10.1007/s11042-024-19652-4
IF: 2.577
2024-06-19
Multimedia Tools and Applications
Abstract:The identification and classification of brain tumors from medical images is a challenging task, which plays a crucial role in treatment planning. In recent times, the transformer models earned greater attention in image classification tasks because of their tendency to learn complex and spatial interconnections in medical images. In this study, a novel fine-tuned transformer model was developed by combining the benefits of a data-efficient image transformer (DeiT) and the firefly algorithm (FA) for the precise classification of tumors. This framework utilizes the transformer's self-attention and feed-forward layers to learn and capture complex patterns and high-level features. It employs the Firefly fitness solution to fine-tune the DeiT architecture by updating its hyperparameters, thereby maximizing the system performance and reducing training loss. The novelty of the paper lies in its proposal and implementation of a fine-tuned transformer model specifically tailored for the detection and classification of brain tumors.. By leveraging the capabilities of transformer architectures, such as the data-efficient image transformer (DeiT), the proposed model can effectively capture intricate spatial dependencies and complex patterns within brain MRI images. The experimental analysis utilizes the benchmark brain tumor MRI database from the Kaggle source and the developed model achieved greater classification accuracy of 99.7%. In addition, a comprehensive comparative analysis was performed with existing models to demonstrate the robustness of the proposed work. This intensive analysis verifies that this model facilitates the accurate and timely detection of brain tumors in medical diagnosis.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?