From Laboratory to Everyday Life: Personalized Stress Prediction via Smartwatches

Published: 17 Jun 2024, Last Modified: 16 Jul 2024ML4LMS PosterEveryoneRevisionsBibTeXCC BY-SA 4.0
Keywords: Stress Prediction, Machine Learning, Preventive Healtcare
TL;DR: This paper introduces MUSTP, a machine learning pipeline that predicts stress using heart rate and ECG data from smartwatches, achieving higher F1 score through model transfer and personalization in everday setting.
Abstract: Accurate prediction of stress in everyday life is essential to prevent chronic stress and maintain health and well-being through early and personalized intervention. With the goal of enabling reliable prediction suitable for everyday life, we present MuStP, a two-stage machine learning pipeline designed to predict stress from low-resolution heart rate (HR) and high-resolution electrocardiography (ECG) measurements from commercial smartwatches. Our model is pre-trained with labeled data collected in a controlled laboratory stress study. Subsequently, we transfer the model for everyday use, enabling it to operate with everyday smartwatch data in various environments. The model transfer strategy effectively addresses the domain shift from laboratory data to highly imbalanced smartwatch data and allows personalization. The empirical results on smartwatch data show that MuStP can predict stress everyday with an F1 score of $0.52$, despite the measurements having sparse labels for stress.
Poster: pdf
Submission Number: 33
Loading