TOCOL:通过 Token 级对比学习改进预训练语言模型的上下文表示
TOCOL: Improving Contextual Representation of Pre-trained Language Models via Token-Level Contrastive Learning
基于摘要分析。本文研究 BERT 类预训练语言模型中,词表示空间的各向异性如何影响注意力与上下文编码,并提出 token 级对比学习框架 TOCOL,将自监督目标融入注意力机制,以重塑词表示空间并促进句子全局语义建模。 作者提出的机制解释是:BERT 的最大似然预训练目标可能使高频 token 在表示空间中过于接近,由此产生偏高的相似度与注意力分数,进而影响全局上下文信息编码。TOCOL 试图通过对比学习缓解这一问题。摘要未交代正负样本构造、损失具体形式、与原预训练目标的组合方式,以及该目标作用于哪些表示或注意力计算环节,这些细节尚未验证。 作者报告,在 GLUE Benchmark 上,TOCOL 相较原始 BERT 获得改善,并通过进一步分析展示了低资源场景下的鲁棒性。摘要未提供具体任务分数、训练数据规模或低资源划分,因而无法量化收益,也不能判断各任务改善是否一致;实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG Transformer 的时序或通道 token 也存在表示过度聚集、相似度主导注意力分配的问题,可考虑检验 token 级对比约束是否有助于上下文建模。但这是迁移假设:语言中的词频与全局语义并不直接对应 EEG 的生理结构,首先需要核对 TOCOL 的样本构造及其对文本语义、数据规模的依赖。可复用候选是表示几何约束,需改造的是 EEG token 定义与正负样本关系;低信噪比、跨被试及通道差异可能使对比目标强化伪迹或被试身份,小数据下也可能产生不稳定估计。一个可证伪的小实验是在固定跨被试划分、相同 EEG 骨干与训练预算下,比较加入该约束前后的任务指标及表示各向异性,检验几何变化是否伴随泛化收益。已有 EEG 相关工作尚未检索确认。
值得核对其如何将 token 级自监督对比目标融入注意力机制,以及表示空间变化是否对应下游收益;摘要报告了 GLUE 与低资源场景的改善,但具体协议及其对 EEG 的适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net