软件导刊2024，Vol.23Issue(5)：24-32,9.DOI:10.11907/rjdk.232271
基于自注意力机制与双向LSTM的睡眠分期模型
A Sleep Staging Model Based on Self-Attention Mechanism and Bi-Directional LSTM
曹科研 1王莹莹 1陶杭波1
作者信息
- 1. 沈阳建筑大学 计算机科学与工程学院,辽宁 沈阳 110168
- 折叠
摘要
Abstract
A sleep staging model based on self attention mechanism and bidirectional long short-term memory network is proposed to address the problem that existing models cannot fully capture transient and random waveforms in samples,as well as cannot focus on typical and impor-tant waveforms,which affects staging results.Firstly,a single stream time-frequency information learning module is constructed to automati-cally express the low-level representations of PSG signals,and to mine the time-invariant information and frequency features of EEG data.Then,design an adaptive feature recalibration learning module to calibrate and train the instantaneous and key waveform features that appear in the 30 second sample,giving more attention to these features and assigning greater weights.Finally,the features are sent to the sequence dependency learning module between the associated samples to learn the contextual relationships between each sleep sample,fully utilizing ad-jacent samples before and after to determine the current sample category.The results show that this method performs better than other main-stream models,with accuracy rates of 85.5%and 84.3%on the Sleep-edf-2013 and Sleep-edf-2018 public sleep datasets,and MF1 values of 82.1%and 79.6%,respectively,which can provide technical reference for sleep staging tasks.关键词
睡眠分期/Transformer/双向LSTM/脑电信号/单通道Key words
sleep staging/Transformer/bi-directional LSTM/electroencephalogram/single channel分类
信息技术与安全科学引用本文复制引用
曹科研,王莹莹,陶杭波..基于自注意力机制与双向LSTM的睡眠分期模型[J].软件导刊,2024,23(5):24-32,9.
