A Machine Learning Framework for Major Depressive Disorder (MDD) Detection Using Non-invasive EEG Signals

Abstract

According to World Health Organization (WHO) report, every 40 seconds a person attempts suicide globally. Depression, one of the world’s most prevailing diseases has become a reason behind these suicides. It is believed that early diagnosis of major depressive disorder (MDD) can reduce the adversity of this heinous deformity. For few years various machine learning and advanced neurocomputing techniques are being utilized in Electroencephalogram (EEG) based detection of multiple neurological diseases. In the proposed study, an EEG based screening of MDD is presented while using various Machine Learning and one Deep Learning approach. The majority of previous EEG based MDD decoding research has concentrated on a limited features. It was necessary to conduct in-depth comparisons of different approaches, besides more detailed feature-based EEG analysis. This research starts with the creation of a complete feature-based framework, which is then further compared against the state of the art end to end techniques. The K-nearest neighbors (KNN) model outperformed the other models and gained an accuracy of 87.5%. While long short term memory (LSTM) model acquired an accuracy of 83.3%. This study can further support in clinical diagnosis of multiple stages of MDD and can attempt to provide an early intervention.