Brain tumor classification using deep convolutional autoencoder-based neural network: multi-task approach

被引:20
作者
Bashir-Gonbadi, Fatemh [1 ]
Khotanlou, Hassan [1 ]
机构
[1] Bu Ali Sina Univ, Dept Comp Engn, RIV Lab, Hamadan, Hamadan, Iran
关键词
Auto-encoder; Brain tumor; Deep learning; Convolutional neural network; IMAGES; MACHINE; SYSTEM; SEGMENTATION;
D O I
10.1007/s11042-021-10637-1
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diagnosis, detection and classification of tumors, in the brain MRI images, are important because misdiagnosis can lead to death. This paper proposes a method that can diagnose brain tumors in the MRI images and classify them into 5 categories using a Convolutional Neural Network (CNN). The proposed network uses a Convolutional Auto-Encoder Neural Network (CANN) to extract and learn deep features of input images. Extracted deep features from each level are combined to make desirable features and improve results. To classify brain tumor into three categories (Meningioma, Glioma, and Pituitary) the proposed method was applied on Cheng dataset and has reached a considerable performance accuracy of 99.3%. To diagnosis and grading Glioma tumors, the proposed method was applied on IXI and BraTS 2017 datasets, and to classify brain images into six classes including Meningioma, Pituitary, Astrocytoma, High-Grade Glioma, Low-Grade Glioma and Normal images (No tumor), the all datasets including IXI, BraTS2017, Cheng and Hazrat-e-Rassol, was used by the proposed network, and it has reached desirable performance accuracy of 99.1% and 98.5%, respectively.
引用
收藏
页码:19909 / 19929
页数:21
相关论文
共 45 条
[1]  
Afshar P, 2019, INT CONF ACOUST SPEE, P1368, DOI [10.1109/icassp.2019.8683759, 10.1109/ICASSP.2019.8683759]
[2]   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
[3]  
[Anonymous], 2019, IXI BRAIN DEV
[4]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[5]  
Bakas S., 2018, ARXIV
[6]   Automated 3D segmentation of brain tumor using visual saliency [J].
Banerjee, Subhashis ;
Mitra, Sushmita ;
Shankar, B. Uma .
INFORMATION SCIENCES, 2018, 424 :337-353
[7]   Classification of magnetic resonance brain images using wavelets as input to support vector machine and neural network [J].
Chaplot, Sandeep ;
Patnaik, L. M. ;
Jagannathan, N. R. .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2006, 1 (01) :86-92
[8]   Post-Traumatic Stress Symptoms and Post-Traumatic Growth: Evidence from a Longitudinal Study following an Earthquake Disaster [J].
Chen, Jieling ;
Zhou, Xiao ;
Zeng, Min ;
Wu, Xinchun .
PLOS ONE, 2015, 10 (06)
[9]   Comprehensive Predictions of Tourists' Next Visit Location Based on Call Detail Records using Machine Learning and Deep Learning methods [J].
Chen, Nai Chun ;
Xie, Wanqin ;
Xie, Jenny ;
Larson, Kent ;
Welsch, Roy E. .
2017 IEEE 6TH INTERNATIONAL CONGRESS ON BIG DATA (BIGDATA CONGRESS 2017), 2017, :1-6
[10]  
Cheng J., 2017, Brain Tumor Dataset Figshare