BTC-fCNN: Fast Convolution Neural Network for Multi-class Brain Tumor Classification

被引:35
作者
Abd El-Wahab, Basant S. S. [1 ]
Nasr, Mohamed E. E. [1 ]
Khamis, Salah [1 ]
Ashour, Amira S. S. [1 ]
机构
[1] Tanta Univ, Fac Engn, Dept Elect & Elect Commun Engn, Tanta, Egypt
关键词
Brain tumor classification; Convolution neural network; Average pooling layer; Convolution layer; Transfer learning;
D O I
10.1007/s13755-022-00203-w
中图分类号
R-058 [];
学科分类号
摘要
Timely prognosis of brain tumors has a crucial role for powerful healthcare of remedy-making plans. Manual classification of the brain tumors in magnetic resonance imaging (MRI) images is a challenging task, which relies on the experienced radiologists to identify and classify the brain tumor. Automated classification of different brain tumors is significant based on designing computer-aided diagnosis (CAD) systems. Existing classification methods suffer from unsatisfactory performance and/or large computational cost/ time. This paper proposed a fast and efficient classification process, called BTC-fCNN, which is a deep learning-based system to distinguish between different views of three brain tumor types, namely meningioma, glioma, and pituitary tumors. The proposed system's model was applied on MRI images from the Figshare dataset. It consists of 13 layers with few trainable parameters involving convolution layer, 1 x 1 convolution layer, average pooling, fully connected layer, and softmax layer. Five iterations including transfer learning and five-fold cross-validation for retraining are considered to increase the proposed model performance. The proposed model achieved 98.63% average accuracy, using five iterations with transfer learning, and 98.86% using retrained five-fold cross-validation (internal transfer learning between the folds). Various evaluation metrics were measured to evaluate the proposed model, such as precision, F-score, recall, specificity and confusion matrix. The proposed BTC-fCNN model outstrips the state-of-the-art and other well-known convolution neural networks (CNN).
引用
收藏
页数:22
相关论文
共 36 条
[1]   Local feature descriptors based ECG beat classification [J].
Abdullah, Daban Abdulsalam ;
Akpinar, Muhammed H. ;
Sengur, Abdulkadir .
HEALTH INFORMATION SCIENCE AND SYSTEMS, 2020, 8 (01)
[2]   Genetic algorithm based adaptive histogram equalization (GAAHE) technique for medical image enhancement [J].
Acharya, Upendra Kumar ;
Kumar, Sandeep .
OPTIK, 2021, 230
[3]   Enhanced brain tumor classification using an optimized multi-layered convolutional neural network architecture [J].
Alshayeji, Mohammad ;
Al-Buloushi, Jassim ;
Ashkanani, Ali ;
Abed, Sa'ed .
MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (19) :28897-28917
[4]   Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithms [J].
Anaraki, Amin Kabir ;
Ayati, Moosa ;
Kazemi, Foad .
BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2019, 39 (01) :63-74
[5]   Deep learning based brain tumor classification and detection system [J].
Ari, Ali ;
Hanbay, Davut .
TURKISH JOURNAL OF ELECTRICAL ENGINEERING AND COMPUTER SCIENCES, 2018, 26 (05) :2275-2286
[6]   Eggshell crack detection using deep convolutional neural networks [J].
Botta, Bhavya ;
Gattam, Sai Swaroop Reddy ;
Datta, Ashis Kumar .
JOURNAL OF FOOD ENGINEERING, 2022, 315
[7]  
Cheng Jun, 2017, Figshare
[8]   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)
[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]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861