用假设类描述指导上下文学习,提升模型泛化与准确率
In-Context Learning with Hypothesis-Class Guidance
- 输入包含假设类描述和样本对,引导模型理解任务
- Transformer 可泛化到未见假设类,且准确率显著提升
- 适合研究上下文学习机制与指令作用的学者
近期研究多基于简单函数类探讨上下文学习(ICL)机制,但通常仅使用标注样本序列。现实中,标注样本常伴随任务说明,提供额外信息。本文提出假设类引导的上下文学习(ICL-HCG),其输入包含有限假设类 H 的文字描述及从 H 中某假设采样的 (x,y) 对。在 ICL-HCG 框架下,我们系统评估了:(i) 新假设类的泛化能力;(ii) 不同模型架构;(iii) 样本复杂度;(iv) 上下文数据不平衡;(v) 指令的作用;(vi) 预训练假设多样性的影响。结果表明:(a) Transformer 能有效学习 ICL-HCG 并泛化至未见假设与假设类;(b) 相较于无指令的 ICL,ICL-HCG 显著提升准确率,验证了指令的关键作用。
原文摘要 · Abstract (English)
Recent research has investigated the underlying mechanisms of in-context learning (ICL) both theoretically and empirically, often using data generated from simple function classes. However, the existing work often focuses on the sequence consisting solely of labeled examples, while in practice, labeled examples are typically accompanied by an instruction, providing some side information about the task. In this work, we propose ICL with hypothesis-class guidance (ICL-HCG), a novel synthetic data model for ICL where the input context consists of the literal description of a (finite) hypothesis class H and $(x,y)$ pairs from a hypothesis chosen from H. Under our framework ICL-HCG, we conduct extensive experiments to explore: (i) a variety of generalization abilities to new hypothesis classes; (ii) different model architectures; (iii) sample complexity; (iv) in-context data imbalance; (v) the role of instruction; and (vi) the effect of pretraining hypothesis diversity. As a result, we show that (a) Transformers can successfully learn ICL-HCG and generalize to unseen hypotheses and unseen hypothesis classes, and (b) compared with ICL without instruction, ICL-HCG achieves significantly higher accuracy, demonstrating the role of instructions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。