Vol.69, No.2, 2021, pp.2355-2366, doi:10.32604/cmc.2021.019409
OPEN ACCESS
ARTICLE
A Multi-Category Brain Tumor Classification Method Bases on Improved ResNet50
  • Linguo Li1,2, Shujing Li1,*, Jian Su3
1 Fuyang Normal University, Fuyang, 236041, China
2 Nanjing University of Posts and Telecommunications, Nanjing, 210003, China
3 University of Michigan, Ann Arbor, MI, 48109, USA
* Corresponding Author: Shujing Li. Email:
Received 11 April 2021; Accepted 12 May 2021; Issue published 21 July 2021
Abstract
Brain tumor is one of the most common tumors with high mortality. Early detection is of great significance for the treatment and rehabilitation of patients. The single channel convolution layer and pool layer of traditional convolutional neural network (CNN) structure can only accept limited local context information. And most of the current methods only focus on the classification of benign and malignant brain tumors, multi classification of brain tumors is not common. In response to these shortcomings, considering that convolution kernels of different sizes can extract more comprehensive features, we put forward the multi-size convolutional kernel module. And considering that the combination of average-pooling with max-pooling can realize the complementary of the high-dimensional information extracted by the two structures, we proposed the dual-channel pooling layer. Combining the two structures with ResNet50, we proposed an improved ResNet50 CNN for the applications in multi-category brain tumor classification. We used data enhancement before training to avoid model over fitting and used five-fold cross-validation in experiments. Finally, the experimental results show that the network proposed in this paper can effectively classify healthy brain, meningioma, diffuse astrocytoma, anaplastic oligodendroglioma and glioblastoma.
Keywords
Brain tumor; convolutional neural network; multi-size convolutional kernel module; dual-channel pooling layer; ResNet50
Cite This Article
Li, L., Li, S., Su, J. (2021). A Multi-Category Brain Tumor Classification Method Bases on Improved ResNet50. CMC-Computers, Materials & Continua, 69(2), 2355–2366.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.