A novel EEG-based major depressive disorder detection framework with two-stage feature selection

Yujie Li,Yingshan Shen,Xiaomao Fan,Xingxian Huang,Haibo Yu,Gansen Zhao,Wenjun Ma
DOI: https://doi.org/10.1186/s12911-022-01956-w
2022-08-06
Abstract:Background: Major depressive disorder (MDD) is a common mental illness, characterized by persistent depression, sadness, despair, etc., troubling people's daily life and work seriously. Methods: In this work, we present a novel automatic MDD detection framework based on EEG signals. First of all, we derive highly MDD-correlated features, calculating the ratio of extracted features from EEG signals at frequency bands between [Formula: see text] and [Formula: see text]. Then, a two-stage feature selection method named PAR is presented with the sequential combination of Pearson correlation coefficient (PCC) and recursive feature elimination (RFE), where the advantages lie in minimizing the feature searching space. Finally, we employ widely used machine learning methods of support vector machine (SVM), logistic regression (LR), and linear regression (LNR) for MDD detection with the merit of feature interpretability. Results: Experiment results show that our proposed MDD detection framework achieves competitive results. The accuracy and [Formula: see text] score are up to 0.9895 and 0.9846, respectively. Meanwhile, the regression determination coefficient [Formula: see text] for MDD severity assessment is up to 0.9479. Compared with existing MDD detection methods with the best accuracy of 0.9840 and [Formula: see text] score of 0.97, our proposed framework achieves the state-of-the-art MDD detection performance. Conclusions: Development of this MDD detection framework can be potentially deployed into a medical system to aid physicians to screen out MDD patients.
What problem does this paper attempt to address?