期刊
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS
卷 26, 期 10, 页码 4948-4956出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/JBHI.2022.3157262
关键词
Sleep; Electroencephalography; Hidden Markov models; Feature extraction; Convolution; Brain modeling; Kernel; Electroencephalogram; sleep staging; convolutional neural network; attention mechanism; hidden Markov model
类别
资金
- Zhejiang Gongshang University, Zhejiang Provincial Key Laboratory of New Network Standards and Technologies [2013E10012]
This study presents an automatic sleep staging model with an improved attention module and hidden Markov model (HMM). The model achieves good performance in feature extraction and classification using single-channel electroencephalogram data.
Sleep staging is an important step in analyzing sleep quality. Traditional manual analysis by psychologists is time-consuming. In this paper, we propose an automatic sleep staging model with an improved attention module and hidden Markov model (HMM). The model is driven by single-channel electroencephalogram (EEG) data. It automatically extracts features through two convolution kernels with different scales. Subsequently, an improved attention module based on Squeeze-and-Excitation Networks (SENet) will perform feature fusion. The neural network will give a preliminary sleep stage based on the learned features. Finally, an HMM will apply sleep transition rules to refine the classification. The proposed method is tested on the sleep-EDFx dataset and achieves excellent performance. The accuracy on the Fpz-Cz channel is 84.6%, and the kappa coefficient is 0.79. For the Pz-Oz channel, the accuracy is 82.3% and kappa is 0.76. The experimental results show that the attention mechanism plays a positive role in feature fusion. And our improved attention module improves the classification performance. In addition, applying sleep transition rules through HMM helps to improve performance, especially N1, which is difficult to identify.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据