In plain words: The agent turns game images into tokens and uses a Transformer to predict what happens next, letting it practice in imagination. With two hours of Atari play, it beat humans on 10 of 26 games, ahead of earlier agents that don't plan ahead.
Abstract · Transformers are Sample-Efficient World Models
Deep reinforcement learning agents are notoriously sample inefficient, which considerably limits their application to real-world problems. Recently, many model-based methods have been designed to address this issue, with learning in the imagination of a world model being one of the most prominent approaches. However, while virtually unlimited interaction with a simulated environment sounds appealing, the world model has to be accurate over extended periods of time. Motivated by the success of Transformers in sequence modeling tasks, we introduce IRIS, a data-efficient agent that learns in a world model composed of a discrete autoencoder and an autoregressive Transformer. With the equivalent of only two hours of gameplay in the Atari 100k benchmark, IRIS achieves a mean human normalized score of 1.046, and outperforms humans on 10 out of 26 games, setting a new state of the art for methods without lookahead search. To foster future research on Transformers and world models for sample-efficient reinforcement learning, we release our code and models at https://github.com/eloialonso/iris.
Vincent Micheli, Eloi Alonso, François Fleuret
arXiv:2209.00588 · cs.LG, cs.AI, cs.CV · submitted Sep 1, 2022 · updated Mar 1, 2023
abstract · pdf · html · ICLR 2023 (notable top 5%)
Does that include Montezuma's Revenge? Because that's the Atari game RL agents completely fail at.
Edit: Yup, Atari 100k doesn't include MR.
https://twitter.com/arankomatsuzaki/status/14553550319019089...