模式识别(心理学)
脑电图
深度学习
卷积神经网络
特征提取
注意力网络
作者
Ziyu Jia,Youfang Lin,Jing Wang,Xuehui Wang,Peiyi Xie,Yingbin Zhang
出处
期刊:International Joint Conference on Artificial Intelligence
日期:2021-08-09
卷期号:3: 2614-2620
被引量:2
标识
DOI:10.24963/ijcai.2021/360
摘要
Sleep staging is fundamental for sleep assessment and disease diagnosis. Although previous attempts to classify sleep stages have achieved high classification performance, several challenges remain open: 1) How to effectively extract salient waves in multimodal sleep data; 2) How to capture the multi-scale transition rules among sleep stages; 3) How to adaptively seize the key role of specific modality for sleep staging. To address these challenges, we propose SalientSleepNet, a multimodal salient wave detection network for sleep staging. Specifically, SalientSleepNet is a temporal fully convolutional network based on the $\rm U^2$-Net architecture that is originally proposed for salient object detection in computer vision. It is mainly composed of two independent $\rm U^2$-like streams to extract the salient features from multimodal data, respectively. Meanwhile, the multi-scale extraction module is designed to capture multi-scale transition rules among sleep stages. Besides, the multimodal attention module is proposed to adaptively capture valuable information from multimodal data for the specific sleep stage. Experiments on the two datasets demonstrate that SalientSleepNet outperforms the state-of-the-art baselines. It is worth noting that this model has the least amount of parameters compared with the existing deep neural network models.
科研通智能强力驱动
Strongly Powered by AbleSci AI