Multi-class brain tumor classification using residual network and global average pooling

doi:10.1007/s11042-020-10335-4

Multi-class brain tumor classification using residual network and global average pooling

2021-01-15

137

PRE

AI

Bala Venkateswarlu Isunuri

M

Munesh Singh

DOI：10.1007/s11042-020-10335-4

Original

Original request for help

Save

Abstract

En 中文

A rapid increase in brain tumor cases mandates researchers for the automation of brain tumor detection and diagnosis. Multi-tumor brain image classification became a contemporary research task due to the diverse characteristics of tumors. Recently, deep neural networks are commonly used for medical image classification to assist neurologists. Vanishing gradient problem and overfitting are the demerits of the deep networks. In this paper, we have proposed a deep network model that uses ResNet-50 and global average pooling to resolve the vanishing gradient and overfitting problems. To evaluate the efficiency of the proposed model simulation has been carried out using a three-tumor brain magnetic resonance image dataset consisting of 3064 images. Key performance metrics have used to analyze the performance of the proposed model and its competitive models. We have achieved a mean accuracy of 97.08% and 97.48% with data augmentation and without data augmentation, respectively. Our proposed model outperforms existing models in classification accuracy.

Keywords:

Multi-class classification

Brain tumor classification

Deep learning

Residual network

Global average pooling

Multi-class brain tumor classification using residual network and global average pooling

Abstract

Journal

Organization