用协方差矩阵和Transformer自动分睡眠阶段,效果优于现有方法。
Automatic Classification of Sleep Stages from EEG Signals Using Riemannian Metrics and Transformer Networks
- 将脑区间协方差矩阵作为时序输入,用Transformer建模
- 在单数据集和多数据集上均超越当前最佳模型
- 适合需要处理协方差时序数据的分类任务
目的:在睡眠医学中,评估个体睡眠进程通常依赖于耗时的手动评分脑电图(EEG)信号。近年来,许多深度学习方法被提出以自动化该过程,主要通过从信号中提取特征。然而,尽管在脑机接口等邻近领域取得进展,脑区间协方差分析在睡眠分期任务中仍被低估。方法:在先前工作的基础上,我们研究了SPDTransNet——一种基于Transformer的网络,用于通过协方差矩阵时序对睡眠阶段进行分类。此外,我们提出一种新方法,将学习到的信号级特征融入协方差矩阵,同时保持其对称正定(SPD)性质。结果:通过与多种先进模型在优化类别性能的方法下比较,我们的模型在单数据集及尤其在多数据集实验中达到或超过现有最优水平。结论:本文证明了SPDTransNet在脑电睡眠分期中的能力,特别是其对多数据集任务的适应性,且该方法可轻松扩展至任何涉及协方差矩阵时序的分类任务。
原文摘要 · Abstract (English)
Purpose: In sleep medicine, assessing the evolution of a subject's sleep often involves the costly manual scoring of electroencephalographic (EEG) signals. In recent years, a number of Deep Learning approaches have been proposed to automate this process, mainly by extracting features from said signals. However, despite some promising developments in related problems, such as Brain-Computer Interfaces, analyses of the covariances between brain regions remain underutilized in sleep stage scoring.Methods: Expanding upon our previous work, we investigate the capabilities of SPDTransNet, a Transformer-derived network designed to classify sleep stages from EEG data through timeseries of covariance matrices. Furthermore, we present a novel way of integrating learned signal-wise features into said matrices without sacrificing their Symmetric Definite Positive (SPD) nature.Results: Through comparison with other State-of-the-Art models within a methodology optimized for class-wise performance, we achieve a level of performance at or beyond various State-of-the-Art models, both in single-dataset and - particularly - multi-dataset experiments.Conclusion: In this article, we prove the capabilities of our SPDTransNet model, particularly its adaptability to multi-dataset tasks, within the context of EEG sleep stage scoring - though it could easily be adapted to any classification task involving timeseries of covariance matrices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。