A High-Accuracy Brain Tumor Classification Method Based onVIT in the Context of Small Sample Size
This paper proposes a brain tumor classification method based on VIT (Vision Transformer), which achieves high accuracy in the context of small sample size by classifying and recognizing MRI images. The study utilizes the VIT model to capture the relationship between different regions in the image using self-attention mechanism and enhances the diversity of the dataset through data augmentation techniques. Experimental results demonstrate that this method achieves 100% accuracy in brain tumor image classification tasks, outperforming traditional convolutional neural network models such as VGG16, Alex-Net, ResNet-50, and Inception-V3. This research is of significant importance for automated identification of brain tumor images and holds promise for improving diagnostic efficiency, treatment outcomes, and patient survival rates.
Authors
- Bingmei M. Fu (ORCID: https://orcid.org/0000-0001-9343-5895)
- Jiahao Ding (ORCID: https://orcid.org/0000-0002-2867-4133)
- Xiaoyu Xie (ORCID: https://orcid.org/0000-0003-1234-1032)
Institutions
- Nanjing University of Posts and Telecommunications (CN)
Publication Details
- Published
- 2024-12-13
- DOI
- https://doi.org/10.1109/ichih63459.2024.11064889
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00