Classification of Stress via Ambulatory ECG and GSR Data

Zachary Dair,Muhammad Muneeb Saad,Urja Pawar,Samantha Dockray,Ruairi O'Reilly
2023-06-08
Abstract:In healthcare, detecting stress and enabling individuals to monitor their mental health and wellbeing is challenging. Advancements in wearable technology now enable continuous physiological data collection. This data can provide insights into mental health and behavioural states through psychophysiological analysis. However, automated analysis is required to provide timely results due to the quantity of data collected. Machine learning has shown efficacy in providing an automated classification of physiological data for health applications in controlled laboratory environments. Ambulatory uncontrolled environments, however, provide additional challenges requiring further modelling to overcome. This work empirically assesses several approaches utilising machine learning classifiers to detect stress using physiological data recorded in an ambulatory setting with self-reported stress annotations. A subset of the training portion SMILE dataset enables the evaluation of approaches before submission. The optimal stress detection approach achieves 90.77% classification accuracy, 91.24 F1-Score, 90.42 Sensitivity and 91.08 Specificity, utilising an ExtraTrees classifier and feature imputation methods. Meanwhile, accuracy on the challenge data is much lower at 59.23% (submission #54 from BEaTS-MTU, username ZacDair). The cause of the performance disparity is explored in this work.
Computers and Society,Machine Learning,Systems and Control
What problem does this paper attempt to address?
The paper aims to address the issue of detecting individual psychological stress through ECG (Electrocardiogram) and GSR (Galvanic Skin Response) data collected via mobile wearable devices. Specifically: 1. **Research Background**: Monitoring mental health is challenging, especially in timely and accurately detecting stress states. With the development of wearable technology, continuous collection of physiological data is possible, providing new perspectives for mental health and behavioral state analysis. However, automated analysis is key to handling large amounts of data. 2. **Target Problem**: The goal of the paper is to develop a method that can automatically identify stress levels using machine learning techniques based on ECG and GSR data collected in daily life. The main focus includes: - Selecting features that best reflect stress states for hourly and minute-level stress detection. - Addressing data loss issues caused by sensor detachment or movement. - Exploring the differences between training data and testing data to improve the model's generalization ability across different datasets. 3. **Main Contributions**: The paper proposes an efficient stress detection method using the ExtraTrees classifier and feature imputation techniques, achieving a high classification accuracy (90.77%) on the SMILE dataset, but significantly lower performance (59.23%) on challenge data. The study also found a covariate shift between the training set and the test set, indicating the need for further improvements to enhance the model's robustness and generalization ability.