Machine Learning and Electroencephalogram Signal based Diagnosis of Dipression

Neurosci Lett. 2023 Jul 13:809:137313. doi: 10.1016/j.neulet.2023.137313. Epub 2023 May 29.

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.

Keywords: Band power; Depression; Detrended Fluctuation Analysis (DFA); Temporal region.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Case-Control Studies
  • Datasets as Topic
  • Decision Trees
  • Depression* / diagnosis
  • Depression* / psychology
  • Depressive Disorder, Major / diagnosis
  • Depressive Disorder, Major / psychology
  • Electroencephalography*
  • Humans
  • Machine Learning*
  • Neural Networks, Computer
  • Random Forest