Results in Control and Optimization (Jun 2023)
StressNet: Hybrid model of LSTM and CNN for stress detection from electroencephalogram signal (EEG)
Abstract
Everyday tasks can cause stress, which can lead to serious medical conditions, such as depression. EEG signal processing can assist medical professionals in managing the emotional state of patients. To detect stress states in EEG signals, we propose a new architecture, StressNet, which is a combination of a two-dimensional convolutional neural network (CNN) and a long short-term memory (LSTM) network. The EEG signals are first decomposed into alpha, beta, and theta signals, which are then used to generate azimuthal projection-based images. These images are fed into the 2D CNN for feature extraction, which are then passed to the LSTM for further processing. The LSTM’s ability to remember past states makes it particularly effective in modeling the temporal dynamics of EEG signals. Finally, the StressNet model classifies the features using fully connected layers into either stress or normal classes. We evaluate the performance of our model on the DEEP and SEED datasets and compare it to other methods. The results show that the proposed StressNet model outperforms the human stress detection accuracy, achieving an accuracy of 97.8%.