用强弱大模型密度比代替人工标注,大幅降低偏好训练成本。
Dr. SoW: Density Ratio of Strong-over-weak LLMs for Reducing the Cost of Human Annotation in Preference Tuning
- 通过强弱模型的对数密度比生成奖励信号,替代人工标注。
- 使用两个Mistral-7B模型实现82.6分的RewardBench得分,优于同类模型。
- 适用于需要低成本定制化奖励函数的领域,尤其适合安全与推理任务。
偏好训练依赖高质量的人工偏好数据,但其收集成本高、耗时长。本文提出Dr.SoW(强模型对弱模型的密度比),一种无需人工标注的低成本方法,利用现成的大语言模型自动生成偏好数据。Dr.SoW采用更对齐与更不对齐模型之间的对数密度比作为奖励信号。我们在221组不同大模型组合上评估该方法,发现模型性能差距与奖励信号质量间存在强相关性,为选择用于标注的模型提供了实用指导。此外,我们构建了端到端管道,可根据用户查询领域定制奖励函数。无需微调即可提升特定领域评估准确率;使用一对Mistral-7B模型,Dr.SoW在RewardBench上取得82.6分,超过同模型类别最佳训练奖励函数,并在安全(91.0)和推理(88.0)领域表现媲美顶尖模型。进一步地,我们使用Dr.SoW标注的数据对Llama-3-8B-Instruct进行偏好训练,使其在ArenaHard上的胜率提升至37.4%(+15.1%),在长度控制的AlpacaEval 2.0上达40.7%(+17.8%)。
原文摘要 · Abstract (English)
Preference tuning relies on high-quality human preference data, which is often expensive and time-consuming to gather. In this paper, we introduce Dr.SoW (Density Ratio of Strong over Weak) a cost-effective method that eliminates the reliance for human annotation by leveraging off-the-shelf LLMs for preference data annotation. Dr.SoW uses the log-density ratio between a better-aligned and a less-aligned LLM as a reward signal. We evaluate Dr.SoW across 221 different LLM pairs and empirically find a strong correlation between the performance gap of the paired models and the quality of the reward signal. This insight provides a practical guideline for selecting LLMs for data annotation. Additionally, we introduce an end-to-end pipeline that customizes reward functions based on user query domains. Without fine-tuning, it improves accuracy on domain-specific evaluations. With a pair of Mistral-7B models, Dr.SoW achieves a RewardBench score of 82.6, outperforming the best trained reward functions from same model class and demonstrating competitive performance against SoTA models in Safety (91.0) and Reasoning (88.0) domains. Further, we preference-tune Llama-3-8B-Instruct using data annotated by Dr.SoW. Our approach pushes Llama-3-8B to achieve a 37.4 % (+15.1 %) win rate on ArenaHard and a 40.7 % (+17.8 %) win rate on length-controlled AlpacaEval 2.0.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。