arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.00644eess.SPcs.LG
收藏

特征匹配作为可穿戴EEG的改进迁移学习技术

Feature matching as improved transfer learning technique for wearable EEG

UZ Leuven(鲁汶大学医院)

Elisabeth R. M. Heremans, Huy Phan, Amir H. Ansari, Pascal Borzée, Bertien Buyse, Dries Testelmans, Maarten De Vos

摘要:

目的:随着具有非常规电极配置的可穿戴睡眠监测设备的快速兴起,需要能够对具有少量标记数据的配置执行睡眠分期的自动算法。迁移学习具有将神经网络权重从源模态(例如,标准电极配置)适配到新的目标模态(例如,非常规电极配置)的能力。方法:我们提出了特征匹配,这是一种新的迁移学习策略,可以替代常用的微调方法。该方法包括用来自源模态的大量数据和源和目标模态的少量配对样本来训练模型。对于那些配对的样本,模型提取目标模态的特征,将这些特征与来自源模态的对应样本的特征进行匹配。结果如下:我们比较了三个不同目标域的特征匹配和微调,使用两种不同的神经网络架构,以及不同数量的训练数据。特别是在小的队列(即非传统记录设置中的2 - 5个标记记录)中,特征匹配系统地优于微调,对于不同的场景和数据集,准确度的平均相对差异范围为0.4%至4.7%。结论:我们的研究结果表明,特征匹配作为一种迁移学习方法优于微调,特别是在数据非常少的情况下。意义:因此,我们得出结论,特征匹配是一种很有前途的新方法,可穿戴睡眠分期与新设备。

英文摘要:

Objective: With the rapid rise of wearable sleep monitoring devices with non-conventional electrode configurations, there is a need for automated algorithms that can perform sleep staging on configurations with small amounts of labeled data. Transfer learning has the ability to adapt neural network weights from a source modality (e.g. standard electrode configuration) to a new target modality (e.g. non-conventional electrode configuration). Methods: We propose feature matching, a new transfer learning strategy as an alternative to the commonly used finetuning approach. This method consists of training a model with larger amounts of data from the source modality and few paired samples of source and target modality. For those paired samples, the model extracts features of the target modality, matching these to the features from the corresponding samples of the source modality. Results: We compare feature matching to finetuning for three different target domains, with two different neural network architectures, and with varying amounts of training data. Particularly on small cohorts (i.e. 2 - 5 labeled recordings in the non-conventional recording setting), feature matching systematically outperforms finetuning with mean relative differences in accuracy ranging from 0.4% to 4.7% for the different scenarios and datasets. Conclusion: Our findings suggest that feature matching outperforms finetuning as a transfer learning approach, especially in very low data regimes. Significance: As such, we conclude that feature matching is a promising new method for wearable sleep staging with novel devices.

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

↑