arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.16104stat.MLcs.LGeess.SPq-bio.NCq-bio.QM
收藏

用自监督学习揭示临床EEG信号的结构

Uncovering the structure of clinical EEG signals with self-supervised learning

Université Paris-Saclay(巴黎萨克雷大学);Inria(法国国家信息与自动化研究所);CEA(法国原子能和替代能源委员会);InteraXon Inc.(InteraXon公司);University of Helsinki(赫尔辛基大学);Max Planck Institute for Human Cognitive and Brain Sciences(马克斯·普朗克人类认知与脑科学研究所)

Hubert Banville, Omar Chehab, Aapo Hyvärinen, Denis-Alexander Engemann, Alexandre Gramfort

摘要:

Objective.监督学习范式通常受到可用标记数据量的限制。这种现象在临床相关数据中尤其成问题,例如脑电图(EEG),其中标记在专业知识和人工处理时间方面可能是昂贵的。因此,设计用于在EEG数据上学习的深度学习架构产生了相对较浅的模型和性能,最多与传统的基于特征的方法相似。然而,在大多数情况下,未标记的数据是丰富的。通过从这些未标记的数据中提取信息,尽管对标签的访问有限,但深度神经网络可能会达到有竞争力的性能。Approach.我们研究了自监督学习(SSL),一种很有前途的技术,用于发现未标记数据中的结构,学习EEG信号的表示。具体来说,我们探讨了两个任务的基础上的时间上下文预测以及对比预测编码的两个临床相关的问题:基于EEG的睡眠分期和病理检测。我们在两个大型公共数据集上进行了实验,这些数据集有数千个记录,并使用纯监督和手工设计的方法进行了基线比较。主要结果。在SSL学习特征上训练的线性分类器在低标签数据体系中始终优于纯监督深度神经网络,同时在所有标签可用时达到竞争性性能。此外,用每种方法学习的嵌入揭示了与生理和临床现象相关的清晰的潜在结构,例如年龄效应。意义我们证明了自我监督学习方法对EEG数据的好处。我们的研究结果表明,SSL可能为在EEG数据上更广泛地使用深度学习模型铺平道路。

英文摘要:

Objective. Supervised learning paradigms are often limited by the amount of labeled data that is available. This phenomenon is particularly problematic in clinically-relevant data, such as electroencephalography (EEG), where labeling can be costly in terms of specialized expertise and human processing time. Consequently, deep learning architectures designed to learn on EEG data have yielded relatively shallow models and performances at best similar to those of traditional feature-based approaches. However, in most situations, unlabeled data is available in abundance. By extracting information from this unlabeled data, it might be possible to reach competitive performance with deep neural networks despite limited access to labels. Approach. We investigated self-supervised learning (SSL), a promising technique for discovering structure in unlabeled data, to learn representations of EEG signals. Specifically, we explored two tasks based on temporal context prediction as well as contrastive predictive coding on two clinically-relevant problems: EEG-based sleep staging and pathology detection. We conducted experiments on two large public datasets with thousands of recordings and performed baseline comparisons with purely supervised and hand-engineered approaches. Main results. Linear classifiers trained on SSL-learned features consistently outperformed purely supervised deep neural networks in low-labeled data regimes while reaching competitive performance when all labels were available. Additionally, the embeddings learned with each method revealed clear latent structures related to physiological and clinical phenomena, such as age effects. Significance. We demonstrate the benefit of self-supervised learning approaches on EEG data. Our results suggest that SSL may pave the way to a wider use of deep learning models on EEG data.

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

↑