小模型通过示例学习,能组合新技能生成文本。
Can Models Learn Skill Composition from Examples?
- 用大模型生成含多种技能的文本,小模型再据此微调。
- 训练时只接触2~3种技能,测试却能组合4~5种新技能。
- 即使从未见过的技能类别,也能有效组合使用。
随着大语言模型日益先进,其组合泛化能力——即在训练中未出现过的全新方式组合已学技能——受到广泛关注。近期研究提出SKILL-MIX评估任务,要求模型生成展示指定k元技能组合的短文。小模型在k=3时表现不佳,而GPT-4在k=5和6时表现良好。本文采用类似设定,评估小模型从示例中学习组合泛化的能力。利用包括修辞、文学、推理、心智理论和常识在内的多样化语言技能,由GPT-4生成随机组合的k种技能文本。随后对7B和13B参数模型在不同k值下进行微调,发现:(1) 在包含k=2和3技能的文本上微调后,模型在生成包含k=4和5技能的文本时有明显提升,尽管这些组合从未在训练中出现;(2) 当技能类别分为训练集与保留集时,模型在测试阶段能显著提升对保留技能的组合能力,即使微调时仅接触过训练类技能。研究还表明,将富含技能(可能为合成)的文本纳入训练,可大幅提升模型的组合能力。
原文摘要 · Abstract (English)
As large language models (LLMs) become increasingly advanced, their ability to exhibit compositional generalization -- the capacity to combine learned skills in novel ways not encountered during training -- has garnered significant attention. This type of generalization, particularly in scenarios beyond training data, is also of great interest in the study of AI safety and alignment. A recent study introduced the SKILL-MIX evaluation, where models are tasked with composing a short paragraph demonstrating the use of a specified $k$-tuple of language skills. While small models struggled with composing even with $k=3$, larger models like GPT-4 performed reasonably well with $k=5$ and $6$. In this paper, we employ a setup akin to SKILL-MIX to evaluate the capacity of smaller models to learn compositional generalization from examples. Utilizing a diverse set of language skills -- including rhetorical, literary, reasoning, theory of mind, and common sense -- GPT-4 was used to generate text samples that exhibit random subsets of $k$ skills. Subsequent fine-tuning of 7B and 13B parameter models on these combined skill texts, for increasing values of $k$, revealed the following findings: (1) Training on combinations of $k=2$ and $3$ skills results in noticeable improvements in the ability to compose texts with $k=4$ and $5$ skills, despite models never having seen such examples during training. (2) When skill categories are split into training and held-out groups, models significantly improve at composing texts with held-out skills during testing despite having only seen training skills during fine-tuning, illustrating the efficacy of the training approach even with previously unseen skills. This study also suggests that incorporating skill-rich (potentially synthetic) text into training can substantially enhance the compositional capabilities of models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。