← Search

Adrià Puigdomènech Badia

8 accepted papers

2022

Retrieval-Augmented Reinforcement Learning

ICML 2022spotlight

Most deep reinforcement learning (RL) algorithms distill experience into parametric behavior policies or value functions via gradient updates. While effective, this approach has several disadvantages: (1) it is computationally expensive, (2) it can take many updates to integrate experiences into the…

Cited by 59SourcePDFScholar
2022

The CLRS Algorithmic Reasoning Benchmark

ICML 2022spotlight

Learning representations of algorithms is an emerging area of machine learning, seeking to bridge concepts from neural networks with classical algorithms. Several important works have investigated whether neural networks can effectively reason like algorithms, typically by learning to execute them.…

2020

Agent57: Outperforming the Atari Human Benchmark

ICML 2020poster

Atari games have been a long-standing benchmark in the reinforcement learning (RL) community for the past decade. This benchmark was proposed to test general competency of RL algorithms. Previous work has achieved good average performance by doing outstandingly well on many games of the set, but ver…

Cited by 759SourcePDFScholar
2020

MEMO: A Deep Network for Flexible Combination of Episodic Memories

ICLR 2020poster

Recent research developing neural network architectures with external memory have often used the benchmark bAbI question and answering dataset which provides a challenging number of tasks requiring reasoning. Here we employed a classic associative inference task from the human neuroscience literatur…

Cited by 0SourceScholar
2020

Never Give Up: Learning Directed Exploration Strategies

ICLR 2020poster

We propose a reinforcement learning agent to solve hard exploration games by learning a range of directed exploratory policies. We construct an episodic memory-based intrinsic reward using k-nearest neighbors over the agent's recent experience to train the directed exploratory policies, thereby enco…

Cited by 410SourceScholar
2019

Generalization of Reinforcement Learners with Working and Episodic Memory

NeurIPS 2019poster

Memory is an important aspect of intelligence and plays a role in many deep reinforcement learning models. However, little progress has been made in understanding when specific memory systems help more than others and how well they generalize. The field also has yet to see a prevalent consistent and…

2017

Imagination-Augmented Agents for Deep Reinforcement Learning

NeurIPS 2017oral

We introduce Imagination-Augmented Agents (I2As), a novel architecture for deep reinforcement learning combining model-free and model-based aspects. In contrast to most existing model-based reinforcement learning and planning methods, which prescribe how a model should be used to arrive at a polic…

Cited by 763SourcePDFScholar
2017

Neural Episodic Control

ICML 2017poster

Deep reinforcement learning methods attain super-human performance in a wide range of environments. Such methods are grossly inefficient, often taking orders of magnitudes more data than humans to achieve reasonable performance. We propose Neural Episodic Control: a deep reinforcement learning agent…

Cited by 448SourcePDFScholar