A deep transfer learning approach for wearable sleep stage classification with photoplethysmography

Mustafa G. Radha, Pedro Fonseca, Arnaud Moreau, Marco Ross, Andreas Cerny, Peter Anderer, Xi Long (Corresponding author), Ronald M. Aarts

Research output: Contribution to journalArticleAcademicpeer-review

6 Citations (Scopus)
25 Downloads (Pure)

Abstract

Unobtrusive home sleep monitoring using wrist-worn wearable photoplethysmography (PPG) could open the way for better sleep disorder screening and health monitoring. However, PPG is rarely included in large sleep studies with gold-standard sleep annotation from polysomnography. Therefore, training data-intensive state-of-the-art deep neural networks is challenging. In this work a deep recurrent neural network is first trained using a large sleep data set with electrocardiogram (ECG) data (292 participants, 584 recordings) to perform 4-class sleep stage classification (wake, rapid-eye-movement, N1/N2, and N3). A small part of its weights is adapted to a smaller, newer PPG data set (60 healthy participants, 101 recordings) through three variations of transfer learning. Best results (Cohen’s kappa of 0.65 ± 0.11, accuracy of 76.36 ± 7.57%) were achieved with the domain and decision combined transfer learning strategy, significantly outperforming the PPG-trained and ECG-trained baselines. This performance for PPG-based 4-class sleep stage classification is unprecedented in literature, bringing home sleep stage monitoring closer to clinical use. The work demonstrates the merit of transfer learning in developing reliable methods for new sensor technologies by reusing similar, older non-wearable data sets. Further study should evaluate our approach in patients with sleep disorders such as insomnia and sleep apnoea.
Original languageEnglish
Article number135
Number of pages11
JournalNPJ Digital Medicine
Volume4
DOIs
Publication statusPublished - 15 Sep 2021

Keywords

  • PPG
  • Sleep stages
  • Transfer learning
  • Wearable
  • LSTM
  • ECG

Fingerprint

Dive into the research topics of 'A deep transfer learning approach for wearable sleep stage classification with photoplethysmography'. Together they form a unique fingerprint.

Cite this