← Search

Alessandro Lazaric

61 accepted papers

2026

BFM-Zero: A Promptable Behavioral Foundation Model for Humanoid Control Using Unsupervised Reinforcement Learning

ICLR 2026poster

Building Behavioral Foundation Models (BFMs) for humanoid robots has the potential to unify diverse control tasks under a single, promptable generalist policy. However, existing approaches are either exclusively deployed on simulated humanoid characters, or specialized to specific tasks such as trac…

Cited by 0SourcecodeScholar
2026

Compositional Planning with Jumpy World Models

ICML 2026poster

The ability to plan with temporal abstractions is central to intelligent decision-making. Rather than reasoning over primitive actions, we study agents that compose pre-trained policies as temporally extended actions, enabling solutions to complex tasks that no constituent alone could solve. Such co…

Cited by 1SourceScholar
2026

TD-JEPA: Latent-predictive Representations for Zero-Shot Reinforcement Learning

ICLR 2026oral

Latent prediction–where agents learn by predicting their own latents–has emerged as a powerful paradigm for training general representations in machine learning. In reinforcement learning (RL), this approach has been explored to define auxiliary losses for a variety of settings, including reward-bas…

Cited by 0SourcecodeScholar
2025

Temporal Difference Flows

ICML 2025oral

Predictive models of the future are fundamental for an agent's ability to reason and plan. A common strategy learns a world model and unrolls it step-by-step at inference, where small errors can rapidly compound. Geometric Horizon Models (GHMs) offer a compelling alternative by directly making predi…

Cited by 0SourcePDFScholar
2025

Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Models

ICLR 2025poster

Unsupervised reinforcement learning (RL) aims at pre-training models that can solve a wide range of downstream tasks in complex environments. Despite recent advancements, existing approaches suffer from several limitations: they may require running an RL process on each task to achieve a satisfactor…

Cited by 4SourcePDFScholar
2024

Fast Imitation via Behavior Foundation Models

ICLR 2024spotlight

Imitation learning (IL) aims at producing agents that can imitate any behavior given a few expert demonstrations. Yet existing approaches require many demonstrations and/or running (online or offline) reinforcement learning (RL) algorithms for each new imitation task. Here we show that recent RL fou…

Cited by 11SourcePDFScholar
2024

Simple Ingredients for Offline Reinforcement Learning

ICML 2024poster

Offline reinforcement learning algorithms have proven effective on datasets highly connected to the target downstream task. Yet, by leveraging a novel testbed (MOOD) in which trajectories come from heterogeneous sources, we show that existing methods struggle with diverse data: their performance con…

Cited by 4SourcePDFScholar
2023

Contextual bandits with concave rewards, and an application to fair ranking

ICLR 2023poster

We consider Contextual Bandits with Concave Rewards (CBCR), a multi-objective bandit problem where the desired trade-off between the rewards is defined by a known concave objective function, and the reward vector depends on an observed stochastic context. We present the first algorithm with provably…

Cited by 4SourcePDFScholar
2023

Layered State Discovery for Incremental Autonomous Exploration

ICML 2023poster

We study the autonomous exploration (AX) problem proposed by Lim & Auer (2012). In this setting, the objective is to discover a set of $\epsilon$-optimal policies reaching a set $\mathcal{S}\_L^{\rightarrow}$ of incrementally $L$-controllable states. We introduce a novel layered decomposition of the…

Cited by 0SourcePDFScholar
2023

Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policies

ICLR 2023poster

We consider infinite-horizon discounted Markov decision processes and study the convergence rates of the natural policy gradient (NPG) and the Q-NPG methods with the log-linear policy class. Using the compatible function approximation framework, both methods with log-linear policies can be written a…

Cited by 51SourcePDFScholar
2023

On the Complexity of Representation Learning in Contextual Linear Bandits

AISTATS 2023poster

In contextual linear bandits, the reward function is assumed to be a linear combination of an unknown reward vector and a given embedding of context-arm pairs. In practice, the embedding is often learned at the same time as the reward vector, thus leading to an online representation learning problem…

Cited by 2SourcePDFScholar
2022

A Reduction-Based Framework for Conservative Bandits and Reinforcement Learning

ICLR 2022poster

We study bandits and reinforcement learning (RL) subject to a conservative constraint where the agent is asked to perform at least as well as a given baseline policy. This setting is particular relevant in real-world domains including digital marketing, healthcare, production, finance, etc. In this…

Cited by 7SourcePDFScholar
2022

A general sample complexity analysis of vanilla policy gradient

AISTATS 2022poster

We adapt recent tools developed for the analysis of Stochastic Gradient Descent (SGD) in non-convex optimization to obtain convergence and sample complexity guarantees for the vanilla policy gradient (PG). Our only assumptions are that the expected return is smooth w.r.t. the policy parameters, that…

Cited by 83SourcePDFScholar
2022

Adaptive Multi-Goal Exploration

AISTATS 2022poster

We introduce a generic strategy for provably efficient multi-goal exploration. It relies on AdaGoal, a novel goal selection scheme that leverages a measure of uncertainty in reaching states to adaptively target goals that are neither too difficult nor too easy. We show how AdaGoal can be used to tac…

Cited by 5SourcePDFScholar
2022

Direct then Diffuse: Incremental Unsupervised Skill Discovery for State Covering and Goal Reaching

ICLR 2022poster

Learning meaningful behaviors in the absence of reward is a difficult problem in reinforcement learning. A desirable and challenging unsupervised objective is to learn a set of diverse skills that provide a thorough coverage of the state space while being directed, i.e., reliably reaching distinct r…

Cited by 25SourcePDFScholar
2022

Learning Goal-Conditioned Policies Offline with Self-Supervised Reward Shaping

CoRL 2022poster

Developing agents that can execute multiple skills by learning from pre-collected datasets is an important problem in robotics, where online interaction with the environment is extremely time-consuming. Moreover, manually designing reward functions for every single desired skill is prohibitive. Prio…

Cited by 22SourcecodeScholar
2022

Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learning

ICLR 2022poster

We present DrQ-v2, a model-free reinforcement learning (RL) algorithm for visual continuous control. DrQ-v2 builds on DrQ, an off-policy actor-critic approach that uses data augmentation to learn directly from pixels. We introduce several improvements that yield state-of-the-art results on the DeepM…

2022

Scalable Representation Learning in Linear Contextual Bandits with Constant Regret Guarantees

NeurIPS 2022accept

We study the problem of representation learning in stochastic contextual linear bandits. While the primary concern in this domain is usually to find \textit{realizable} representations (i.e., those that allow predicting the reward function at any context-action pair exactly), it has been recently sh…

Cited by 7SourcePDFScholar
2022

Temporal abstractions-augmented temporally contrastive learning: An alternative to the Laplacian in RL

UAI 2022poster

In reinforcement learning, the graph Laplacian has proved to be a valuable tool in the task-agnostic setting, with applications ranging from skill discovery to reward shaping. Recently, learning the Laplacian representation has been framed as the optimization of a temporally-contrastive objective to…

Cited by 8SourcePDFScholar
2022

Top K Ranking for Multi-Armed Bandit with Noisy Evaluations

AISTATS 2022poster

We consider a multi-armed bandit setting where, at the beginning of each round, the learner receives noisy independent, and possibly biased, evaluations of the true reward of each arm and it selects $K$ arms with the objective of accumulating as much reward as possible over $T$ rounds. Under the ass…

Cited by 8SourcePDFScholar
2021

A Provably Efficient Sample Collection Strategy for Reinforcement Learning

NeurIPS 2021spotlight

One of the challenges in online reinforcement learning (RL) is that the agent needs to trade off the exploration of the environment and the exploitation of the samples to optimize its behavior. Whether we optimize for regret, sample complexity, state-space coverage or model estimation, we need to st…

Cited by 21SourcePDFScholar
2021

Leveraging Good Representations in Linear Contextual Bandits

ICML 2021spotlight

The linear contextual bandit literature is mostly focused on the design of efficient learning algorithms for a given representation. However, a contextual bandit problem may admit multiple linear representations, each one with different characteristics that directly impact the regret of the learning…

Cited by 35SourcePDFScholar
2021

Reinforcement Learning in Linear MDPs: Constant Regret and Representation Selection

NeurIPS 2021poster

We study the role of the representation of state-action value functions in regret minimization in finite-horizon Markov Decision Processes (MDPs) with linear structure. We first derive a necessary condition on the representation, called universally spanning optimal features (UNISOFT), to achieve con…

Cited by 23SourcePDFScholar
2021

Reinforcement Learning with Prototypical Representations

ICML 2021spotlight

Learning effective representations in image-based environments is crucial for sample efficient Reinforcement Learning (RL). Unfortunately, in RL, representation learning is confounded with the exploratory experience of the agent – learning a useful representation requires diverse data, while effecti…

2021

Stochastic Shortest Path: Minimax, Parameter-Free and Towards Horizon-Free Regret

NeurIPS 2021spotlight

We study the problem of learning in the stochastic shortest path (SSP) setting, where an agent seeks to minimize the expected cost accumulated before reaching a goal state. We design a novel model-based algorithm EB-SSP that carefully skews the empirical transitions and perturbs the empirical costs…

Cited by 37SourcePDFScholar
2020

A Novel Confidence-Based Algorithm for Structured Bandits

AISTATS 2020poster

We study finite-armed stochastic bandits where the rewards of each arm might be correlated to those of other arms. We introduce a novel phased algorithm that exploits the given structure to build confidence sets over the parameters of the true bandit problem and rapidly discard all sub-optimal arms.…

Cited by 17SourcePDFScholar
2020

A single algorithm for both restless and rested rotting bandits

AISTATS 2020poster

In many application domains (e.g., recommender systems, intelligent tutoring systems), the rewards associated to the available actions tend to decrease over time. This decay is either caused by the actions executed in the past (e.g., a user may get bored when songs of the same genre are recommended…

2020

Active Model Estimation in Markov Decision Processes

UAI 2020poster

We study the problem of efficient exploration in order to learn an accurate model of an environment, modeled as a Markov decision process (MDP). Efficient exploration in this problem requires the agent to identify the regions in which estimating the model is more difficult and then exploit this know…

Cited by 31SourcePDFScholar
2020

Adversarial Attacks on Linear Contextual Bandits

NeurIPS 2020poster

Contextual bandit algorithms are applied in a wide range of domains, from advertising to recommender systems, from clinical trials to education. In many of these domains, malicious agents may have incentives to force a bandit algorithm into a desired behavior For instance, an unscrupulous ad publish…

Cited by 68SourcePDFScholar
2020

An Asymptotically Optimal Primal-Dual Incremental Algorithm for Contextual Linear Bandits

NeurIPS 2020poster

In the contextual linear bandit setting, algorithms built on the optimism principle fail to exploit the structure of the problem and have been shown to be asymptotically suboptimal. In this paper, we follow recent approaches of deriving asymptotically optimal algorithms from problem-dependent regret…

Cited by 51SourcePDFScholar
2020

Conservative Exploration in Reinforcement Learning

AISTATS 2020poster

While learning in an unknown Markov Decision Process (MDP), an agent should trade off exploration to discover new information about the MDP, and exploitation of the current knowledge to maximize the reward. Although the agent will eventually learn a good or optimal policy, there is no guarantee on t…

Cited by 34SourcePDFScholar
2020

Efficient Optimistic Exploration in Linear-Quadratic Regulators via Lagrangian Relaxation

ICML 2020poster

We study the exploration-exploitation dilemma in the linear quadratic regulator (LQR) setting. Inspired by the extended value iteration algorithm used in optimistic algorithms for finite MDPs, we propose to relax the optimistic optimization of \ofulq and cast it into a constrained \emph{extended} LQ…

Cited by 46SourcePDFScholar
2020

Frequentist Regret Bounds for Randomized Least-Squares Value Iteration

AISTATS 2020poster

We consider the exploration-exploitation dilemma in finite-horizon reinforcement learning (RL). When the state space is large or continuous, traditional tabular approaches are unfeasible and some form of function approximation is mandatory. In this paper, we introduce an optimistically-initialized v…

Cited by 161SourcePDFScholar
2020

Improved Sample Complexity for Incremental Autonomous Exploration in MDPs

NeurIPS 2020oral

We study the problem of exploring an unknown environment when no reward function is provided to the agent. Building on the incremental exploration setting introduced by Lim and Auer (2012), we define the objective of learning the set of $\epsilon$-optimal goal-conditioned policies attaining all stat…

Cited by 17SourcePDFScholar
2020

Learning Near Optimal Policies with Low Inherent Bellman Error

ICML 2020poster

We study the exploration problem with approximate linear action-value functions in episodic reinforcement learning under the notion of low inherent Bellman error, a condition normally employed to show convergence of approximate value iteration. First we relate this condition to other common framewor…

Cited by 266SourcePDFScholar
2020

Near-linear time Gaussian process optimization with adaptive batching and resparsification

ICML 2020poster

Gaussian processes (GP) are one of the most successful frameworks to model uncertainty. However, GP optimization (e.g., GP-UCB) suffers from major scalability issues. Experimental time grows linearly with the number of evaluations, unless candidates are selected in batches (e.g., using GP-BUCB) and…

2020

No-Regret Exploration in Goal-Oriented Reinforcement Learning

ICML 2020poster

Many popular reinforcement learning problems (e.g., navigation in a maze, some Atari games, mountain car) are instances of the episodic setting under its stochastic shortest path (SSP) formulation, where an agent has to achieve a goal state while minimizing the cumulative cost. Despite the popularit…

Cited by 48SourcePDFScholar
2020

Provably Efficient Reward-Agnostic Navigation with Linear Value Iteration

NeurIPS 2020poster

There has been growing progress on theoretical analyses for provably efficient learning in MDPs with linear function approximation, but much of the existing work has made strong assumptions to enable exploration by conventional exploration frameworks. Typically these assumptions are stronger than wh…

Cited by 71SourcePDFScholar
2019

A Structured Prediction Approach for Generalization in Cooperative Multi-Agent Reinforcement Learning

NeurIPS 2019spotlight

Effective coordination is crucial to solve multi-agent collaborative (MAC) problems. While centralized reinforcement learning methods can optimally solve small MAC instances, they do not scale to large problems and they fail to generalize to scenarios different from those seen during training. In t…

2019

Exploration Bonus for Regret Minimization in Discrete and Continuous Average Reward MDPs

NeurIPS 2019poster

The exploration bonus is an effective approach to manage the exploration-exploitation trade-off in Markov Decision Processes (MDPs). While it has been analyzed in infinite-horizon discounted and finite-horizon problems, we focus on designing and analysing the exploration bonus in the more challengin…

2019

Limiting Extrapolation in Linear Approximate Value Iteration

NeurIPS 2019poster

We study linear approximate value iteration (LAVI) with a generative model. While linear models may accurately represent the optimal value function using a few parameters, several empirical and theoretical studies show the combination of least-squares projection with the Bellman operator may be expa…

Cited by 38SourcePDFScholar
2019

Regret Bounds for Learning State Representations in Reinforcement Learning

NeurIPS 2019poster

We consider the problem of online reinforcement learning when several state representations (mapping histories to a discrete state space) are available to the learning agent. At least one of these representations is assumed to induce a Markov decision process (MDP), and the performance of the agent…

Cited by 16SourcePDFScholar
2019

Rotting bandits are no harder than stochastic ones

AISTATS 2019poster

In stochastic multi-armed bandits, the reward distribution of each arm is assumed to be stationary. This assumption is often violated in practice (e.g., in recommendation systems), where the reward of an arm may change whenever is selected, i.e., rested bandit setting. In this paper, we consider the…

Cited by 73SourcePDFScholar
2018

Efficient Bias-Span-Constrained Exploration-Exploitation in Reinforcement Learning

ICML 2018oral

We introduce SCAL, an algorithm designed to perform efficient exploration-exploration in any unknown weakly-communicating Markov Decision Process (MDP) for which an upper bound c on the span of the optimal bias function is known. For an MDP with $S$ states, $A$ actions and $\Gamma \leq S$ possible n…

2018

Fighting Boredom in Recommender Systems with Linear Reinforcement Learning

NeurIPS 2018poster

A common assumption in recommender systems (RS) is the existence of a best fixed recommendation strategy. Such strategy may be simple and work at the item level (e.g., in multi-armed bandit it is assumed one best fixed arm/item exists) or implement more sophisticated RS (e.g., the objective of A/B t…

Cited by 57SourcePDFScholar
2018

Improved Regret Bounds for Thompson Sampling in Linear Quadratic Control Problems

ICML 2018oral

Thompson sampling (TS) is an effective approach to trade off exploration and exploration in reinforcement learning. Despite its empirical success and recent advances, its theoretical analysis is often limited to the Bayesian setting, finite state-action spaces, or finite-horizon problems. In this pa…

Cited by 115SourcePDFScholar
2018

Improved large-scale graph learning through ridge spectral sparsification

ICML 2018oral

The representation and learning benefits of methods based on graph Laplacians, such as Laplacian smoothing or harmonic function solution for semi-supervised learning (SSL), are empirically and theoretically well supported. Nonetheless, the exact versions of these methods scale poorly with the number…

Cited by 46SourcePDFScholar
2018

Near Optimal Exploration-Exploitation in Non-Communicating Markov Decision Processes

NeurIPS 2018spotlight

While designing the state space of an MDP, it is common to include states that are transient or not reachable by any policy (e.g., in mountain car, the product space of speed and position contains configurations that are not physically reachable). This results in weakly-communicating or multi-chain…

2017

Active Learning for Accurate Estimation of Linear Models

ICML 2017poster

We explore the sequential decision making problem where the goal is to estimate uniformly well a number of linear models, given a shared budget of random contexts independently sampled from a known distribution. The decision maker must query one of the linear models for each incoming context, and re…

Cited by 19SourcePDFScholar
2017

Distributed Adaptive Sampling for Kernel Matrix Approximation

AISTATS 2017poster

Most kernel-based methods, such as kernel regression, kernel PCA, ICA, or $k$-means clustering, do not scale to large datasets, because constructing and storing the kernel matrix $K_n$ requires at least $O(n^2)$ time and space for $n$ samples. Recent works (Alaoui 2014, Musco 2016) show that samplin…

Cited by 40SourcePDFScholar
2017

Efficient Second-Order Online Kernel Learning with Adaptive Embedding

NeurIPS 2017poster

Online kernel learning (OKL) is a flexible framework to approach prediction problems, since the large approximation space provided by reproducing kernel Hilbert spaces can contain an accurate function for the problem. Nonetheless, optimizing over this space is computationally expensive. Not only fir…

Cited by 45SourcePDFScholar
2017

Regret Minimization in MDPs with Options without Prior Knowledge

NeurIPS 2017spotlight

The option framework integrates temporal abstraction into the reinforcement learning model through the introduction of macro-actions (i.e., options). Recent works leveraged on the mapping of Markov decision processes (MDPs) with options to semi-MDPs (SMDPs) and introduced SMDP-versions of exploratio…

Cited by 34SourcePDFScholar
2017

Second-Order Kernel Online Convex Optimization with Adaptive Sketching

ICML 2017poster

Kernel online convex optimization (KOCO) is a framework combining the expressiveness of non-parametric kernel models with the regret guarantees of online learning. First-order KOCO methods such as functional gradient descent require only $O(t)$ time and space per iteration, and, when the only inform…

Cited by 48SourcePDFScholar
2017

Trading off Rewards and Errors in Multi-Armed Bandits

AISTATS 2017poster

In multi-armed bandits, the most common objective is the maximization of the cumulative reward. Alternative settings include active exploration, where a learner tries to gain accurate estimates of the rewards of all arms. While these objectives are contrasting, in many scenarios it is desirable to t…

Cited by 35SourcePDFScholar
2016

Improved Learning Complexity in Combinatorial Pure Exploration Bandits

AISTATS 2016poster

We study the problem of combinatorial pure exploration in the stochastic multi-armed bandit problem. We first construct a new measure of complexity that provably characterizes the learning performance of the algorithms we propose for the fixed confidence and the fixed budget setting. We show that th…

Cited by 48SourcePDFScholar