arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.02481cs.CLcs.AI
收藏

Dr. SoW:用于减少偏好调优中人工标注成本的强弱LLM密度比方法

Dr. SoW: Density Ratio of Strong-over-weak LLMs for Reducing the Cost of Human Annotation in Preference Tuning

Guangxuan Xu, Kai Xu, Shivchander Sudalairaj, Hao Wang, Akash Srivastava

AI总结:

本文提出Dr.SoW方法,利用强弱LLM的对数密度比作为奖励信号替代人工标注,并基于221对模型评估提供LLM选择指南,显著提升了奖励模型及下游偏好调优的性能。

摘要:

偏好调优依赖于高质量的偏好数据,而收集这些数据通常既昂贵又耗时。在本文中,我们提出了Dr.SoW(Density Ratio of Strong over Weak,强弱密度比),这是一种高性价比的方法,通过利用现成的LLM进行偏好数据标注,消除了对人工标注的依赖。Dr.SoW使用对齐较好的LLM与对齐较差的LLM之间的对数密度比作为奖励信号。我们在221对不同的LLM上评估了Dr.SoW,并实证发现配对模型的性能差距与奖励信号的质量之间存在强相关性。这一见解为选择用于数据标注的LLM提供了实用指南。此外,我们引入了一个端到端流水线,可根据用户查询领域定制奖励函数。无需微调,它即可提高特定领域评估的准确率。使用一对Mistral-7B模型,Dr.SoW获得了82.6的RewardBench分数,超越了同类模型中训练效果最佳的奖励函数,并在Safety(91.0)和Reasoning(88.0)领域展现出与SoTA模型相当的竞争性能。进一步,我们使用Dr.SoW标注的数据对Llama-3-8B-Instruct进行偏好调优。我们的方法推动Llama-3-8B在ArenaHard上达到37.4%(+15.1%)的胜率,在长度控制的AlpacaEval 2.0上达到40.7%(+17.8%)的胜率。

英文摘要:

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.

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

↑