Optimizing wearable single-channel EEG sleep staging in a heterogeneous sleep-disordered population using transfer learning

Jaap F van der Aar,Merel M van Gilst,Daan A van den Ende,Pedro Fonseca,Bregje N J van Wetten,Hennie C J P Janssen,Elisabetta Peri,Sebastiaan Overeem
DOI: https://doi.org/10.5664/jcsm.11380
2024-09-30
Abstract:Study objectives: While various wearable EEG devices have been developed, performance evaluation of the devices and their associated automated sleep stage classification models is mostly limited to healthy subjects. A major barrier for applying automated wearable EEG sleep staging in clinical populations is the need for large-scale data for model training. We therefore investigated transfer learning as strategy to overcome limited data availability and optimize automated single-channel EEG sleep staging in people with sleep disorders. Methods: We acquired 52 single-channel frontopolar headband EEG recordings from a heterogeneous sleep-disordered population with concurrent PSG. We compared three model training strategies: 'pre-training' (i.e., training on a larger dataset of 901 conventional PSGs), 'training-from-scratch' (i.e., training on wearable headband recordings), and 'fine-tuning' (i.e., training on conventional PSGs, followed by training on headband recordings). Performance was evaluated on all headband recordings using 10-fold cross-validation. Results: Highest performance for 5-stage classification was achieved with fine-tuning (κ = .778), significantly higher than with pre-training (κ = .769) and with training-from-scratch (κ = .733). No significant differences or systematic biases were observed with clinically relevant sleep parameters derived from PSG. All sleep disorder categories showed comparable performance. Conclusions: This study emphasizes the importance of leveraging larger available datasets through deep transfer learning to optimize performance with limited data availability. Findings indicate strong similarity in data characteristics between conventional PSG and headband recordings. Altogether, results suggest the combination of the headband, classification model, and training methodology can be viable for sleep monitoring in the heterogeneous clinical population.
What problem does this paper attempt to address?