arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2312.09623eess.SPcs.LG
收藏

一种用于睡眠阶段分类的新型双流时频对比前置任务框架

A novel dual-stream time-frequency contrastive pretext tasks framework for sleep stage classification

Utrecht University(乌得勒支大学)

Sergio Kazatzidis, Siamak Mehrkanoon

AI总结:

针对EEG领域标注数据不足的问题,提出一种双流时频对比前置任务框架,将新型FS任务与RP、TS任务结合,在PC18数据集的睡眠阶段分类任务上取得准确率提升,且学习到的表示具有可分性。

摘要:

自监督学习解决了许多有监督方法面临的难题,即需要大量标注数据。这一挑战在脑电图(EEG)研究等领域尤为突出。自监督学习则通过利用由前置任务生成的伪标签来工作,以获取丰富且有意义的数据表示。在本研究中,我们旨在引入一种在时域和频域同时运行的双流前置任务架构。具体而言,我们研究了将新型的Frequency Similarity(FS,频率相似度)前置任务整合到两个现有前置任务——Relative Positioning(RP,相对定位)和Temporal Shuffling(TS,时间打乱)——中的效果。我们使用Physionet Challenge 2018(PC18)数据集,在下游任务睡眠阶段分类的场景下评估了这些模型的准确率。加入FS任务后,下游任务准确率得到了显著提升:在RP基础上提升了1.28%,在TS基础上提升了2.02%。此外,使用Uniform Manifold Approximation and Projection(UMAP,均匀流形近似与投影)对学习到的嵌入进行可视化时,出现了清晰的聚类,表明学习到的表示承载了有意义的信息。

英文摘要:

Self-supervised learning addresses the challenge encountered by many supervised methods, i.e. the requirement of large amounts of annotated data. This challenge is particularly pronounced in fields such as the electroencephalography (EEG) research domain. Self-supervised learning operates instead by utilizing pseudo-labels, which are generated by pretext tasks, to obtain a rich and meaningful data representation. In this study, we aim at introducing a dual-stream pretext task architecture that operates both in the time and frequency domains. In particular, we have examined the incorporation of the novel Frequency Similarity (FS) pretext task into two existing pretext tasks, Relative Positioning (RP) and Temporal Shuffling (TS). We assess the accuracy of these models using the Physionet Challenge 2018 (PC18) dataset in the context of the downstream task sleep stage classification. The inclusion of FS resulted in a notable improvement in downstream task accuracy, with a 1.28 percent improvement on RP and a 2.02 percent improvement on TS. Furthermore, when visualizing the learned embeddings using Uniform Manifold Approximation and Projection (UMAP), distinct clusters emerge, indicating that the learned representations carry meaningful information.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑