Prediction of Sleep Stages Based on Wearable Signals Using Machine Learning Techniques

Rodrigo Braga, Daniel Osório, Daniel Osório, Hugo Gamboa, Hugo Gamboa

2023

Abstract

Sleep’s impact on mood and health is widely recognized by medical researchers with such understanding disseminating among average people in recent years. The main objective of this work was the development of machine learning algorithms for automatic sleep cycles detection. The features were selected based on the AASM manual, which is considered the gold standard for human technicians. For training the models we used MESA, a database containing 2056 full overnight unattended polysomnographies. With the goal of developing an algorithm that would only require a photoplethysmography (PPG) device to be able to accurately predict sleep stages and quality, the main channels used from this dataset were peripheral oxygen saturation and PPG. Testing the performance of Random forest, Gradient Boosting, Gaussian Naive-bayes, K-Nearest Neighbours, Support Vector Machine and Multilayer Perceptron classifiers, and using features extracted from the dataset, we achieved 80.50 % accuracy, 0.7586 Cohen’s kappa, and 77.38% F1-score, for five sleep stages, using a Multilayer Perceptron. To assess its performance in a real-world scenario we acquired sleep data and compared the classifications attributed by a popular sleep stage classification android app and our algorithm, resulting in a strong level of agreement (90.96% agreement, 0.8663 Cohen’s kappa), for four sleep stages.

Download


Paper Citation


in Harvard Style

Braga R., Osório D. and Gamboa H. (2023). Prediction of Sleep Stages Based on Wearable Signals Using Machine Learning Techniques. In Proceedings of the 16th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2023) - Volume 4: BIOSIGNALS; ISBN 978-989-758-631-6, SciTePress, pages 195-202. DOI: 10.5220/0011655200003414


in Bibtex Style

@conference{biosignals23,
author={Rodrigo Braga and Daniel Osório and Hugo Gamboa},
title={Prediction of Sleep Stages Based on Wearable Signals Using Machine Learning Techniques},
booktitle={Proceedings of the 16th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2023) - Volume 4: BIOSIGNALS},
year={2023},
pages={195-202},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0011655200003414},
isbn={978-989-758-631-6},
}


in EndNote Style

TY - CONF

JO - Proceedings of the 16th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2023) - Volume 4: BIOSIGNALS
TI - Prediction of Sleep Stages Based on Wearable Signals Using Machine Learning Techniques
SN - 978-989-758-631-6
AU - Braga R.
AU - Osório D.
AU - Gamboa H.
PY - 2023
SP - 195
EP - 202
DO - 10.5220/0011655200003414
PB - SciTePress