Saved in:
Bibliographic Details
Main Authors: Ai, Lejun, Li, Yulong, Yi, Haodong, Xie, Jixuan, Wang, Yue, Liu, Jia, Chen, Min, Wang, Rui
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.06785
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918208014909440
author Ai, Lejun
Li, Yulong
Yi, Haodong
Xie, Jixuan
Wang, Yue
Liu, Jia
Chen, Min
Wang, Rui
author_facet Ai, Lejun
Li, Yulong
Yi, Haodong
Xie, Jixuan
Wang, Yue
Liu, Jia
Chen, Min
Wang, Rui
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2511_06785
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Resource Efficient Sleep Staging via Multi-Level Masking and Prompt Learning
Ai, Lejun
Li, Yulong
Yi, Haodong
Xie, Jixuan
Wang, Yue
Liu, Jia
Chen, Min
Wang, Rui
Machine Learning
Artificial Intelligence
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.
title Resource Efficient Sleep Staging via Multi-Level Masking and Prompt Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2511.06785