扩展指令微调语言模型

Scaling Instruction-Finetuned Language Models

郑亨原 Hyung Won Chung · Google · 2022-10-20 · arXiv:2210.11416 ↗ · 被引 4250

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

摘要 · Abstract

对以指令形式表述的数据集集合进行语言模型微调,已被证明能提升模型性能及对未见任务的泛化能力。本文探索指令微调,重点关注:(1) 扩展任务数量,(2) 扩大模型规模,(3) 在思维链数据上进行微调。我们发现,结合上述方面的指令微调能显著提升多种模型类别(PaLM、T5、U-PaLM)、提示设置(零样本、少样本、思维链)以及评估基准(MMLU、BBH、TyDiQA、MGSM、开放式生成)的性能。例如,在 1.8K 个任务上经过指令微调的 Flan-PaLM 540B,平均性能比 PaLM 540B 高出 9.4%。Flan-PaLM 540B 在多个基准上达到最先进水平,如五次样本 MMLU 上达到 75.2%。我们还公开发布了 Flan-T5 检查点,即使与更大的模型(如 PaLM 62B)相比,也展现出强大的少样本性能。总体而言,指令微调是提升预训练语言模型性能和可用性的通用方法。

Finetuning language models on a collection of datasets phrased as instructions has been shown to improve model performance and generalization to unseen tasks. In this paper we explore instruction finetuning with a particular focus on (1) scaling the number of tasks, (2) scaling the model size, and (3) finetuning on chain-of-thought data. We find that instruction finetuning with the above aspects dramatically improves performance on a variety of model classes (PaLM, T5, U-PaLM), prompting setups (zero-shot, few-shot, CoT), and evaluation benchmarks (MMLU, BBH, TyDiQA, MGSM, open-ended generation). For instance, Flan-PaLM 540B instruction-finetuned on 1.8K tasks outperforms PALM 540B by a large margin (+9.4% on average). Flan-PaLM 540B achieves state-of-the-art performance on several benchmarks, such as 75.2% on five-shot MMLU. We also publicly release Flan-T5 checkpoints, which achieve strong few-shot performance even compared to much larger models, such as PaLM 62B. Overall, instruction finetuning is a general method for improving the performance and usability of pretrained language models.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 11)

阅读逐段中英对照全文 →