← Search

Tom Zahavy

22 accepted papers

2026

Position: LLMs can't jump

ICML 2026poster

How do we fundamentally discover new things? In a letter to Maurice Solovine, Albert Einstein conceptualized discovery as a cyclical process involving an intuitive 'jump' from sensory experience to axioms, followed by logical deduction. While Generative AI has mastered Induction (statistical pattern…

Cited by 0SourceScholar
2025

Generating Creative Chess Puzzles

NeurIPS 2025poster

While Generative AI rapidly advances in various domains, generating truly creative, aesthetic, and counter-intuitive outputs remains a challenge. This paper presents an approach to tackle these difficulties in the domain of chess puzzles. We start by benchmarking Generative AI architectures, and the…

Cited by 0SourceScholar
2025

Mastering Board Games by External and Internal Planning with Language Models

ICML 2025spotlight

Advancing planning and reasoning capabilities of Large Language Models (LLMs) is one of the key prerequisites towards unlocking their potential for performing reliably in complex and impactful domains. In this paper, we aim to demonstrate this across board games (Chess, Fischer Random / Chess960, Co…

Cited by 6SourcePDFScholar
2023

Discovering Evolution Strategies via Meta-Black-Box Optimization

ICLR 2023poster

Optimizing functions without access to gradients is the remit of black-box meth- ods such as evolution strategies. While highly general, their learning dynamics are often times heuristic and inflexible — exactly the limitations that meta-learning can address. Hence, we propose to discover effective…

2023

Discovering Policies with DOMiNO: Diversity Optimization Maintaining Near Optimality

ICLR 2023poster

In this work we propose a Reinforcement Learning (RL) agent that can discover complex behaviours in a rich environment with a simple reward function. We define diversity in terms of state-action occupancy measures, since policies with different occupancy measures visit different states on average. M…

Cited by 42SourcePDFScholar
2023

Optimistic Meta-Gradients

NeurIPS 2023poster

We study the connection between gradient-based meta-learning and convex optimisation. We observe that gradient descent with momentum is a special case of meta-gradients, and building on recent results in optimisation, we prove convergence rates for meta learning in the single task setting. While a m…

Cited by 4SourcePDFScholar
2023

ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPs

ICML 2023poster

In recent years, reinforcement learning (RL) has been applied to real-world problems with increasing success. Such applications often require to put constraints on the agent's behavior. Existing algorithms for constrained RL (CRL) rely on gradient descent-ascent, but this approach comes with a cavea…

Cited by 23SourcePDFScholar
2022

Bootstrapped Meta-Learning

ICLR 2022oral

Meta-learning empowers artificial intelligence to increase its efficiency by learning how to learn. Unlocking this potential involves overcoming a challenging meta-optimisation problem. We propose an algorithm that tackles this problem by letting the meta-learner teach itself. The algorithm first bo…

Cited by 83SourcePDFScholar
2022

Palm up: Playing in the Latent Manifold for Unsupervised Pretraining

NeurIPS 2022accept

Large and diverse datasets have been the cornerstones of many impressive advancements in artificial intelligence. Intelligent creatures, however, learn by interacting with the environment, which changes the input sensory signals and the state of the environment. In this work, we aim to bring the bes…

Cited by 9SourcePDFScholar
2021

Balancing Constraints and Rewards with Meta-Gradient D4PG

ICLR 2021poster

Deploying Reinforcement Learning (RL) agents to solve real-world applications often requires satisfying complex system constraints. Often the constraint thresholds are incorrectly set due to the complex nature of a system or the inability to verify the thresholds offline (e.g, no simulator or reason…

Cited by 26SourcePDFScholar
2021

Discovering a set of policies for the worst case reward

ICLR 2021spotlight

We study the problem of how to construct a set of policies that can be composed together to solve a collection of reinforcement learning tasks. Each task is a different reward function defined as a linear combination of known features. We consider a specific class of policy compositions which we ca…

Cited by 29SourcePDFScholar
2021

Discovery of Options via Meta-Learned Subgoals

NeurIPS 2021poster

Temporal abstractions in the form of options have been shown to help reinforcement learning (RL) agents learn faster. However, despite prior work on this topic, the problem of discovering options through interaction with an environment remains a challenge. In this paper, we introduce a novel meta-gr…

Cited by 44SourcePDFScholar
2021

Emphatic Algorithms for Deep Reinforcement Learning

ICML 2021spotlight

Off-policy learning allows us to learn about possible policies of behavior from experience generated by a different behavior policy. Temporal difference (TD) learning algorithms can become unstable when combined with function approximation and off-policy sampling—this is known as the “deadly triad”.…

2021

Online Limited Memory Neural-Linear Bandits with Likelihood Matching

ICML 2021spotlight

We study neural-linear bandits for solving problems where {\em both} exploration and representation learning play an important role. Neural-linear bandits harnesses the representation power of Deep Neural Networks (DNNs) and combines it with efficient exploration mechanisms by leveraging uncertainty…

2020

A Self-Tuning Actor-Critic Algorithm

NeurIPS 2020poster

Reinforcement learning algorithms are highly sensitive to the choice of hyperparameters, typically requiring significant manual effort to identify hyperparameters that perform well on a new domain. In this paper, we take a step towards addressing this issue by using metagradients to automatically ad…

2020

Unknown mixing times in apprenticeship and reinforcement learning

UAI 2020poster

We derive and analyze learning algorithms for apprenticeship learning, policy evaluation and policy gradient for average reward criteria. Existing algorithms explicitly require an upper bound on the mixing time. In contrast, we build on ideas from Markov chain theory and derive sampling algorithms t…

Cited by 6SourcePDFScholar
2018

Ensemble Robustness and Generalization of Stochastic Deep Learning Algorithms

ICLR 2018workshop

The question why deep learning algorithms generalize so well has attracted increasing research interest. However, most of the well-established approaches, such as hypothesis capacity, stability or sparseness, have not provided complete explanations (Zhang et al., 2016; Kawaguchi et al., 2017). In th…

Cited by 21SourceScholar
2018

Learn What Not to Learn: Action Elimination with Deep Reinforcement Learning

NeurIPS 2018poster

Learning how to act when there are many available actions in each state is a challenging task for Reinforcement Learning (RL) agents, especially when many of the actions are redundant or irrelevant. In such cases, it is easier to learn which actions not to take. In this work, we propose the Action-E…

2017

Shallow Updates for Deep Reinforcement Learning

NeurIPS 2017poster

Deep reinforcement learning (DRL) methods such as the Deep Q-Network (DQN) have achieved state-of-the-art results in a variety of challenging, high-dimensional domains. This success is mainly attributed to the power of deep neural networks to learn rich domain representations for approximating the v…

Cited by 54SourcePDFScholar