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

被引:152
作者
Kumar, R. Lokesh [1 ]
Kakarla, Jagadeesh [1 ]
Isunuri, B. Venkateswarlu [1 ]
Singh, Munesh [1 ]
机构
[1] IIITDM Kancheepuram, Chennai, Tamil Nadu, India
关键词
Multi-class classification; Brain tumor classification; Deep learning; Residual network; Global average pooling;
D O I
10.1007/s11042-020-10335-4
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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.
引用
收藏
页码:13429 / 13438
页数:10
相关论文
共 50 条
[21]   Three-class brain tumor classification using deep dense inception residual network [J].
Kokkalla, Srinath ;
Kakarla, Jagadeesh ;
Venkateswarlu, Isunuri B. ;
Singh, Munesh .
SOFT COMPUTING, 2021, 25 (13) :8721-8729
[22]   Three-class brain tumor classification using deep dense inception residual network [J].
Srinath Kokkalla ;
Jagadeesh Kakarla ;
Isunuri B. Venkateswarlu ;
Munesh Singh .
Soft Computing, 2021, 25 :8721-8729
[23]   Rice plant disease classification using dilated convolutional neural network with global average pooling [J].
Pandi, S. Senthil ;
Senthilselvi, A. ;
Gitanjali, J. ;
ArivuSelvan, K. ;
Gopal, Jagadeesh ;
Vellingiri, J. .
ECOLOGICAL MODELLING, 2022, 474
[24]   Convolutional Neural Network for Multi-class Classification of Diabetic Eye Disease [J].
Sarki, Rubina ;
Ahmed, Khandakar ;
Wang, Hua ;
Zhang, Yanchun ;
Wang, Kate .
EAI ENDORSED TRANSACTIONS ON SCALABLE INFORMATION SYSTEMS, 2022, 9 (04)
[25]   Intelligent Neural Network Schemes for Multi-Class Classification [J].
You, Ying-Jie ;
Wu, Chen-Yu ;
Lee, Shie-Jue ;
Liu, Ching-Kuan .
APPLIED SCIENCES-BASEL, 2019, 9 (19)
[26]   Detecting steganography using multi-class classification [J].
Rodriguez, Benjamin ;
Peterson, Gilbert .
ADVANCES IN DIGITAL FORENSIC III, 2007, 242 :193-+
[27]   Multi-class classification using a signomial function [J].
Hwang, Kyoungmi ;
Lee, Kyungsik ;
Lee, Chungmok ;
Park, Sungsoo .
JOURNAL OF THE OPERATIONAL RESEARCH SOCIETY, 2015, 66 (03) :434-449
[28]   Multi-Class Classification of Lung Diseases Using CNN Models [J].
Hong, Min ;
Rim, Beanbonyka ;
Lee, Hongchang ;
Jang, Hyeonung ;
Oh, Joonho ;
Choi, Seongjun .
APPLIED SCIENCES-BASEL, 2021, 11 (19)
[29]   Disparity Autoencoders for Multi-class Brain Tumor Segmentation [J].
Yogananda, Chandan Ganesh Bangalore ;
Das, Yudhajit ;
Wagner, Benjamin C. ;
Nalawade, Sahil S. ;
Reddy, Divya ;
Holcomb, James ;
Pinho, Marco C. ;
Fei, Baowei ;
Madhuranthakam, Ananth J. ;
Maldjian, Joseph A. .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2021, PT II, 2022, 12963 :116-124
[30]   Multi-class Review Rating Classification using Deep Recurrent Neural Network [J].
Junaid Hassan ;
Umar Shoaib .
Neural Processing Letters, 2020, 51 :1031-1048