在《我的世界》里挖钻石把AI难哭,DeepMind最新算法终于扳回颜面

不靠攻略和“外挂”

鱼羊 Alex 发自 凹非寺

量子位 | 公众号 QbitAI

又一个AI老大难问题,被DeepMind攻克了:

一只名叫DreamerV3的AI,在啥也不知道的情况下被丢进《我的世界》(MineCraft)里,摸爬滚打17天,还真就学会了如何从0开始挖钻石。

△就是从撸树开始的那种

要知道,之前为了攻克这个问题,CMU、微软、DeepMind和OpenAI还联手在NeurIPS上拉了个比赛,叫MineRL。

结果搞了三四年,AI们也没能在不参考人类经验的情况下,完成挖钻石任务。

此前表现最好的VPT选手,为了达成这一成就,可是狂看了70000+小时《我的世界》游戏视频,并且用上了720个V100……

这不DeepMind的最新结果一出,研究人员都开心坏了。

MineRL的发起人之一、前OpenAI研究科学家William Guss就第一时间跑来发表贺电:

4年了,“钻石挑战”终于被攻克了!

今年刚拿到了NeurIPS杰出数据集和基准论文奖的MineDojo作者、英伟达AI科学家范麟熙则表示:

AI玩转MineCraft背后,有个莫拉维克悖论:一些任务对于人类而言很困难(比如围棋),但对AI来说很简单。但像MineCraft这样人类高玩无数的游戏,情况却相反。
DreamerV3能在没有任何人工数据辅助的情况下收集钻石,这让我感到非常兴奋。

如何做到

所以,这个0基础挖钻石任务,到底有啥难点?

首先,在《我的世界》里,初始世界是完全随机生成的。

即使是人类玩家,想要快速挖到钻石,也得有相当丰富的经验。比如,知道怎么推算钻石的位置、掌握一些挖掘窍门(如鱼骨挖矿法)等。

排除掉经验的因素,对于AI来说,这个挖钻石的过程也挺复杂,至少得要7个步骤

第一步,玩家在空手进入《我的世界》中时,需要先撸树来获取木块:

第二步,得用木块用来合成工作台:

第三步,在工作台上合成木镐,用来挖圆石:

第四步,获得圆石之后,需要合成一个石镐,用来快速挖铁矿:

第五步,为了将铁矿合成铁锭,还需要做一个熔炉来烧铁:

第六步,合成铁镐,用来挖钻石:

第七步,寻找钻石,然后用铁镐挖出钻石:

p.s. 有玩家测试过,在AI知道钻石等各种资源坐标的情况下(开外挂),也需要2-3分钟才能搞定。

这也就意味着,AI必须在有限时间里,做出大量决策。

那么既然不能参考人类高手的经验,自然就需要强化学习(Reinforcement Learning)出马。

具体而言,DeepMind的研究人员提出了一种基于世界模型的通用算法

在整体架构上,DreamerV3由3个神经网络组成:世界模型、评委(critic)和演员(actor)。

世界模型要做的,是把环境输入编码为离散的表征,并通过预测来指导下一步要执行的操作。

而评委和演员则会根据抽象出来的表征进行学习。其中,评委网络会输出一个标量值来代表行动价值,从而帮助演员网络选择最优的行动。

这里面的一个核心点在于,DeepMind的研究人员希望DreamerV3不仅仅能处理同类型的问题,还可以用固定超参数,掌握跨领域任务

因此,研究人员需要系统地解决世界模型、评委和演员等各个组件中信号大小不同,以及稳定平衡目标的问题。

研究人员发现:

以前的世界模型,需要根据复杂3D环境的视觉输入,对表征损失进行不同的缩放,在训练过程中还得调整不少超参数,但这里面其实有许多没必要的细节。

而如果把自由比特(free bits)和KL平衡(KL balancing)结合起来,就可以在不调整超参数的情况下让DreamerV3在不同领域中学习。

KL平衡是上一代DreamerV2中提出的一项新技术。能够使预测向表征移动的速度比表征向预测移动的速度更快,带来更精确的预测。

自由比特避免了简单环境下的过度拟合。

DreamerV3的三大块都用上了固定超参数,具体如下:

实验结果

也就是说,DreamerV3如今成了世界上第一个纯靠自己摸索,就能在《我的世界》里速挖钻石的AI。

并且DreamerV3的本事可不只是玩MC。

在另外7项基准测试中,DreamerV3都取得了成功,并且在BSuite、Crafter上达到了SOTA。

值得一提的是,在这些任务中,训练智能体所用到的GPU资源都仅为1块V100。研究人员表示,这意味着有更多的实验室能跑得动这一模型。

另外,在需要时空推理的三维空间中,DreamerV3也能快速进行学习。

在DeepMind为强化学习专门打造的3D平台DMLab上,DreamerV3在任务中使用的交互次数仅为IMPALA的1/130。

目前,DreamerV3的代码是coming soon的状态。

感兴趣的小伙伴可以蹲一波~

参考链接:
[1]https://danijar.com/project/dreamerv3/
[2]https://twitter.com/DeepMind/status/1613159943040811010

版权所有,未经授权不得以任何形式转载及使用,违者必究。