用大模型自我迭代优化数据,让微调更高效
Refine-n-Judge: Curating High-Quality Preference Chains for LLM-Fine-Tuning
- 让一个大模型轮流当编辑和裁判,自动改进回答质量
- 在多个任务上使微调模型胜率超74%,部分评测提升19%
- 无需人工标注,适合大规模数据构建与模型训练
大型语言模型通过基于偏好微调取得显著进展,其效果高度依赖训练数据质量。虽然人类反馈能提升数据质量,但成本高且难扩展。本文提出Refine-n-Judge,一种仅用单个大模型作为生成器和评判者、实现自动化迭代优化的方法。该方法在每轮中由模型生成改进后的回答,并判断是否优于原回答,直至不再有改进为止,从而形成高质量、带偏好标签的响应序列。我们在涵盖编码、数学和对话等五类任务的多个公开数据集上验证了该方法的有效性。使用Llama 3.1-8B和Llama 3.3-70B在经Refine-n-Judge增强的数据集上微调的模型,在与GPT-4标注的原始数据微调模型对比中,获得超过74%的偏好胜率;在AlpacaEval和AlpacaEval 2.0上分别提升5%,在MT-Bench上提升19%。结果表明,该方法可生成高质量数据并实现可扩展的模型性能提升。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have demonstrated remarkable progress through preference-based fine-tuning, which critically depends on the quality of the underlying training data. While human feedback is essential for improving data quality, it is costly and does not scale well. In this paper, we introduce Refine-n-Judge, an automated iterative approach that leverages a single LLM as both a refiner and a judge to enhance dataset quality. Unlike existing iterative refinement methods, Refine-n-Judge employs an LLM to both generate refinements and explicitly evaluate each improvement, ensuring that every iteration meaningfully enhances the dataset without requiring additional human annotation or a separate reward model. At each step, the LLM refines a response and judges whether the refinement is an improvement over the previous answer. This process continues until the LLM prefers the initial answer over the refinement, indicating no further improvements. This produces sequences of increasing quality, preference-labeled responses ideal for fine-tuning. We demonstrate the effectiveness of Refine-n-Judge across a range of public datasets spanning five corpora, targeting tasks such as coding, math, and conversation. Models (Llama 3.1-8B and Llama 3.3-70B) fine-tuned on Refine-n-Judge-enhanced datasets were preferred by LLM judges in over 74% of comparisons against models tuned on the original dataset by GPT-4. Additionally, we report performance gains: +5% on AlpacaEval and AlpacaEval 2.0, and +19% on MT-Bench. Our results indicate that Refine-n-Judge produces high-quality datasets and scalable model improvements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。