Back

HypnoGPT: A Hypnogram Language Model for Sleep Staging Enhancement and Sleep Disorder Diagnosis

Yu, T.; Gu, Z.; Wang, F.; Huang, R.; Li, M.; Yu, J.-g.; Yu, Z.; Zhang, J.; Xu, Y.; Jiang, H.; Liu, W.; Deng, G.; Gao, Z.; Wu, Y.; Liu, J.; Zhang, Y.; Duan, S.; Li, Y.; Xiao, J.; Wu, W.

2024-10-28 psychiatry and clinical psychology
10.1101/2024.10.26.24316166 medRxiv
Show abstract

Accurate and scalable sleep assessment is crucial for diagnosing sleep disorders and advancing personalized medicine. However, current approaches heavily rely on manual scoring of polysomnography (PSG) recordings, which is labor-intensive, costly, and difficult to scale. Here, we introduce SleepGPT, the first sleep-specific language model trained on over 5.8 million sleep stage annotations from 5,793 whole-night recordings to model the sequential dynamics of sleep macrostructure. We demonstrate that SleepGPT consistently enhances the performance of multiple state-of-the-art sleep staging models across diverse independent datasets (total N=1,320,654), and enables high staging accuracy from low-density wearable EEG recordings (N=120,095). We further show that SleepGPT supports robust diagnosis of sleep disorders when embedded into a sequence-level Transformer framework. The model accurately identifies abnormal sleep stage dynamics and Type-1 narcolepsy across multiple clinical datasets (total N=685). Attention-based visualization of model outputs revealed interpretable macrostructural patterns associated with diagnostic labels. Together, these results demonstrate that SleepGPT serves as a general-purpose foundation model for sleep sequence modeling, offering a scalable and clinically translatable tool for automated sleep staging and sleep disorder diagnosis.

Matching journals

The top 5 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.