通过挑选差异大的回复对,提升大模型对齐效率与质量
REAL: Response Embedding-based Alignment for LLMs
- 基于回复嵌入相似度筛选差异显著的响应对进行标注
- 在SHP2和HH-RLHF上实现更高胜率与更优得分差距
- 可减少65%标注工作量,降低标签错误率
将大语言模型(LLMs)对齐人类偏好是构建有用且安全AI工具的关键步骤,通常依赖监督数据集训练。主流方法如直接偏好优化(DPO)依赖人工标注的生成回复对。但回复对标注过程易引入人为偏差,构建高质量偏好数据集成本高昂。为此,我们提出REAL:基于回复嵌入的对齐方法,聚焦从候选回复中筛选出歧义较小的偏好对进行标注。该选择过程基于独立于提示的回复嵌入相似度,可在离策略设置下完成,避免训练中动态测量相似性。在真实数据集SHP2与合成数据集HH-RLHF上的实验表明,选择差异较大的回复对能提升直接对齐效果,并减少继承的标注误差。使用差异对训练的模型在对话任务中取得更高得分差与胜率。结果表明,聚焦差异对可降低标签错误并提高对齐效率,最多节省65%标注人力。
原文摘要 · Abstract (English)
Aligning large language models (LLMs) to human preferences is a crucial step in building helpful and safe AI tools, which usually involve training on supervised datasets. Popular algorithms such as Direct Preference Optimization (DPO) rely on pairs of AI-generated responses ranked according to human annotation. The response pair annotation process might bring human bias. Building a correct preference dataset is the costly part of the alignment pipeline. To improve annotation efficiency and quality in the LLMs alignment, we propose REAL: Response Embedding-based Alignment for LLMs, a strategy for constructing a high-quality training dataset that focuses on acquiring the less ambiguous preference pairs for labeling out of a set of response candidates. Our selection process is based on the similarity of embedding responses independently of prompts, which guarantees the selection process in an off-policy setting, avoiding adaptively measuring the similarity during the training. Experimental results on real-world dataset SHP2 and synthetic HH-RLHF benchmarks indicate that choosing dissimilar response pairs enhances the direct alignment of LLMs while reducing inherited labeling errors. The model aligned with dissimilar response pairs obtained a better margin and win rate on the dialogue task. Our findings suggest that focusing on distinct pairs can reduce the label error and improve LLM alignment efficiency, saving up to $65\%$ of annotators' work.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。