Brain Tumor Classification Model of ResNet-50 Network Based on Different Data Enhancement Algorithms

Menghan Zou,Mingze Ma,Anan Wang,Yujie Li,Teoh Teik Toe
DOI: https://doi.org/10.1109/icsp58490.2023.10248544
2023-01-01
Abstract:The accuracy and stability of brain tumor MRI image classification is significant for the healthcare system, but the traditional models have the defects of difficulty in handling complex features and unstable classification. In this paper, we propose a novel brain tumor classification model based on residual neural network, and use three different data enhancement algorithms: geometric transformation, mixup, and SamplePairing to process a dataset containing four different types of brain tumor images, and finally evaluate the classification effect of our proposed model. Among them, the best results were achieved by using geometric transformations for data enhancement.
What problem does this paper attempt to address?