用分层掩码与提示学习,少采数据也能准判睡眠阶段。
Resource Efficient Sleep Staging via Multi-Level Masking and Prompt Learning
- 分层掩码让模型在不完整数据上仍能学好特征。
- 在极少量数据下仍达顶尖性能,最低仅需1/8原始数据。
- 适合可穿戴设备等资源受限的睡眠监测场景。
自动睡眠分期对评估睡眠质量与诊断睡眠障碍至关重要。现有方法通常依赖长时连续的脑电(EEG)记录,在可穿戴或家庭监测等资源受限系统中面临数据采集难题。本文提出资源高效睡眠分期任务,旨在减少每个睡眠周期所需采集的信号量,同时保持可靠分类性能。为此,我们采用掩码与提示学习策略,提出新型框架Mask-Aware Sleep Staging(MASS)。设计多层级掩码策略,以促进在部分且不规则观测下的有效特征建模;为缓解掩码带来的上下文信息损失,进一步提出分层提示学习机制,将未被掩码的数据聚合为全局提示,作为指导局部与整体特征建模的语义锚点。MASS在四个数据集上评估,表现优于现有方法,尤其在数据极度有限时优势显著,展现了其在真实低资源睡眠监测环境中高效可扩展部署的潜力。
原文摘要 · Abstract (English)
Automatic sleep staging plays a vital role in assessing sleep quality and diagnosing sleep disorders. Most existing methods rely heavily on long and continuous EEG recordings, which poses significant challenges for data acquisition in resource-constrained systems, such as wearable or home-based monitoring systems. In this paper, we propose the task of resource-efficient sleep staging, which aims to reduce the amount of signal collected per sleep epoch while maintaining reliable classification performance. To solve this task, we adopt the masking and prompt learning strategy and propose a novel framework called Mask-Aware Sleep Staging (MASS). Specifically, we design a multi-level masking strategy to promote effective feature modeling under partial and irregular observations. To mitigate the loss of contextual information introduced by masking, we further propose a hierarchical prompt learning mechanism that aggregates unmasked data into a global prompt, serving as a semantic anchor for guiding both patch-level and epoch-level feature modeling. MASS is evaluated on four datasets, demonstrating state-of-the-art performance, especially when the amount of data is very limited. This result highlights its potential for efficient and scalable deployment in real-world low-resource sleep monitoring environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。