LIMA:少即是多——对齐的简约之道

LIMA: Less Is More for Alignment

周春婷 Chunting Zhou · Meta AI · 2023-05-18 · arXiv:2305.11206 ↗ · 被引 1329

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

大型语言模型的训练分为两个阶段:(1)从原始文本进行无监督预训练,学习通用表示;(2)大规模指令微调和强化学习,以更好地对齐最终任务和用户偏好。我们通过训练 LIMA 来衡量这两个阶段的相对重要性。LIMA 是一个 65B 参数的 LLaMa 语言模型,仅使用 1000 个精心挑选的提示和响应进行标准监督损失微调,未使用任何强化学习或人类偏好建模。LIMA 表现出色,仅从训练数据中的少量示例就学会了遵循特定的响应格式,包括从规划旅行行程到推测替代历史等复杂查询。此外,该模型对训练数据中未出现的未见任务也具有良好的泛化能力。在受控人类研究中,LIMA 的响应在 43%的情况下与 GPT-4 相当或更受青睐;与 Bard 相比这一比例为 58%,与经过人类反馈训练的 DaVinci003 相比则为 65%。这些结果强烈表明,大型语言模型中的几乎所有知识都是在预训练阶段学到的,只需有限的指令微调数据即可教会模型生成高质量输出。

Large language models are trained in two stages: (1) unsupervised pretraining from raw text, to learn general-purpose representations, and (2) large scale instruction tuning and reinforcement learning, to better align to end tasks and user preferences. We measure the relative importance of these two stages by training LIMA, a 65B parameter LLaMa language model fine-tuned with the standard supervised loss on only 1,000 carefully curated prompts and responses, without any reinforcement learning or human preference modeling. LIMA demonstrates remarkably strong performance, learning to follow specific response formats from only a handful of examples in the training data, including complex queries that range from planning trip itineraries to speculating about alternate history. Moreover, the model tends to generalize well to unseen tasks that did not appear in the training data. In a controlled human study, responses from LIMA are either equivalent or strictly preferred to GPT-4 in 43% of cases; this statistic is as high as 58% when compared to Bard and 65% versus DaVinci003, which was trained with human feedback. Taken together, these results strongly suggest that almost all knowledge in large language models is learned during pretraining, and only limited instruction tuning data is necessary to teach models to produce high quality output.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 12)

阅读逐段中英对照全文 →