用游戏框架训练更可信的AI助手,解决人类反馈强化学习的欺骗问题。
AssistanceZero: Scalably Solving Assistance Games
- 扩展AlphaZero,加入预测人类行为与奖励的神经网络,实现不确定性下的规划。
- 在超过10^400种目标的Minecraft任务中,性能优于无模型强化学习和模仿学习。
- 人类实验显示助手能显著减少用户完成建筑任务的操作次数,适合复杂环境应用。
协助游戏是训练人工智能助手的一种有前景替代方案,可克服强化学习人类反馈(RLHF)中诱导欺骗行为等关键缺陷。其核心是将助手与用户的交互建模为二人博弈,且助手无法观察共同目标。尽管潜力巨大,现有研究仅限于简单场景。将其拓展至复杂环境面临双重挑战:在不确定下求解难解决策问题,以及准确建模人类行为。本文提出首个可扩展的协助游戏求解方法——AssistanceZero,通过在AlphaZero基础上引入预测人类动作与奖励的神经网络,实现不确定性下的规划。我们在一个基于Minecraft、包含超10^400种可能目标的新颖协助游戏中验证该方法。结果表明,AssistanceZero在性能上超越了无模型强化学习算法和模仿学习。人类实验显示,经其训练的助手能显著降低用户完成建造任务所需的操作次数。研究证明,协助游戏是复杂环境中训练高效AI助手的可行框架。代码与模型已开源。
原文摘要 · Abstract (English)
Assistance games are a promising alternative to reinforcement learning from human feedback (RLHF) for training AI assistants. Assistance games resolve key drawbacks of RLHF, such as incentives for deceptive behavior, by explicitly modeling the interaction between assistant and user as a two-player game where the assistant cannot observe their shared goal. Despite their potential, assistance games have only been explored in simple settings. Scaling them to more complex environments is difficult because it requires both solving intractable decision-making problems under uncertainty and accurately modeling human users' behavior. We present the first scalable approach to solving assistance games and apply it to a new, challenging Minecraft-based assistance game with over $10^{400}$ possible goals. Our approach, AssistanceZero, extends AlphaZero with a neural network that predicts human actions and rewards, enabling it to plan under uncertainty. We show that AssistanceZero outperforms model-free RL algorithms and imitation learning in the Minecraft-based assistance game. In a human study, our AssistanceZero-trained assistant significantly reduces the number of actions participants take to complete building tasks in Minecraft. Our results suggest that assistance games are a tractable framework for training effective AI assistants in complex environments. Our code and models are available at https://github.com/cassidylaidlaw/minecraft-building-assistance-game.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。