arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.03751cs.LGcs.AI
收藏

零阶优化遇上人类反馈:通过排序预言机实现可证明的学习

Zeroth-Order Optimization Meets Human Feedback: Provable Learning via Ranking Oracles

Shenzhen Research Institute of Big Data(深圳大数据研究院)

Zhiwei Tang, Dmitry Rybin, Tsung-Hui Chang

AI总结:

本研究针对仅能通过排序预言机评估的黑盒优化问题,提出带理论保证的零阶优化算法ZO-RankSGD,可应用于强化学习策略优化与扩散模型图像生成,能以少量人类反馈显著提升效果,推动零阶优化与AI人类意图对齐研究。

摘要:

本研究深入探讨了一类新兴的优化挑战:黑盒目标函数仅能通过排序预言机(ranking oracle)进行评估——这是现实场景中十分常见的情况,尤其是当函数由人类评判者进行评估时。该挑战的灵感来源于人类反馈强化学习(Reinforcement Learning with Human Feedback, RLHF),这是近期被用于借助人类指导提升大语言模型(Large Language Models, LLMs)性能的方法。我们提出了ZO-RankSGD,这是一种专为解决该优化问题设计的创新零阶优化算法,并附带理论保证。我们的算法采用一种新颖的基于排序的随机估计器来确定下降方向,且能够保证收敛到驻点。此外,ZO-RankSGD可直接应用于强化学习(Reinforcement Learning, RL)中的策略优化问题,特别是在仅能获得回合奖励的排序预言机的场景下。最后但同样重要的是,我们在一项新颖的应用中验证了ZO-RankSGD的有效性:利用人类排序反馈提升扩散生成模型的生成图像质量。在所有实验中我们发现,仅需少数几轮人类反馈,ZO-RankSGD就能显著提升生成图像的细节表现。总体而言,我们的工作通过解决仅具备排序反馈的函数优化问题,推动了零阶优化领域的发展,同时为人工智能(Artificial Intelligence, AI)对齐人类意图提供了一种全新且有效的方法。

英文摘要:

In this study, we delve into an emerging optimization challenge involving a black-box objective function that can only be gauged via a ranking oracle-a situation frequently encountered in real-world scenarios, especially when the function is evaluated by human judges. Such challenge is inspired from Reinforcement Learning with Human Feedback (RLHF), an approach recently employed to enhance the performance of Large Language Models (LLMs) using human guidance. We introduce ZO-RankSGD, an innovative zeroth-order optimization algorithm designed to tackle this optimization problem, accompanied by theoretical assurances. Our algorithm utilizes a novel rank-based random estimator to determine the descent direction and guarantees convergence to a stationary point. Moreover, ZO-RankSGD is readily applicable to policy optimization problems in Reinforcement Learning (RL), particularly when only ranking oracles for the episode reward are available. Last but not least, we demonstrate the effectiveness of ZO-RankSGD in a novel application: improving the quality of images generated by a diffusion generative model with human ranking feedback. Throughout experiments, we found that ZO-RankSGD can significantly enhance the detail of generated images with only a few rounds of human feedback. Overall, our work advances the field of zeroth-order optimization by addressing the problem of optimizing functions with only ranking feedback, and offers a new and effective approach for aligning Artificial Intelligence (AI) with human intentions.

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

↑