Search papers, labs, and topics across Lattice.
This paper introduces GamSleepNet, a lightweight automatic sleep staging framework designed for single-channel EEG data, addressing challenges like overfitting and low accuracy in classifying difficult sleep stages. By employing a novel FEB module that integrates improved Gabor kernels and learnable filters, along with a Mamba architecture for temporal classification, the framework achieves state-of-the-art performance with only 30.86 thousand parameters. Experimental results on the Sleepedf dataset demonstrate an overall accuracy of 87.86%, significantly enhancing the identification of challenging sleep stages such as N1 and REM.
GamSleepNet achieves 87.86% accuracy in sleep staging with just 30.86K parameters, setting a new standard for lightweight EEG models.
Automatic sleep staging is a key technology for precise diagnosis and treatment of sleep disorders as well as long-term home sleep monitoring. Portable electroencephalogram (EEG) devices have become the focus of research due to their convenience in data collection. However, current methods still face three major challenges: large parameter sizes that easily lead to overfitting on small datasets, low accuracy in classifying difficult stages such as N1 and REM, unclear optimal training dataset size, and difficulty in deployment. This paper proposes GamSleepNet, a lightweight and low-latency automatic sleep staging framework for single-channel EEG. The framework features the FEB module, which combines improved Gabor kernels with learnable filters for feature extraction, uses the Mamba architecture to build a temporal classification network, introduces a novel contrastive loss and a two-stage training strategy, and experimentally validates the optimal dataset size for single-channel EEG sleep staging models. On the Sleepedf dataset, this model achieves an overall accuracy of 87.86 percent with only 30.86 thousand parameters, with all metrics reaching SOTA levels and significantly improving the identification accuracy of challenging sleep stages.
