A machine learning framework involving EEG-based functional connectivity to diagnose major depressive disorder (MDD)

作者：Mumtaz Wajid; Ali Syed Saad Azhar; Yasin Mohd Azhar Mohd; Malik Aamir Saeed^*

来源：MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2018, 56(2): 233-246.

DOI：10.1007/s11517-017-1685-z

• Major depressive disorder
• Machine learning-based diagnosis
• Computer-aided diagnosis (CAD) for depression
• EEG-based machine learning techniques for depression
• Synchronization likelihood (SL) features

Major depressive disorder (MDD), a debilitating mental illness, could cause functional disabilities and could become a social problem. An accurate and early diagnosis for depression could become challenging. This paper proposed a machine learning framework involving EEG-derived synchronization likelihood (SL) features as input data for automatic diagnosis of MDD. It was hypothesized that EEG-based SL features could discriminate MDD patients and healthy controls with an acceptable accuracy better than measures such as interhemispheric coherence and mutual information. In this work, classification models such as support vector machine (SVM), logistic regression (LR) and Na < ve Bayesian (NB) were employed to model relationship between the EEG features and the study groups (MDD patient and healthy controls) and ultimately achieved discrimination of study participants. The results indicated that the classification rates were better than chance. More specifically, the study resulted into SVM classification accuracy = 98%, sensitivity = 99.9%, specificity = 95% and f-measure = 0.97; LR classification accuracy = 91.7%, sensitivity = 86.66%, specificity = 96.6% and f-measure = 0.90; NB classification accuracy = 93.6%, sensitivity = 100%, specificity = 87.9% and f-measure = 0.95. In conclusion, SL could be a promising method for diagnosing depression. The findings could be generalized to develop a robust CAD-based tool that may help for clinical purposes.

• 出版日期2018-2