arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.11907cs.CL
收藏

通过自奖励对比提示蒸馏实现大语言模型直接对齐

Direct Large Language Model Alignment Through Self-Rewarding Contrastive Prompt Distillation

Tsinghua University(清华大学);Apple(苹果公司)

Aiwei Liu, Haoping Bai, Zhiyun Lu, Xiang Kong, Simon Wang, Jiulong Shan, Meng Cao, Lijie Wen

AI总结:

针对无人工标注偏好数据的大语言模型对齐问题,提出基于对比提示对的偏好评估与自奖励的DLMA方法,结合DPO算法实现对齐,性能超越RLHF。

摘要:

在无需人工标注偏好数据的情况下让大语言模型(LLM)与人类期望对齐是一个重要问题。本文提出一种利用对比提示对下响应对的输出概率来评估响应偏好的方法,该方法在LLaMA2-7B和LLaMA2-13B上相比RLAIF能取得更优性能。基于此,我们提出自动对齐方法Direct Large Model Alignment(DLMA,大模型直接对齐):首先用对比提示对自动生成偏好数据,接着继续用对比提示对评估生成的偏好数据并计算自奖励分数,最后结合该自奖励分数使用DPO算法有效对齐LLM。实验阶段,我们的DLMA方法无需依赖人工标注的偏好数据即可超越RLHF方法。

英文摘要:

Aligning large language models (LLMs) with human expectations without human-annotated preference data is an important problem. In this paper, we propose a method to evaluate the response preference by using the output probabilities of response pairs under contrastive prompt pairs, which could achieve better performance on LLaMA2-7B and LLaMA2-13B compared to RLAIF. Based on this, we propose an automatic alignment method, Direct Large Model Alignment (DLMA). First, we use contrastive prompt pairs to automatically generate preference data. Then, we continue to evaluate the generated preference data using contrastive prompt pairs and calculate a self-rewarding score. Finally, we use the DPO algorithm to effectively align LLMs by combining this self-rewarding score. In the experimental stage, our DLMA method could surpass the \texttt{RLHF} method without relying on human-annotated preference data.

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

↑