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

被引:14
|
作者
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
相关论文
共 50 条
  • [1] Brain tumor classification using deep convolutional autoencoder-based neural network: multi-task approach
    Fatemh Bashir-Gonbadi
    Hassan Khotanlou
    [J]. Multimedia Tools and Applications, 2021, 80 : 19909 - 19929
  • [2] Convolutional Autoencoder-Based Transfer Learning for Multi-Task Image Inferences
    Lu, Jie
    Verma, Naveen
    Jha, Niraj K.
    [J]. IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTING, 2022, 10 (02) : 1045 - 1057
  • [3] Autoencoder-based multi-task learning for imputation and classification of incomplete data
    Lai, Xiaochen
    Wu, Xia
    Zhang, Liyong
    [J]. Applied Soft Computing, 2021, 98
  • [4] Autoencoder-based multi-task learning for imputation and classification of incomplete data
    Lai, Xiaochen
    Wu, Xia
    Zhang, Liyong
    [J]. APPLIED SOFT COMPUTING, 2021, 98
  • [5] Simultaneous Object Classification and Viewpoint Estimation using Deep Multi-task Convolutional Neural Network
    Afifi, Ahmed J.
    Hellwich, Olaf
    Soomro, Toufique A.
    [J]. PROCEEDINGS OF THE 13TH INTERNATIONAL JOINT CONFERENCE ON COMPUTER VISION, IMAGING AND COMPUTER GRAPHICS THEORY AND APPLICATIONS (VISIGRAPP 2018), VOL 5: VISAPP, 2018, : 177 - 184
  • [6] Brain MRI Image Classification for Cancer Detection Using Deep Wavelet Autoencoder-Based Deep Neural Network
    Mallick, Pradeep Kumar
    Ryu, Seuc Ho
    Satapathy, Sandeep Kumar
    Mishra, Shruti
    Gia Nhu Nguyen
    Tiwari, Prayag
    [J]. IEEE ACCESS, 2019, 7 : 46278 - 46287
  • [7] Brain Networks Classification Based on an Adaptive Multi-Task Convolutional Neural Networks
    Xing X.
    Ji J.
    Yao Y.
    [J]. Jisuanji Yanjiu yu Fazhan/Computer Research and Development, 2020, 57 (07): : 1449 - 1459
  • [8] Multi-task deep convolutional neural network for cancer diagnosis
    Liao, Qing
    Ding, Ye
    Jiang, Zoe L.
    Wang, Xuan
    Zhang, Chunkai
    Zhang, Qian
    [J]. NEUROCOMPUTING, 2019, 348 : 66 - 73
  • [10] Indoor Localization With an Autoencoder-Based Convolutional Neural Network
    Arslantas, Hatice
    Okdem, Selcuk
    [J]. IEEE ACCESS, 2024, 12 : 46059 - 46066