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

被引:19
|
作者
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 条
  • [41] Brain tumor classification in MRI image using convolutional neural network
    Khan, Hassan Ali
    Jue, Wu
    Mushtaq, Muhammad
    Mushtaq, Muhammad Umer
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2020, 17 (05) : 6203 - 6216
  • [42] Brain tumor classification by using a novel convolutional neural network structure
    Polat, Ozlem
    Dokur, Zumray
    Olmez, Tamer
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2022, 32 (05) : 1646 - 1660
  • [43] Detection and Classification of Brain Tumor Using Convolutional Neural Network (CNN)
    Deshmukh, Smita
    Tiwari, Divya
    MACHINE LEARNING AND BIG DATA ANALYTICS (PROCEEDINGS OF INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND BIG DATA ANALYTICS (ICMLBDA) 2021), 2022, 256 : 289 - 303
  • [44] Brain Tumor Segmentation and Identification Using Particle Imperialist Deep Convolutional Neural Network in MRI Images
    Khemchandani, Maahi Amit
    Jadhav, Shivajirao Manikrao
    Iyer, B. R.
    INTERNATIONAL JOURNAL OF INTERACTIVE MULTIMEDIA AND ARTIFICIAL INTELLIGENCE, 2022, 7 (07): : 38 - 47
  • [45] Accurate brain tumor detection using deep convolutional neural network
    Khan, Md Saikat Islam
    Rahman, Anichur
    Debnath, Tanoy
    Karim, Md Razaul
    Nasir, Mostofa Kamal
    Band, Shahab S.
    Mosavi, Amir
    Dehzangi, Iman
    COMPUTATIONAL AND STRUCTURAL BIOTECHNOLOGY JOURNAL, 2022, 20 : 4733 - 4745
  • [46] Brain Tumor Segmentation using Cascaded Deep Convolutional Neural Network
    Hussain, Saddam
    Anwar, Syed Muhammad
    Majid, Muhammad
    2017 39TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2017, : 1998 - 2001
  • [47] Automatic Segmentation of MR Brain Images With a Convolutional Neural Network
    Moeskops, Pim
    Viergever, Max A.
    Mendrik, Adrienne M.
    de Vries, Linda S.
    Benders, Manon J. N. L.
    Isgum, Ivana
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) : 1252 - 1261
  • [48] GT-Net: global transformer network for multiclass brain tumor classification using MR images
    Dutta, Tapas Kumar
    Nayak, Deepak Ranjan
    Pachori, Ram Bilas
    BIOMEDICAL ENGINEERING LETTERS, 2024, 14 (05) : 1069 - 1077
  • [49] Breast Lesion Classification in Ultrasound Images Using Deep Convolutional Neural Network
    Zeimarani, Bashir
    Fernandes Costa, Marly Guimaraes
    Nurani, Nilufar Zeimarani
    Bianco, Sabrina Ramos
    De Albuquerque Pereira, Wagner Coelho
    Costa Filho, Cicero Ferreira Fernandes
    IEEE ACCESS, 2020, 8 : 133349 - 133359
  • [50] Classification of First Trimester Ultrasound Images Using Deep Convolutional Neural Network
    Singh, Rishi
    Mahmud, Mufti
    Yovera, Luis
    APPLIED INTELLIGENCE AND INFORMATICS, AII 2021, 2021, 1435 : 92 - 105