正在加载视频...
视频加载失败
Can reinforcement learning from AI feedback unlock new capabilities in AI agents? Introducing Motif, an LLM-powered method for intrinsic motivation from AI feedback. Motif extracts reward functions from Llama 2's preferences and uses them to train agents with reinforcement learning. On the complex NetHack game, Motif solves previously unsolved... show more
10 条评论

To benchmark the capabilities of Motif, we apply it to NetHack, a challenging rogue-like videogame, in which a player has to go through different levels of a dungeon, killing monsters, gathering objects and overcoming significant difficulties. Yet common sense can take you very far in such an environment! We use the messages from the game (i.e., even captions shown in 20% of interactions) to ask Llama 2 about its preferences about game situations. In this image, for instance, the event caption is "You kill the yellow mold!", which is understood by the Llama 2 model due to its knowledge of NetHack. / 2

Motif leverages recent ideas from RLAIF, asking an LLM to rank event captions and then distilling those preferences into a reward function. Motif has three phases: • Dataset annotation: given a dataset of observations with event captions, Motif uses Llama 2 to give preferences on sampled pairs according to its perception of how good and promising they are in the environment • Reward training: the resulting dataset of annotated pairs is used to learn a reward model from preferences • RL training: the reward function is given to an agent interacting with the environment and used to train it with RL, possibly alongside an external reward / 3

Our prompting strategy is very simple: we ask the model (with a form of chain of thought) to reason about its knowledge of NetHack and to give preferences about two messages coming from NetHack. We add the possibility to have a modifier in the prompt, to further influence and steer the behavior of an agent. / 4

We measure the performance of Motif in the task of NetHack score maximization, finding, surprisingly, that the reward extracted from Llama 2 yields an agent obtaining better score than an agent directly optimizing the score itself. This is the first time an intrinsically motivated agent outperforms the task-driven baseline on such a complex environment. When combining the score reward to the one coming from the LLM, the performance improves even further, both in terms of sample efficiency and final score. / 5

Motif also solves extremely sparse reward tasks, when other methods never find the solution. Motif does not need any demonstration or expert data, and simply bootstraps its knowledge from the LLM's common sense, to effectively explore the environment. Motif distills the capability of finding new information into AI agents. For instance, they will already know that opening a door is good for exploration even before having ever interacted with one. / 6

To understand where the performance of Motif is coming from, we analyzed the behavior of agents trained with Motif’s and the extrinsic reward function. We discovered that, when compared to an agent just trained using the score from the game, the behavior of agents trained with Motif is more aligned with human intuitions on playing NetHack: instead of going to the levels as fast as possible, encountering monsters too strong for its experience level, Motif’s reward encourages survival. An agent trained with both the intrinsic and extrinsic rewards is thus able to leverage this survival attitude to get significantly more score than an agent trained to maximize the score. / 7

We show that by just modifying the prompt given to the model with a single sentence, you can easily change the resulting behavior of the AI agent. Do you want an agent that maximizes gold, goes down the dungeon or kills more monsters? No problem, just tell the LLM to prefer that particular type of agent. / END

@AlphaSignalAI @DrJimFan @TalkRLPodcast @MLStreetTalk @TheTuringPost @_akhaliq @arankomatsuzaki @natolambert @Mila_Quebec @AIatMeta @ylecun @egrefen @_rockt

🔥 Congrats! Would one of the co-authors want to tell us about this work on @TalkRLPodcast ?

A new era of MMO bots are upon us! Great work.
