无线电工程2024，Vol.54Issue(8)：2030-2039,10.DOI:10.3969/j.issn.1003-3106.2024.08.022
基于单通道上下文编码的轻量化睡眠分期模型
A Lightweight Sleep Staging Model Based on Single-channel Context Encoding
摘要
Abstract
Utilizing the frequency relationships between different signals in multimodal signals to handle sleep staging tasks has become the mainstream of today.However,the difficulty of multimodal signal acquisition limits further development of sleep staging tasks.In addition,the unbalanced classification of the data set and the large magnitude of parameters are also important reasons that limit the development of sleep staging tasks.In response to the above problems,a fully convolutional network model based on single-channel Electroencephalogram(EEG)data is proposed.The waveform information of EEG signals is learned through the U-shaped structure and feature fusion module.The parallel multi-scale time features are used to instead the recurrent networks to extract time series information.The bottleneck structure is used to reduce the amount of parameters,and the focal loss function is used to reduce the impact of uneven distribution between data categories on classification accuracy.Experimental results on the public data set Sleep-EDF show that the proposed model achieves a classification accuracy of 87.5％for single-channel EEG data,alleviating the problem of imbalanced data set.It is worth mentioning that the number of parameters of the proposed method is only 2.86％of which of the Deepsleep network.关键词
脑电信号/睡眠分期/空洞卷积/类别失衡/焦变函数Key words
EEG signal/sleep staging/dilated convolution/class imbalance/focal loss function分类
信息技术与安全科学引用本文复制引用
仝爽,王琪,孙久淞..基于单通道上下文编码的轻量化睡眠分期模型[J].无线电工程,2024,54(8):2030-2039,10.基金项目
江苏省研究生培养创新工程项目(SJCX23_0373)Postgraduate Training Innovation Engineering Project of Jiangsu Province(SJCX23_0373) （SJCX23_0373）
