Brain stimulation outcome prediction in Major Depressive Disorder by deep learning models using EEG representations

Major Depressive Disorder (MDD) is known as a widespread illness and needs a timely treatment. The treatment procedure is currently based on the trial and error between various treatments. An individualized treatment selection is crucial for saving time and financial resources and preventing possible side effects. Because of the complex nature of this problem, a Deep Learning (DL) approach, as a promising method for the precision medicine, was utilized to identify the responders to the treatment using pre-treatment EEG signals. Eighty-three patients with MDD participated in this study to receive treatment using Repetitive Transcranial Magnetic Stimulation (rTMS). A deep hybrid neural network was developed based on three pre-trained convolutional neural networks named DenseNet121, EfficientNetB0, and Xception. The training of each model was performed by feeding three types of EEG representations as the inputs into the models including the Wavelet Transform (WT) images, the connectivity matrix between electrode pairs, and the raw EEG signals. The performance of the proposed models were assessed for the three different input types and achieved the highest accuracy of 94.7% in classifying patients as responders or non-responders when utilizing a sequence of raw EEG images. For the WT and connectivity inputs the best accuracy of model was 94.38% and 94.25% respectively. Therefore, the proposed model can be a step forward towards the clinical implementation of an end-to-end treatment selection framework using raw EEG signals.