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 条
[31]   Multi-class Review Rating Classification using Deep Recurrent Neural Network [J].
Hassan, Junaid ;
Shoaib, Umar .
NEURAL PROCESSING LETTERS, 2020, 51 (01) :1031-1048
[32]   Teeth category classification via seven-layer deep convolutional neural network with max pooling and global average pooling [J].
Li, Zhi ;
Wang, Shui-Hua ;
Fan, Rui-Rui ;
Cao, Gang ;
Zhang, Yu-Dong ;
Guo, Ting .
INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2019, 29 (04) :577-583
[33]   An efficient deep neural network based abnormality detection and multi-class breast tumor classification [J].
Rakesh Chandra Joshi ;
Divyanshu Singh ;
Vaibhav Tiwari ;
Malay Kishore Dutta .
Multimedia Tools and Applications, 2022, 81 :13691-13711
[34]   An efficient deep neural network based abnormality detection and multi-class breast tumor classification [J].
Joshi, Rakesh Chandra ;
Singh, Divyanshu ;
Tiwari, Vaibhav ;
Dutta, Malay Kishore .
MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (10) :13691-13711
[35]   Neural network for multi-class classification by boosting composite stumps [J].
Nie, Qingfeng ;
Jin, Lizuo ;
Fei, Shumin ;
Ma, Junyong .
Neurocomputing, 2015, 149 (PB) :949-956
[36]   Neural network for multi-class classification by boosting composite stumps [J].
Nie, Qingfeng ;
Jin, Lizuo ;
Fei, Shumin ;
Ma, Junyong .
NEUROCOMPUTING, 2015, 149 :949-956
[37]   A genetically optimized neural network model for multi-class classification [J].
Bhardwaj, Arpit ;
Tiwari, Aruna ;
Bhardwaj, Harshit ;
Bhardwaj, Aditi .
EXPERT SYSTEMS WITH APPLICATIONS, 2016, 60 :211-221
[38]   Binary and multi-class classification of Android applications using static features [J].
Dhalaria, Meghna ;
Gandotra, Ekta .
INTERNATIONAL JOURNAL OF APPLIED MANAGEMENT SCIENCE, 2023, 15 (02) :117-140
[39]   Multi-class classification using quantum transfer learning [J].
Bidisha Dhara ;
Monika Agrawal ;
Sumantra Dutta Roy .
Quantum Information Processing, 23
[40]   Multi-class pattern classification using neural networks [J].
Ou, Guobin ;
Murphey, Yi Lu .
PATTERN RECOGNITION, 2007, 40 (01) :4-18