[1]CHEN Lingling,BI Xiaojun.Sleep staging model based on multimodal fusion[J].CAAI Transactions on Intelligent Systems,2022,17(6):1194-1200.[doi:10.11992/tis.202202018]
Copy
CAAI Transactions on Intelligent Systems[ISSN 1673-4785/CN 23-1538/TP] Volume:
17
Number of periods:
2022 6
Page number:
1194-1200
Column:
学术论文—智能系统
Public date:
2022-11-05
- Title:
-
Sleep staging model based on multimodal fusion
- Author(s):
-
CHEN Lingling1; BI Xiaojun2
-
1. College of Information and Communication Engineering, Harbin Engineering University, Harbin 150001, China;
2. School of Information Engineering, Minzu University of China, Beijing 100081, China
-
- Keywords:
-
deep learning; sleep stage; channel attention mechanism; polysomnogram; multimodal; bidirectional long short-term memory; sleep-European data format dataset; residual shrinkage network
- CLC:
-
TP18; TH79
- DOI:
-
10.11992/tis.202202018
- Abstract:
-
To solve the problem of insufficient feature use due to the differences in modal information in sleep polygraphs at different stages, this paper proposes a sleep staging method based on the channel attention mechanism and multimodal gating module. First, a residual shrinkage network is used to design each modal feature extraction network to extract the features of all modules, which are spliced and fused in the channel dimension, and the fused features are further recalibrated using the channel attention mechanism to obtain time-invariant polysomnography features. Then, a multimodal gating module based on an adaptive gating mechanism is proposed, which weighs and fuses all modal and time-invariant features according to their importance, thereby realizing the organic fusion of features. Finally, the time sequence characteristics of polysomnography are obtained using a bidirectional long short-term memory network. The experimental result shows that the accuracy of the sleep staging model proposed in this paper is 87.6% on the Sleep-EDF dataset, with an $M_{F_1} $ of 82.0%, thus achieving the best staging effect to date.