← Search

Matteo Pirotta

39 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

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

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

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

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 Kernel-Based Approach to Non-Stationary Reinforcement Learning in Metric Spaces

AISTATS 2021poster

In this work, we propose KeRNS: an algorithm for episodic reinforcement learning in non-stationary Markov Decision Processes (MDPs) whose state-action set is endowed with a metric. Using a non-parametric model of the MDP built with time-dependent kernels, we prove a regret bound that scales with the…

Cited by 48SourcePDFScholar
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

Kernel-Based Reinforcement Learning: A Finite-Time Analysis

ICML 2021spotlight

We consider the exploration-exploitation dilemma in finite-horizon reinforcement learning problems whose state-action space is endowed with a metric. We introduce Kernel-UCBVI, a model-based optimistic algorithm that leverages the smoothness of the MDP and a non-parametric kernel estimator of the re…

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

Local Differential Privacy for Regret Minimization in Reinforcement Learning

NeurIPS 2021poster

Reinforcement learning algorithms are widely used in domains where it is desirable to provide a personalized service. In these domains it is common that user data contains sensitive information that needs to be protected from third parties. Motivated by this, we study privacy in the context of finit…

Cited by 52SourcePDFScholar
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

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

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

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

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
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

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
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

Importance Weighted Transfer of Samples in Reinforcement Learning

ICML 2018oral

We consider the transfer of experience samples (i.e., tuples < s, a, s’, r >) in reinforcement learning (RL), collected from a set of source tasks to improve the learning process in a given target task. Most of the related approaches focus on selecting the most relevant source samples for solving th…

Cited by 68SourcePDFScholar
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…

2018

Stochastic Variance-Reduced Policy Gradient

ICML 2018oral

In this paper, we propose a novel reinforcement-learning algorithm consisting in a stochastic variance-reduced version of policy gradient for solving Markov Decision Processes (MDPs). Stochastic variance-reduced gradient (SVRG) methods have proven to be very successful in supervised learning. Howeve…

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