Prediction of Depression from EEG Signal Using Long Short Term Memory(LSTM)

Abstract

Depression, a neurological disorder is the leading cause of disability worldwide. EEG recordings have found wide use in the diagnosis and analysis of various neurological disorders including depression. In this paper, LSTM (Long-short term memory) deep learning models are used in the prediction of trends of depression for the next time instants, based on the features extracted. The statistical time-domain feature encompassing the mean of amplitude of the data is extracted employing moving window segmentation from the acquired EEG signals. The model uses one LSTM layer with 10 hidden neurons for the prediction. Out of a total of 7000 mean values calculated from a sample of 30 patient records from each resting states, 5600 sample means were used to train the model. The proposed LSTM network could predict the next 1400 sample mean values accurately with root mean square error of 0.000064. The performance of the model is compared with CNN-LSTM and ConvLSTM. It is observed that LSTM predictor model works best for the prediction of trends of depression.