A comparative study of brain tumor classification on unbalanced dataset using deep neural networks

被引:7
作者
Agrawal, Tarun [1 ]
Choudhary, Prakash [2 ]
Kumar, Vijay [3 ]
Singh, Prabhishek [4 ]
Diwakar, Manoj [5 ,6 ]
Kumar, Sandeep [7 ]
机构
[1] Jaypee Inst Informat Technol, Dept Comp Sci Engn & IT, Noida, India
[2] Cent Univ Rajasthan, Dept Comp Sci Engn, Ajmer, India
[3] Dr B R Ambedkar Natl Inst Technol, Dept Informat Technol, Jalandhar, India
[4] Bennett Univ, Sch Comp Sci Engn & Technol, Greater Noida, India
[5] G Era Deemed be Univ, Dept Comp Sci Engn, Dehra Dun, India
[6] Graph Era Hill Univ, Dehra Dun, India
[7] Maharaja Surajmal Inst Technol, Dept Comp Sci & Engn, Delhi, India
关键词
Brain tumor classification; Deep learning; MRI images; Convolutional neural network; Unbalanced dataset;
D O I
10.1016/j.bspc.2024.106256
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
A brain tumor can pose a significant risk to a person's life if it is not promptly detected and diagnosed. The transfer learning approach is used in the majority of research studies on the identification of brain tumors using magnetic resonance imaging (MRI). In this study, ten state-of-the-art deep learning models are evaluated and compared on an unbalanced dataset for three-class brain tumor classification. Experimentation results reveal that the Inception models outperformed all the models for the three-class classification. However, the EfficientNet model does not perform well in comparison to all other models. Moreover, this study provides insight into the well-known deep neural networks on the MRI dataset for brain tumor classification. This study will allow the readers to analyze the performance of these models provided through detailed tables, confusion metrics, and loss graphs in their future research.
引用
收藏
页数:19
相关论文
共 47 条
[1]   ALCNN: Attention based lightweight convolutional neural network for pneumothorax detection in chest X-rays [J].
Agrawal, Tarun ;
Choudhary, Prakash .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 79
[2]   FocusCovid: automated COVID-19 detection using deep learning with chest X-ray images [J].
Agrawal, Tarun ;
Choudhary, Prakash .
EVOLVING SYSTEMS, 2022, 13 (04) :519-533
[3]   An enhanced deep learning method for multi-class brain tumor classification using deep transfer learning [J].
Asif, Sohaib ;
Zhao, Ming ;
Tang, Fengxiao ;
Zhu, Yusen .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (20) :31709-31736
[4]   Deep CNN for Brain Tumor Classification [J].
Ayadi, Wadhah ;
Elhamzi, Wajdi ;
Charfi, Imen ;
Atri, Mohamed .
NEURAL PROCESSING LETTERS, 2021, 53 (01) :671-700
[5]   Joint training of two-channel deep neural network for brain tumor classification [J].
Bodapati, Jyostna Devi ;
Shaik, Nagur Shareef ;
Naralasetti, Veeranjaneyulu ;
Mundukur, Nirupama Bhat .
SIGNAL IMAGE AND VIDEO PROCESSING, 2021, 15 (04) :753-760
[6]   Enhanced Performance of Brain Tumor Classification via Tumor Region Augmentation and Partition [J].
Cheng, Jun ;
Huang, Wei ;
Cao, Shuangliang ;
Yang, Ru ;
Yang, Wei ;
Yun, Zhaoqiang ;
Wang, Zhijian ;
Feng, Qianjin .
PLOS ONE, 2015, 10 (10)
[7]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[8]   DFU_SPNet: A stacked parallel convolution layers based CNN to improve Diabetic Foot Ulcer classification [J].
Das, Sujit Kumar ;
Roy, Pinki ;
Mishra, Arnab Kumar .
ICT EXPRESS, 2022, 8 (02) :271-275
[9]   Brain tumor classification using deep CNN features via transfer learning [J].
Deepak, S. ;
Ameer, P. M. .
COMPUTERS IN BIOLOGY AND MEDICINE, 2019, 111
[10]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848