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

被引:137
|
作者
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] Deep Decision Network for Multi-Class Image Classification
    Murthy, Venkatesh N.
    Singh, Vivek
    Chen, Terrence
    Manmatha, R.
    Comaniciu, Dorin
    2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 2240 - 2248
  • [32] Intelligent Neural Network Schemes for Multi-Class Classification
    You, Ying-Jie
    Wu, Chen-Yu
    Lee, Shie-Jue
    Liu, Ching-Kuan
    APPLIED SCIENCES-BASEL, 2019, 9 (19):
  • [33] Hierarchical Learning for Large Multi-class Network Classification
    Liu, Lei
    2016 23RD INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2016, : 2307 - 2312
  • [34] Steganographic domain classification using multi-class
    Xu Bo
    Wang Jiazhen
    Liu Xiaqin
    Yang Sumin
    ISTM/2007: 7TH INTERNATIONAL SYMPOSIUM ON TEST AND MEASUREMENT, VOLS 1-7, CONFERENCE PROCEEDINGS, 2007, : 1270 - 1273
  • [35] Detecting steganography using multi-class classification
    Rodriguez, Benjamin
    Peterson, Gilbert
    ADVANCES IN DIGITAL FORENSIC III, 2007, 242 : 193 - +
  • [36] Multi-class classification using a signomial function
    Hwang, Kyoungmi
    Lee, Kyungsik
    Lee, Chungmok
    Park, Sungsoo
    JOURNAL OF THE OPERATIONAL RESEARCH SOCIETY, 2015, 66 (03) : 434 - 449
  • [37] Using discriminant analysis for multi-class classification
    Li, T
    Zhu, SH
    Ogihara, M
    THIRD IEEE INTERNATIONAL CONFERENCE ON DATA MINING, PROCEEDINGS, 2003, : 589 - 592
  • [38] Multi-class Review Rating Classification using Deep Recurrent Neural Network
    Junaid Hassan
    Umar Shoaib
    Neural Processing Letters, 2020, 51 : 1031 - 1048
  • [39] Disparity Autoencoders for Multi-class Brain Tumor Segmentation
    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
  • [40] Multi-class Review Rating Classification using Deep Recurrent Neural Network
    Hassan, Junaid
    Shoaib, Umar
    NEURAL PROCESSING LETTERS, 2020, 51 (01) : 1031 - 1048