TensorX
返回文献探索

Paper · arXiv 2406.13542

Self-play with Execution Feedback: Improving Instruction-following Capabilities of Large Language Models

Guanting Dong, Keming Lu, Chengpeng Li, Tingyu Xia, Bowen Yu, Chang Zhou, Jingren Zhou

17 upvotesJune 19, 2024arXiv 预印本
AI 摘要

AutoIF is a scalable method for automatically generating high-quality instruction-following training data by using code verification and execution feedback-based rejection sampling, improving various training algorithms for LLMs.

Large Language Models (LLMs)instruction-followingtraining datacode verificationexecution feedback-based rejection samplingSupervised Fine-Tuning (SFT)Reinforcement Learning from Human Feedback (RLHF)Offline DPOOnline DPOQwen2LLaMA3self-alignmentstrong-to-weak distillation

Abstract

One core capability of large language models (LLMs) is to follow natural language instructions. However, the issue of automatically constructing high-quality training data to enhance the complex instruction-following abilities of LLMs without manual annotation remains unresolved. In this paper, we introduce AutoIF, the first scalable and reliable method for automatically generating instruction-following training data. AutoIF transforms the validation of instruction-following data quality into code verification, requiring LLMs to generate instructions, the corresponding code to check the correctness of the instruction responses, and unit test samples to verify the code's correctness. Then, execution feedback-based rejection sampling can generate data for Supervised Fine-Tuning (SFT) and Reinforcement Learning from Human Feedback (RLHF) training. AutoIF achieves significant improvements across three training algorithms, SFT, Offline DPO, and Online DPO, when applied to the top open-source LLMs, Qwen2 and LLaMA3, in self-alignment and strong-to-weak distillation settings. Our code is publicly available at https://github.com/QwenLM/AutoIF.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Self-play with Execution Feedback: Improving Instruction-following Capabilities of Large Language Models | TensorX