Post by Yuzhe Yang
AI Prof @ UCLA | Scientist @ Google | PhD @ MIT
Meet ๐ข๐ฆ๐ โ a fully open benchmark and state-of-the-art family of sleep foundation models. ๐ Recent sleep FMs have shown strong promise, but one basic question remains open: which ๐ฑ๐ณ๐ฆ-๐ต๐ณ๐ข๐ช๐ฏ๐ช๐ฏ๐จ and ๐ด๐ค๐ข๐ญ๐ช๐ฏ๐จ choices really improve generalization in real-world settings? With ๐ข๐ฆ๐, we study this directly under ๐ค๐ฐ๐ฉ๐ฐ๐ณ๐ต ๐ด๐ฉ๐ช๐ง๐ต and ๐ฎ๐ช๐ด๐ด๐ช๐ฏ๐จ-๐ค๐ฉ๐ข๐ฏ๐ฏ๐ฆ๐ญ ๐ช๐ฏ๐ง๐ฆ๐ณ๐ฆ๐ฏ๐ค๐ฆ, two major challenges for deployable sleep AI. ๐ฅ To make this possible, we built ๐ฆ๐น๐ฒ๐ฒ๐ฝ๐๐ฒ๐ป๐ฐ๐ต, a fully open benchmark aggregated from public resources with: โฑ๏ธ 166,500 hours of sleep recordings ๐งโ๐คโ๐ง 21,000+ sleep studies ๐ 9 public datasets ๐พ ~20M 30-second epochs. Across major self-supervised learning families, we identify the design choices that consistently matter for sleep FM pre-training. Our findings show that missing-channel inference can cause major drops for existing sleep FMs, but also that the right pre-training recipe can greatly improve robustness. We further find that scaling does help โ in ๐ฆ pre-training data, ๐ง model size, and ๐ multi-source data mixture โ but only when paired with the right SSL design. Guided by these insights, we build ๐ข๐ฆ๐, which beats state-of-the-arts across diverse downstream sleep and health tasks. ๐ ๐ข๐ฆ๐ offers a practical recipe for building more generalizable and deployable sleep AI! ๐ ๐ Paper: https://lnkd.in/gnYewCn7 ๐ Website: https://lnkd.in/gC8rAtmA ๐ป Code: https://lnkd.in/g4NZmDEf ๐ค Models: https://lnkd.in/gwatigQ7 Great work led by Zitao Shuai, Zongzhe Xu, David Yang, with collaborator Wei Wang! UCLA UCLA Computer Science Computational Medicine Department UCLA Henry Samueli School of Engineering and Applied Science #AI #Sleep #HealthAI #MultimodalAI #FoundationModels