Self-Supervised Transformer Model Training for a Sleep-EEG Foundation Model. 2024

Mattson Ogg, and William G Coon

The American Academy of Sleep Medicine (AASM) recognizes five sleep/wake states (Wake, N1, N2, N3, REM), yet this classification schema provides only a high-level summary of sleep and likely overlooks important neurological or health information. New, data-driven approaches are needed to more deeply probe the information content of sleep signals. Here we present a self-supervised approach that learns the structure embedded in large quantities of neurophysiological sleep data. This masked transformer training procedure is inspired by high performing self-supervised methods developed for speech transcription. We show that self-supervised pre-training matches or outperforms supervised sleep stage classification, especially when labeled data or compute-power is limited. Perhaps more importantly, we also show that our pretrained model is flexible and can be fine-tuned to perform well on new tasks including distinguishing individuals and quantifying "brain age" (a potential health biomarker). This suggests that modern methods can automatically learn information that is potentially overlooked by the 5-class sleep staging schema, laying the groundwork for new schemas and further data-driven exploration of sleep.

UI MeSH Term Description Entries

Related Publications

Mattson Ogg, and William G Coon
February 2023, Journal of the American Chemical Society,
Mattson Ogg, and William G Coon
December 2023, IEEE transactions on pattern analysis and machine intelligence,
Mattson Ogg, and William G Coon
January 2024, IEEE transactions on pattern analysis and machine intelligence,
Mattson Ogg, and William G Coon
November 2023, IEEE transactions on pattern analysis and machine intelligence,
Mattson Ogg, and William G Coon
December 2023, Nature computational science,
Mattson Ogg, and William G Coon
July 2022, IEEE journal of biomedical and health informatics,
Mattson Ogg, and William G Coon
January 2024, IEEE transactions on neural systems and rehabilitation engineering : a publication of the IEEE Engineering in Medicine and Biology Society,
Mattson Ogg, and William G Coon
March 2024, IEEE transactions on pattern analysis and machine intelligence,
Mattson Ogg, and William G Coon
May 2022, Mathematical biosciences and engineering : MBE,
Copied contents to your clipboard!