Contrastive Self-Supervised Learning for Stress Detection from ECG Data

Autor: Suha Rabbani, Naimul Khan
Jazyk: angličtina
Rok vydání: 2022
Předmět:
Zdroj: Bioengineering, Vol 9, Iss 8, p 374 (2022)
Druh dokumentu: article
ISSN: 2306-5354
DOI: 10.3390/bioengineering9080374
Popis: In recent literature, ECG-based stress assessment has become popular due to its proven correlation to stress and increased accessibility of ECG data through commodity hardware. However, most ECG-based stress assessment models use supervised learning, relying on manually-annotated data. Limited research is done in the area of self-supervised learning (SSL) approaches that leverage unlabelled data and none that utilize contrastive SSL. However, with the dominance of contrastive SSL in domains such as computer vision, it is essential to see if the same excellence in performance can be obtained on an ECG-based stress assessment dataset. In this paper, we propose a contrastive SSL model for stress assessment using ECG signals based on the SimCLR framework. We test our model on two ECG-based stress assessment datasets. We show that our proposed solution results in a 9% improvement in accuracy on the WESAD dataset and 3.7% on the RML dataset when compared with SOTA ECG-based SSL models for stress assessment. The development of more accurate stress assessment models, particularly those that employ non-invasive data such as ECG for assessment, leads to developments in wearable technology and the creation of better health monitoring applications in areas such as stress management and relaxation therapy.
Databáze: Directory of Open Access Journals
Nepřihlášeným uživatelům se plný text nezobrazuje