跳到正文
OpenReview · Representation learning· Ruifeng Ren; Yong Liu·· 2024-01-01精选AI 评分78

从表征学习视角理解 Transformers 如何实现上下文学习

Towards Understanding How Transformers Learn In-context Through a Representation Learning Lens

AI 导读

基于摘要分析。本文研究 Transformer 的上下文学习(In-context Learning,ICL)机制,核心贡献是借助核方法构造单个 softmax attention 层的对偶模型,将无需参数更新的 ICL 推理与该对偶模型的训练过程联系起来。主要研究对象是 Transformer 的表征学习机制,并非 EEG 或 BCI 解码。 作者报告,在所研究的设定下,attention 层的 ICL 推理过程对应于对偶模型的训练过程,其生成的 token 表征预测与对偶模型的测试输出等价。在此基础上,作者从表征学习角度分析对偶模型的训练,并推导与 demonstration token 数量有关的泛化误差界。摘要未给出等价关系的具体假设、核函数形式、误差界表达式及其适用范围,尚不能据此推断增加示例数量必然改善任意任务表现。 作者进一步将理论结论扩展至一个 Transformer 层及多个 attention 层的情形,并受表征学习、尤其是对比学习方法启发,提出 attention 层的潜在修改。作者报告设计了实验支持上述发现,但具体修改、数据集、对照基线、评估指标、实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 模型同样通过上下文示例完成任务适配,该对偶分析可能提供研究示例数量与表征预测关系的理论切入点;这不等于已证实的 EEG/BCI 方法。其是否适用于连续信号 token、低信噪比和跨被试分布变化,仍需核对原文理论假设后评估,已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对其将 attention 的 ICL 推理与对偶模型训练建立等价关系的理论条件,以及 demonstration token 数量相关的泛化界;摘要尚不足以验证其对 EEG/BCI 的适用性。

来源:OpenReview · Representation learning · openreview.net