arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.08925cs.CLcs.AIcs.LGcs.RO
收藏

MaxMin-RLHF:面向多样化人类偏好的对齐方法

MaxMin-RLHF: Alignment with Diverse Human Preferences

University of Maryland(马里兰大学);Princeton University(普林斯顿大学);JP Morgan Chase AI Research(摩根大通人工智能研究院);University of Central Florida(中佛罗里达大学)

Souradip Chakraborty, Jiahao Qiu, Hui Yuan, Alec Koppel, Furong Huang, Dinesh Manocha, Amrit Singh Bedi, Mengdi Wang

AI总结:

本文针对单一奖励RLHF无法表示多样化人类偏好的问题,提出基于期望最大化和MaxMin目标的RLHF方法,在提升整体胜率的同时显著改善少数群体表现,兼顾鲁棒性与公平性。

摘要:

基于人类反馈的强化学习(RLHF)通过使用从偏好数据中导出的单一奖励模型,使语言模型与人类偏好对齐。然而,这种方法忽略了从多个用户收集的数据中固有的人类偏好多样性。在本工作中,我们首先推导出单一奖励RLHF对齐的不可能性结果,从而凸显其在表示多样化人类偏好方面的不足。为了为该问题提供公平的解决方案,我们通过期望最大化算法学习偏好分布的混合模型,并受社会选择理论中平等主义原则的启发,提出用于策略学习的MaxMin对齐目标,以更好地表示多样化的人类偏好。我们阐明了所提方法与分布鲁棒优化和通用效用强化学习之间的联系,从而突出了所提解决方案的通用性和鲁棒性。我们在小规模(GPT-2)和大规模语言模型(使用Tulu2-7B)上给出了全面的实验结果,并展示了所提方法在人类偏好存在多样性情况下的有效性。与常规RLHF算法相比,我们的算法在胜率上平均提高了16%以上,并将少数群体的胜率(准确率)提高了33%以上,同时不影响多数群体的性能,展示了我们方法的鲁棒性和公平性。我们指出,本工作中的发现不仅限于语言模型,还可推广到一般的强化学习。

英文摘要:

Reinforcement Learning from Human Feedback (RLHF) aligns language models to human preferences by employing a singular reward model derived from preference data. However, such an approach overlooks the rich diversity of human preferences inherent in data collected from multiple users. In this work, we first derive an impossibility result of alignment with single reward RLHF, thereby highlighting its insufficiency in representing diverse human preferences. To provide an equitable solution to the problem, we learn a mixture of preference distributions via an expectation-maximization algorithm and propose a MaxMin alignment objective for policy learning inspired by the Egalitarian principle in social choice theory to better represent diverse human preferences. We elucidate the connection of our proposed approach to distributionally robust optimization and general utility RL, thereby highlighting the generality and robustness of our proposed solution. We present comprehensive experimental results on small-scale (GPT-2) and large-scale language models (with Tulu2-7B) and show the efficacy of the proposed approach in the presence of diversity among human preferences. Our algorithm achieves an average improvement of more than 16% in win-rates over conventional RLHF algorithms and improves the win-rate (accuracy) for minority groups by over 33% without compromising the performance of majority groups, showcasing the robustness and fairness of our approach. We remark that our findings in this work are not only limited to language models but also extend to reinforcement learning in general.

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

↑