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

基于对比强化学习的故事讲述偏好学习

Robust Preference Learning for Storytelling via Contrastive Reinforcement Learning

CarperAI;Brown University(布朗大学);Georgia Tech(佐治亚理工学院);University of Waterloo(滑铁卢大学);EleutherAI

Louis Castricato, Alexander Havrilla, Shahbuland Matiana, Michael Pieler, Anbang Ye, Ian Yang, Spencer Frazier, Mark Riedl

摘要:

受控自动故事生成寻求生成满足来自自然语言评论或偏好的约束的自然语言故事。控制楼层偏好的现有方法利用劳动密集型且经常不一致的提示工程。他们也可以使用logit操作方法,该方法需要注释的数据集存在于所需的属性中。为了解决这些问题,我们首先训练一个对比双编码器模型,将故事与相应的人类评论(CARP)对齐,建立一个通用的偏好模型。这随后被用作奖励函数,以通过强化学习微调生成语言模型。然而,简单地用对比奖励模型微调生成语言模型并不总是可靠地导致能够生成满足用户偏好的故事的故事生成系统。为了提高故事生成的鲁棒性,我们使用一种学习技术进一步微调对比奖励模型。然后进行一项人类参与者研究,比较我们的完整系统、消融和两个基线的世代。我们表明,完整的微调管道的结果在故事发生器优于LLM 20倍大以及基于logit的方法。这激发了将对比学习用于通用人类偏好建模。

英文摘要:

Controlled automated story generation seeks to generate natural language stories satisfying constraints from natural language critiques or preferences. Existing methods to control for story preference utilize prompt engineering which is labor intensive and often inconsistent. They may also use logit-manipulation methods which require annotated datasets to exist for the desired attributes. To address these issues, we first train a contrastive bi-encoder model to align stories with corresponding human critiques, named CARP, building a general purpose preference model. This is subsequently used as a reward function to fine-tune a generative language model via reinforcement learning. However, simply fine-tuning a generative language model with a contrastive reward model does not always reliably result in a story generation system capable of generating stories that meet user preferences. To increase story generation robustness we further fine-tune the contrastive reward model using a prompt-learning technique. A human participant study is then conducted comparing generations from our full system, ablations, and two baselines. We show that the full fine-tuning pipeline results in a story generator preferred over a LLM 20x as large as well as logit-based methods. This motivates the use of contrastive learning for general purpose human preference modeling.

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

↑