Brain Magnetic Resonance Imaging Classification Using Deep Learning Architectures with Gender and Age.

Journal: Sensors (Basel, Switzerland)
Published Date:

Abstract

Usage of effective classification techniques on Magnetic Resonance Imaging (MRI) helps in the proper diagnosis of brain tumors. Previous studies have focused on the classification of normal (nontumorous) or abnormal (tumorous) brain MRIs using methods such as Support Vector Machine (SVM) and AlexNet. In this paper, deep learning architectures are used to classify brain MRI images into normal or abnormal. Gender and age are added as higher attributes for more accurate and meaningful classification. A deep learning Convolutional Neural Network (CNN)-based technique and a Deep Neural Network (DNN) are also proposed for effective classification. Other deep learning architectures such as LeNet, AlexNet, ResNet, and traditional approaches such as SVM are also implemented to analyze and compare the results. Age and gender biases are found to be more useful and play a key role in classification, and they can be considered essential factors in brain tumor analysis. It is also worth noting that, in most circumstances, the proposed technique outperforms both existing SVM and AlexNet. The overall accuracy obtained is 88% (LeNet Inspired Model) and 80% (CNN-DNN) compared to SVM (82%) and AlexNet (64%), with best accuracy of 100%, 92%, 92%, and 81%, respectively.

Authors

  • Imayanmosha Wahlang
    Department of Information Technology, North-Eastern Hill University, Shillong 793022, India.
  • Arnab Kumar Maji
    Department of Information Technology, North-Eastern Hill University, Shillong 793022, India.
  • Goutam Saha
    Department of Electronics and Electrical Communication Engineering, Indian Institute of Technology, Kharagpur, Kharagpur 721302, India. Electronic address: gsaha@ece.iitkgp.ernet.in.
  • Prasun Chakrabarti
    Deputy Provost, ITM SLS Baroda University, Vadodara, India.
  • Michal Jasinski
    Department of Electrical Engineering Fundamentals, Faculty of Electrical Engineering, Wroclaw University of Science and Technology, 50-370 Wroclaw, Poland.
  • Zbigniew Leonowicz
    Department of Electrical Engineering Fundamentals, Faculty of Electrical Engineering, Wroclaw University of Science and Technology, 50-370 Wroclaw, Poland.
  • Elzbieta Jasinska
    Department of Operations Research and Business Intelligence, Wrocław University of Science and Technology, 50-370 Wrocław, Poland.