Instruction Pre-Training: Language Models are Supervised Multitask Learners
This work addresses the challenge of scaling supervised multitask learning for language models, offering a method to boost performance and efficiency, though it appears incremental as it builds on existing pre-training paradigms.
The paper tackles the problem of improving language model generalization by proposing Instruction Pre-Training, a supervised multitask pre-training framework that augments raw corpora with instruction-response pairs, resulting in enhanced base models and enabling smaller models like Llama3-8B to match or outperform larger ones like Llama3-70B.
Unsupervised multitask pre-training has been the critical method behind the recent success of language models (LMs). However, supervised multitask learning still holds significant promise, as scaling it in the post-training stage trends towards better generalization. In this paper, we explore supervised multitask pre-training by proposing Instruction Pre-Training, a framework that scalably augments massive raw corpora with instruction-response pairs to pre-train LMs. The instruction-response pairs are generated by an efficient instruction synthesizer built on open-source models. In our experiments, we synthesize 200M instruction-response pairs covering 40+ task categories to verify the effectiveness of Instruction Pre-Training. In pre-training from scratch, Instruction Pre-Training not only consistently enhances pre-trained base models but also benefits more from further instruction tuning. In continual pre-training, Instruction Pre-Training enables Llama3-8B to be comparable to or even outperform Llama3-70B. Our model, code, and data are available at https://github.com/microsoft/LMOps.