面向睡眠阶段分类的自适应时空图卷积神经网络

最新推荐文章于 2023-12-31 11:34:29 发布

PaperWeekly

最新推荐文章于 2023-12-31 11:34:29 发布

阅读量3.2k

点赞数 2

文章标签：网络神经网络大数据计算机视觉机器学习

本文链接：https://blog.csdn.net/c9Yv2cf9I06K2A9E/article/details/110729604

版权

本文介绍了GraphSleepNet，一种应用于睡眠阶段分类的自适应时空图卷积神经网络，旨在解决睡眠分期领域的挑战，如网格数据局限性、大脑连接关系建模及睡眠过渡规则利用。该模型通过自适应睡眠图学习和时空图卷积，结合注意力机制捕获空间和时间特征，实现了睡眠阶段的高精度分类。在Montreal Archive of Sleep Studies数据集上的实验表明，GraphSleepNet超越了传统机器学习和深度学习基线方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

©PaperWeekly 原创 · 作者｜宁晓军

学校｜北京交通大学硕士生

研究方向｜时间序列分析与挖掘；图神经网络

本文介绍一篇北京交通大学网络科学与智能系统研究所贾子钰博士等人，于 IJCAI 2020 发表的论文《GraphSleepNet: Adaptive Spatial-Temporal Graph Convolutional Networks for Sleep Stage Classification》，该研究提出一种多变量时间序列分类的通用图神经网络框架并首次应用于睡眠阶段分类。

论文标题：

GraphSleepNet: Adaptive Spatial-Temporal Graph Convolutional Networks for Sleep Stage Classification

论文链接：

https://www.ijcai.org/Proceedings/2020/0184.pdf

代码链接：

https://github.com/ziyujia/GraphSleepNet

论文作者主页：

https://ziyujia.github.io/

背景简介

睡眠阶段分类对睡眠质量的评估和疾病的诊断具有重要意义，睡眠专家通常根据睡眠分期标准和多导睡眠图（polysomnography, PSG）来判定睡眠状态。在目前的睡眠分期领域研究中，睡眠专家通常使用 R&K 睡眠分期标准和美国睡眠医学会（AASM）制定的睡眠分期标准进行睡眠阶段的识别。

特别是在 AASM 标准中还记录了不同睡眠阶段之间的转换规则，帮助睡眠专家通过时间上下文进行睡眠分期。虽然有标准和规则提供决策支持，但是由睡眠专家人工进行睡眠分期仍然是一项繁琐且耗时的任务，分期结果也容易受到睡眠专家主观意识的影响。

动机

2.1 挑战

2.1.1 网格数据的局限性

目前大多数睡眠分期模型大多使用 CNNs 和 RNNs 模型，尽管 CNNs 和 RNNs 能够实现较高的睡眠分期准确率，但它们的局限性在于模型的输入必须是网格数据（例如 2D 图像表示），这导致了大脑区域之间的联系被忽略。由于大脑区域处于非欧式空间，因此图应当是表示大脑连接性最自然且最适合的数据结构。

2.1.2 大脑连接关系的建模

基于图卷积神经网络在图结构数据中的成功应用，我们采用图结构表示方法研究睡眠分期问题，在多导 EEG 数据中，每个 EEG 通道对应于睡眠图中的一个节点，两个节点之间的连接对应于睡眠图中存在的边。

固定的图结构是图卷积神经网络中的重要输入，也是提取空间信息的关键。但由于人类对人脑的认知是有限的，因此为睡眠分期预定义合适的空间脑连接结构仍然是一个挑战。因此，论文尝试提出一种数据驱动的大脑连接结构学习模块与与时空图卷积集成于统一的图神经网络框架。

2.1.3 睡眠过渡规则

如何利用相邻睡眠阶段之间的过渡规则也是一个挑战。睡眠专家在进行当前睡眠阶段的分期时，往往会结合其相邻睡眠阶段。因此，充分利用睡眠阶段间的过渡规则往往能够提升睡眠分期的准确率。

2.2 贡献

据我们所知，这是首次尝试将 ST-GCN 应用于睡眠分期领域。此外，我们提出了一种新的自适应睡眠图学习机制，它与 ST-GCN 同时被集成在统一的深度神经网络的架构中。
我们设计了一种基于注意力机制的时空卷积结构，它可以有效的捕获不同睡眠阶段的空间特征和睡眠阶段之间的转换规则。
在基准数据集的实验结果表明，提出的 GraphSleepNet 优于传统的基线方法，取得了 SOTA 的性能。