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

被引:21
|
作者
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
    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.
    International Journal of Information Technology, 2023, 15 (2) : 665 - 674
  • [3] 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
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2019, 29 (04) : 577 - 583
  • [4] 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
    CURRENT MEDICAL IMAGING, 2021, 17 (01) : 56 - 63
  • [5] Multi-class brain tumor classification using residual network and global average pooling
    R Lokesh Kumar
    Jagadeesh Kakarla
    B Venkateswarlu Isunuri
    Munesh Singh
    Multimedia Tools and Applications, 2021, 80 : 13429 - 13438
  • [6] Multi-class brain tumor classification using residual network and global average pooling
    Kumar, R. Lokesh
    Kakarla, Jagadeesh
    Isunuri, B. Venkateswarlu
    Singh, Munesh
    MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (09) : 13429 - 13438
  • [7] 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.
    ECOLOGICAL MODELLING, 2022, 474
  • [8] Study on Brain Tumor Classification Through MRI Images Using a Deep Convolutional Neural Network
    Sharma, Kirti
    Khanna, Ketna
    Gambhir, Sapna
    Gambhir, Mohit
    INTERNATIONAL JOURNAL OF INFORMATION RETRIEVAL RESEARCH, 2022, 12 (01)
  • [9] Classification of Brain Tumours in MRI Images using a Convolutional Neural Network
    Gupta, Isha
    Singh, Swati
    Gupta, Sheifali
    Nayak, Soumya Ranjan
    CURRENT MEDICAL IMAGING, 2023, 20
  • [10] Transfer Learning Using Convolutional Neural Network Architectures for Brain Tumor Classification from MRI Images
    Chelghoum, Rayene
    Ikhlef, Ameur
    Hameurlaine, Amina
    Jacquir, Sabir
    ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, AIAI 2020, PT I, 2020, 583 : 189 - 200