Classification of Tumor in Brain MR Images Using Deep Convolutional Neural Network and Global Average Pooling

被引:11
|
作者
Malla, Prince Priya [1 ]
Sahu, Sudhakar [1 ]
Alutaibi, Ahmed I. [2 ]
机构
[1] Kalinga Inst Ind Technol, Sch Elect Engn, Bhubaneswar 751024, India
[2] Majmaah Univ, Coll Comp & Informat Sci, Majmaah 11952, Saudi Arabia
关键词
medical imaging; magnetic resonance imaging; deep learning; transfer learning; tumor detection; global average pooling; SEGMENTATION; CNN;
D O I
10.3390/pr11030679
中图分类号
TQ [化学工业];
学科分类号
0817 ;
摘要
Brain tumors can cause serious health complications and lead to death if not detected accurately. Therefore, early-stage detection of brain tumors and accurate classification of types of brain tumors play a major role in diagnosis. Recently, deep convolutional neural network (DCNN) based approaches using brain magnetic resonance imaging (MRI) images have shown excellent performance in detection and classification tasks. However, the accuracy of DCNN architectures depends on the training of data samples since it requires more precise data for better output. Thus, we propose a transfer learning-based DCNN framework to classify brain tumors for example meningioma tumors, glioma tumors, and pituitary tumors. We use a pre-trained DCNN architecture VGGNet which is previously trained on huge datasets and used to transfer its learning parameters to the target dataset. Also, we employ transfer learning aspects such as fine-tune the convolutional network and freeze the layers of the convolutional network for better performance. Further, this proposed approach uses a Global Average Pooling (GAP) layer at the output to avoid overfitting issues and vanishing gradient problems. The proposed architecture is assessed and compared with competing deep learning based brain tumor classification approaches on the Figshare dataset. Our proposed approach produces 98.93% testing accuracy and outperforms the contemporary learning-based approaches.
引用
收藏
页数:17
相关论文
共 50 条
  • [1] Brain Tumor Classification of MRI Images Using Deep Convolutional Neural Network
    Kuraparthi, Swaraja
    Reddy, Madhavi K.
    Sujatha, C. N.
    Valiveti, Himabindu
    Duggineni, Chaitanya
    Kollati, Meenakshi
    Kora, Padmavathi
    Sravan, V
    [J]. TRAITEMENT DU SIGNAL, 2021, 38 (04) : 1171 - 1179
  • [2] An automated brain tumor classification in MR images using an enhanced convolutional neural network
    Singh R.
    Agarwal B.B.
    [J]. International Journal of Information Technology, 2023, 15 (2) : 665 - 674
  • [3] Three-class classification of brain magnetic resonance images using average-pooling convolutional neural network
    Kakarla, Jagadeesh
    Isunuri, Bala Venkateswarlu
    Doppalapudi, Krishna Sai
    Bylapudi, Karthik Satya Raghuram
    [J]. INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2021, 31 (03) : 1731 - 1740
  • [4] Rice plant disease classification using dilated convolutional neural network with global average pooling
    Pandi, S. Senthil
    Senthilselvi, A.
    Gitanjali, J.
    ArivuSelvan, K.
    Gopal, Jagadeesh
    Vellingiri, J.
    [J]. ECOLOGICAL MODELLING, 2022, 474
  • [5] Study on Brain Tumor Classification Through MRI Images Using a Deep Convolutional Neural Network
    Sharma, Kirti
    Khanna, Ketna
    Gambhir, Sapna
    Gambhir, Mohit
    [J]. INTERNATIONAL JOURNAL OF INFORMATION RETRIEVAL RESEARCH, 2022, 12 (01)
  • [6] Teeth category classification via seven-layer deep convolutional neural network with max pooling and global average pooling
    Li, Zhi
    Wang, Shui-Hua
    Fan, Rui-Rui
    Cao, Gang
    Zhang, Yu-Dong
    Guo, Ting
    [J]. INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2019, 29 (04) : 577 - 583
  • [7] Classification of Brain Tumors using Convolutional Neural Network from MR Images
    Gungen, Cahfer
    Polat, Ozlem
    Karakis, Rukiye
    [J]. 2020 28TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE (SIU), 2020,
  • [8] Brain MR Image Classification for Glioma Tumor detection using Deep Convolutional Neural Network Features
    Latif, Ghazanfar
    Iskandar, D. N. F. Awang
    Alghazo, Jaafar
    Butt, M. Mohsin
    [J]. CURRENT MEDICAL IMAGING, 2021, 17 (01) : 56 - 63
  • [9] Classification of Bone Tumor on CT Images Using Deep Convolutional Neural Network
    Li, Yang
    Zhou, Wenyu
    Lv, Guiwen
    Luo, Guibo
    Zhu, Yuesheng
    Liu, Ji
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2018, PT II, 2018, 11140 : 127 - 136
  • [10] Multi-class brain tumor classification using residual network and global average pooling
    Kumar, R. Lokesh
    Kakarla, Jagadeesh
    Isunuri, B. Venkateswarlu
    Singh, Munesh
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (09) : 13429 - 13438