Study of a Hybrid CNN-SVM Model for Stress Detection with Automated Heart Rate Variability Feature Extraction Method
In the era of big data, stress detection techniques based on electrocardiogram (ECG) data play a crucial role in monitoring mental health and preventing mental disorders. Given that the relationship between various heart rate variability (HRV) parameters extracted from ECG signals and stress levels requires further exploration, this paper presents a hybrid model combining a Convolutional Neural Network (CNN) and a Support Vector Machine (SVM). The proposed model leverages the CNN's robust feature extraction capabilities alongside the SVM's superior performance in high-dimensional classification tasks to achieve accurate stress detection. For this study, two publicly available datasets, WESAD (Wearable Stress and Affect Detection) and SWELL (SWELL-KW), are employed, both containing ECG signals recorded under different stress conditions. Experimental results demonstrate that the CNN-SVM model achieves a stress detection accuracy of 0.99 on both datasets, underscoring its potential for reliable stress detection tasks.
Authors
- Guangshun Chen
- Dalin Li (ORCID: https://orcid.org/0000-0002-3086-9087)
- Tianyuan Zhu
- Mengting Huang (ORCID: https://orcid.org/0000-0001-7583-7970)
- Haidong Yang
- Ning Sun (ORCID: https://orcid.org/0000-0003-0396-1694)
- Yanchun Liang
- Gang Yong Lin (ORCID: https://orcid.org/0000-0002-4780-3973)
Institutions
- Zhuhai Institute of Advanced Technology (CN)
- City University of Macau (MO)
Publication Details
- Published
- 2024-12-13
- DOI
- https://doi.org/10.1109/ichih63459.2024.11064910
- Citations
- 2
- Primary Topic
- Emotion and Mood Recognition
- Type
- article
- Field-Weighted Citation Impact
- 1.02
Funders
- National Natural Science Foundation of China