Machine Learning and Electroencephalogram Signal based Diagnosis of Dipression.

Journal: Neuroscience letters
Published Date:

Abstract

Depression is a psychological condition which hampers day to day activity (Thinking, Feeling or Action). The early detection of this illness will help to save many lives because it is now recognized as a global problem which could even lead to suicide. Electroencephalogram (EEG) signals can be used to diagnose depression using machine learning techniques. The dataset studied is public dataset which consists of 30 healthy people and 34 depression patients. The methods used for detection of depression are Decision Tree, Random Forest, Convolutional Neural Network (CNN), Recurrent Neural Network (RNN), Long Short-Term Memory (LSTM), Gated Recurrent Unit (GRU), Bidirectional Long-Short Term Memory (Bi-LSTM), Gradient Boosting, Extreme Gradient Boosting (XGBoost) along with band power. Among Deep Learning techniques, CNN model got the highest accuracy with 98.13%, specificity of 99%, and sensitivity of 97% using band power features.

Authors

  • Adil O Khadidos
    Department of Information Technology, Faculty of Computing and Information Technology, King Abdulaziz University, Jeddah, Saudi Arabia. Electronic address: akhadidos@kau.edu.sa.
  • Khaled H Alyoubi
    Department of Information Technology, Faculty of Computing and Information Technology, King Abdulaziz University, Jeddah, Saudi Arabia; Department of Computer Science, College of Computers and Information Systems, Umm Al-Qura University, Makkah, Saudi Arabia. Electronic address: kalyoubi@kau.edu.sa.
  • Shalini Mahato
    Department of Computer Science and Engineering, Birla Institute of Technology, Ranchi, Mesra, 835215, India. swarup.shalini@gmail.com.
  • Alaa O Khadidos
    Department of Information Systems, Faculty of Computing and Information Technology, King Abdulaziz University, Jeddah, Saudi Arabia. Electronic address: aokhadidos@kau.edu.sa.
  • Sachi Nandan Mohanty
    Department of Computer Science & Engineering, Vardhaman College of Engineering(Autonomous), Hyderabad, India. Electronic address: sachinandan09@gmail.com.