Highlight

Scaling Instruction-Finetuned Language Models

Studies instruction finetuning at scale across more tasks, larger models, and chain-of-thought data, yielding Flan-PaLM and released Flan-T5 checkpoints.

Based on

Scaling Instruction-Finetuned Language Models

By Hyung Won Chung, Le Hou, S. Longpre et al.Journal of machine learning research
Read original article →

Starting from the observation that finetuning language models on datasets phrased as instructions improves both performance and generalization to unseen tasks, the paper systematically studies instruction finetuning by scaling three factors: the number of finetuning tasks, the size of the model, and the inclusion of chain-of-thought reasoning data. The approach is evaluated across several model classes, including PaLM, T5, and U-PaLM, over multiple prompting setups such as zero-shot, few-shot, and chain-of-thought, and across a range of benchmarks including MMLU, BBH, TyDiQA, MGSM, and open-ended generation.

Instruction finetuning dramatically improves performance across these settings; for example, Flan-PaLM 540B finetuned on 1.8K tasks outperforms PaLM 540B by an average of 9.4% and reaches state-of-the-art results such as 75.2% on five-shot MMLU. The authors also publicly release Flan-T5 checkpoints, which deliver strong few-shot performance even compared with much larger models like PaLM 62B, establishing instruction finetuning as a general and practical method for improving the performance and usability of pretrained language models.

Abstract

Finetuning language models on datasets phrased as instructions improves performance and generalization to unseen tasks. This work scales instruction finetuning along three axes: number of tasks, model size, and chain-of-thought data. Across model classes (PaLM, T5, U-PaLM), prompting setups, and benchmarks, it sharply improves results: Flan-PaLM 540B tuned on 1.8K tasks beats PaLM 540B by +9.4% on average and hits 75.2% on five-shot MMLU. The released Flan-T5 checkpoints show strong few-shot performance even against much larger models.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

instruction finetuningchain-of-thoughtlarge language modelsFlan-T5Flan-PaLMmodel scaling
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.

Scaling Instruction-Finetuned Language Models | Aramai