arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.01335cs.LGcs.AIcs.CLstat.ML
收藏

自我对弈微调将弱语言模型转化为强语言模型

Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Models

University of California, Los Angeles(加利福尼亚大学洛杉矶分校)

Zixiang Chen, Yihe Deng, Huizhuo Yuan, Kaixuan Ji, Quanquan Gu

AI总结:

本文提出SPIN自我对弈微调方法,使LLM利用先前迭代生成的数据并区分人工标注响应,在无额外标注数据下实现显著性能提升。

摘要:

通过监督微调(SFT)利用人工标注数据的力量,对于推进大型语言模型(LLM)至关重要。本文探讨了在无需获取额外人工标注数据的情况下,将弱LLM培养成强LLM的前景。我们提出一种名为Self-Play fIne-tuNing(SPIN)的新型微调方法,它从一个经过监督微调的模型开始。SPIN的核心是一种自我对弈机制,LLM通过与自身的不同实例对弈来提升能力。更具体地说,LLM利用其先前迭代生成自身的训练数据,并通过区分这些自生成响应与从人工标注数据获得的响应来改进其策略。该方法将LLM从初始模型逐步提升为强大模型,充分释放用于SFT的人工标注演示数据的潜力。理论上,我们证明只有当LLM策略与目标数据分布对齐时,该方法训练目标函数的全局最优解才会实现。实验上,我们在多个基准数据集上评估该方法,包括HuggingFace Open LLM Leaderboard、MT-Bench以及来自Big-Bench的数据集。结果表明,SPIN能够在多种基准上显著提升LLM性能,甚至优于通过直接偏好优化(DPO)训练并辅以额外GPT-4偏好数据的模型。这揭示了自我对弈的前景,使LLM无需专家对手即可实现人类水平性能。代码位于https://github.com/uclaml/SPIN。

英文摘要:

Harnessing the power of human-annotated data through Supervised Fine-Tuning (SFT) is pivotal for advancing Large Language Models (LLMs). In this paper, we delve into the prospect of growing a strong LLM out of a weak one without the need for acquiring additional human-annotated data. We propose a new fine-tuning method called Self-Play fIne-tuNing (SPIN), which starts from a supervised fine-tuned model. At the heart of SPIN lies a self-play mechanism, where the LLM refines its capability by playing against instances of itself. More specifically, the LLM generates its own training data from its previous iterations, refining its policy by discerning these self-generated responses from those obtained from human-annotated data. Our method progressively elevates the LLM from a nascent model to a formidable one, unlocking the full potential of human-annotated demonstration data for SFT. Theoretically, we prove that the global optimum to the training objective function of our method is achieved only when the LLM policy aligns with the target data distribution. Empirically, we evaluate our method on several benchmark datasets including the HuggingFace Open LLM Leaderboard, MT-Bench, and datasets from Big-Bench. Our results show that SPIN can significantly improve the LLM's performance across a variety of benchmarks and even outperform models trained through direct preference optimization (DPO) supplemented with extra GPT-4 preference data. This sheds light on the promise of self-play, enabling the achievement of human-level performance in LLMs without the need for expert opponents. Codes are available at https://github.com/uclaml/SPIN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑