arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.13541cs.CLcs.AIcs.LG
收藏

通过多样且具代表性的回复文本实现标注高效的语言模型对齐

Annotation-Efficient Language Model Alignment via Diverse and Representative Response Texts

CyberAgent / Tokyo, Japan(CyberAgent)

Yuu Jinnai, Ukyo Honda

AI总结:

针对有限标注预算下构建有效偏好数据集的问题,提出标注高效偏好优化(AEPO)方法,通过选择最大化多样性和代表性的回复子集进行偏好标注,在相同预算下优于基线。

摘要:

偏好优化是微调大型语言模型以符合人类偏好的标准方法。偏好数据集的数量、多样性和代表性对偏好优化的有效性至关重要。然而,在许多应用中,获取大量偏好标注是很困难的。这就提出了一个问题:如何使用有限的标注预算来创建有效的偏好数据集。为此,我们提出了标注高效偏好优化(AEPO)。AEPO不是对所有可用的回复文本进行详尽的偏好标注,而是从可用回复中选择一个能最大化多样性和代表性的子集,然后对选定的回复进行偏好标注。通过这种方式,AEPO将标注预算集中在为更小但信息丰富的回复子集标注偏好上。我们使用AEPO在三个数据集上评估了偏好学习的性能,并表明它在相同标注预算下优于基线方法。我们的代码可在https://github.com/CyberAgentAILab/annotation-efficient-po获取。

英文摘要:

Preference optimization is a standard approach to fine-tuning large language models to align with human preferences. The quantity, diversity, and representativeness of the preference dataset are critical to the effectiveness of preference optimization. However, obtaining a large amount of preference annotations is difficult in many applications. This raises the question of how to use the limited annotation budget to create an effective preference dataset. To this end, we propose Annotation-Efficient Preference Optimization (AEPO). Instead of exhaustively annotating preference over all available response texts, AEPO selects a subset of responses that maximizes diversity and representativeness from the available responses and then annotates preference over the selected ones. In this way, AEPO focuses the annotation budget on labeling preferences over a smaller but informative subset of responses. We evaluate the performance of preference learning using AEPO on three datasets and show that it outperforms the baselines with the same annotation budget. Our code is available at https://github.com/CyberAgentAILab/annotation-efficient-po

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

↑