Intelligent Stress Detection Using ECG Signals: Power Spectrum Imaging with Continuous Wavelet Transform and CNN
Rodrigo Mateo-Reyes, Irving A. Cruz-Albarrán, Luis A. Morales‐Hernandez
Abstract
Stress is a natural response of the organism to challenging situations, but its accurate detection is challenging due to its subjective nature. This study proposes a model based on depth-separable convolutional neural networks (DSCNN) to analyze heart rate variability (HRV) and detect stress. Electrocardiogram (ECG) signals are pre-processed to remove noise and ensure data quality. The signals are then transformed into two-dimensional images using the continuous wavelet transform (CWT) to identify pattern recognition in the time–frequency domain. These representations are classified using the DSCNN model to determine the presence of stress. The methodology has been validated using the SWELL-KW dataset, achieving an accuracy of 99.9% by analyzing the variability in three states (neutral, time pressure, and interruptions) of the 25 samples in the experiment, scanning the acquired signal every 5 s for 45 min per state. The proposed approach is characterized by its ability to transform ECG signals into time–frequency representations by means of short duration sampling, achieving an accurate classification of stress states without the need for complex feature extraction processes. This model is an efficient and accurate tool for stress analysis from biomedical signals.