DeprNet: A Deep Convolution Neural Network Framework for Detecting Depression Using EEG

Ayan Seal,Rishabh Bajpai,Jagriti Agnihotri,Anis Yazidi,Enrique Herrera-Viedma,Ondrej Krejcar
DOI: https://doi.org/10.1109/tim.2021.3053999
IF: 5.6
2021-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:Depression is a common reason for an increase in suicide cases worldwide. Thus, to mitigate the effects of depression, accurate diagnosis and treatment are needed. An electroencephalogram (EEG) is an instrument used to measure and record the brain's electrical activities. It can be utilized to produce the exact report on the level of depression. Previous studies proved the feasibility of the usage of EEG data and deep learning (DL) models for diagnosing mental illness. Therefore, this study proposes a DL-based convolutional neural network (CNN) called DeprNet for classifying the EEG data of depressed and normal subjects. Here, the Patient Health Questionnaire 9 score is used for quantifying the level of depression. The performance of DeprNet in two experiments, namely, the recordwise split and the subjectwise split, is presented in this study. The results attained by DeprNet have an accuracy of 0.9937, and the area under the receiver operating characteristic curve (AUC) of 0.999 is achieved when recordwise split data are considered. On the other hand, an accuracy of 0.914 and the AUC of 0.956 are obtained, while subjectwise split data are employed. These results suggest that CNN trained on recordwise split data gets overtrained on EEG data with a small number of subjects. The performance of DeprNet is remarkable compared with the other eight baseline models. Furthermore, on visualizing the last CNN layer, it is found that the values of right electrodes are prominent for depressed subjects, whereas, for normal subjects, the values of left electrodes are prominent.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?