Prediction of Treatment Outcome in Major Depressive Disorder Using Ensemble of Hybrid Transfer Learning and Long Short-Term Memory Based on EEG Signal

Abstract

Major Depressive Disorder (MDD) is a widespread global mental disease. The Effectiveness of Selective Serotonin Reuptake Inhibitors (SSRIs) antidepressants prescribed for MDD patients is limited and pre-treatment assessment of treatment outcome is a vital task. In this study, a hybrid model based on Transfer Learning (TL) of powerful pre-trained deep Convolutional Neural Networks (CNNs) empowered with bidirectional Long Short-term Memory (BLSTM) cells and attention mechanism were developed to classify Responders (R) and Non-Responders (NR) to SSRI antidepressants, using raw data images of pre-treatment EEG signal obtained from 30 MDD patients. TL-LSTM-Attention models based on VGG16, Xception, and Densenet121 models were created for the classification of R/NR. Results show that the highest performance of hybrid TL-LSTM-Attention models was achieved by VGG16-LSTM-Attention with an accuracy of 98.21%, sensitivity of 96.22%, and specificity of 99.67%. An Ensemble model based on weighted majority voting among hybrid models is constructed to surpass the performance of each single hybrid model. Our proposed ensemble model gained accuracy, sensitivity, and specificity of 98.84%, 97.80%, and 99.60%, respectively. Hence, the proposed model which is an ensemble of TL-LSTM-Attention models fed by raw data images of EEG signal leads to high confidence in the prediction of antidepressants treatment outcome.