A machine learning based depression screening framework using temporal domain features of the electroencephalography signals

PLoS One. 2024 Mar 27;19(3):e0299127. doi: 10.1371/journal.pone.0299127. eCollection 2024.

Abstract

Depression is a serious mental health disorder affecting millions of individuals worldwide. Timely and precise recognition of depression is vital for appropriate mediation and effective treatment. Electroencephalography (EEG) has surfaced as a promising tool for inspecting the neural correlates of depression and therefore, has the potential to contribute to the diagnosis of depression effectively. This study presents an EEG-based mental depressive disorder detection mechanism using a publicly available EEG dataset called Multi-modal Open Dataset for Mental-disorder Analysis (MODMA). This study uses EEG data acquired from 55 participants using 3 electrodes in the resting-state condition. Twelve temporal domain features are extracted from the EEG data by creating a non-overlapping window of 10 seconds, which is presented to a novel feature selection mechanism. The feature selection algorithm selects the optimum chunk of attributes with the highest discriminative power to classify the mental depressive disorders patients and healthy controls. The selected EEG attributes are classified using three different classification algorithms i.e., Best- First (BF) Tree, k-nearest neighbor (KNN), and AdaBoost. The highest classification accuracy of 96.36% is achieved using BF-Tree using a feature vector length of 12. The proposed mental depressive classification scheme outperforms the existing state-of-the-art depression classification schemes in terms of the number of electrodes used for EEG recording, feature vector length, and the achieved classification accuracy. The proposed framework could be used in psychiatric settings, providing valuable support to psychiatrists.

MeSH terms

  • Algorithms
  • Depression* / diagnosis
  • Electroencephalography
  • Humans
  • Machine Learning
  • Support Vector Machine*

Grants and funding

This study was financially supported by the European Union within the REFRESH project, "Research Excellence for Region Sustainability and High-tech Industries" as part of the European Just Transition Fund, in the form of a grant (CZ.10.03.01/00/22_003/0000048) received by JF. No additional external funding was received for this study.