提出新型混合模型与频域掩码自编码,提升小样本生物信号迁移学习性能。
CiTrus: Squeezing Extra Performance out of Low-data Bio-signal Transfer Learning
- 设计卷积-变压器混合架构,结合频域掩码自编码预训练
- 在低数据量场景下,模型性能超越现有方法,尤其在极端数据量时表现最优
- 适用于小样本生物信号分类任务,对多模态数据有显著增益
针对小样本生物信号迁移学习,本文提出一种新的卷积-变压器混合模型架构,并引入基于频率的掩码自编码预训练任务。通过更全面的评估框架,研究了预训练对下游任务性能的提升效果及其在多模态场景下的作用。提出一种有效方法,将不同时间长度和采样率的下游数据对齐至预训练数据。实验表明,该模型在部分低数据下游任务上达到当前最佳性能;完整模型进一步提升效果。在基于变压器的模型中,预训练显著改善下游性能,多模态预训练带来额外增益,且本方法在最低与最高数据量条件下平均表现最优。
原文摘要 · Abstract (English)
Transfer learning for bio-signals has recently become an important technique to improve prediction performance on downstream tasks with small bio-signal datasets. Recent works have shown that pre-training a neural network model on a large dataset (e.g. EEG) with a self-supervised task, replacing the self-supervised head with a linear classification head, and fine-tuning the model on different downstream bio-signal datasets (e.g., EMG or ECG) can dramatically improve the performance on those datasets. In this paper, we propose a new convolution-transformer hybrid model architecture with masked auto-encoding for low-data bio-signal transfer learning, introduce a frequency-based masked auto-encoding task, employ a more comprehensive evaluation framework, and evaluate how much and when (multimodal) pre-training improves fine-tuning performance. We also introduce a dramatically more performant method of aligning a downstream dataset with a different temporal length and sampling rate to the original pre-training dataset. Our findings indicate that the convolution-only part of our hybrid model can achieve state-of-the-art performance on some low-data downstream tasks. The performance is often improved even further with our full model. In the case of transformer-based models we find that pre-training especially improves performance on downstream datasets, multimodal pre-training often increases those gains further, and our frequency-based pre-training performs the best on average for the lowest and highest data regimes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。