PedSleepMAE: Generative Model for Multimodal Pediatric Sleep Signals

Saurav Raj Pandey, Aaqib Saeed, Harlin Lee

Keywords: sleep, pediatric health, polysomnography, eeg, apnea, generative AI, masked autoencoder, Prader-Willi syndrome

TL;DR: Generative AI for pediatric sleep signals including EEGs and respiratory signals

Abstract: Pediatric sleep is an important but often overlooked area in health informatics. We present PedSleepMAE, a generative model that fully leverages multimodal pediatric sleep signals including multichannel EEGs, respiratory signals, EOGs and EMG. This masked autoencoder-based model performs comparably to supervised learning models in sleep scoring and in the detection of apnea, hypopnea, EEG arousal and oxygen desaturation. Its embeddings are also shown to capture subtle differences in sleep signals coming from a rare genetic disorder. Furthermore, PedSleepMAE generates realistic signals that can be used for sleep segment retrieval, outlier detection, and missing channel imputation. This is the first general-purpose generative model trained on multiple types of pediatric sleep signals.

Track: 5. Biomedical generative AI

Registration Id: PVNZKSJ84DQ

Submission Number: 196