← Search

Marcello Restelli

79 accepted papers

2026

From Parameters to Behaviors: Unsupervised Compression of the Policy Space

ICLR 2026poster

Despite its recent successes, Deep Reinforcement Learning (DRL) is notoriously sample-inefficient. We argue that this inefficiency stems from the standard practice of optimizing policies directly in the high-dimensional and highly redundant parameter space $\\Theta$. This challenge is greatly compou…

Cited by 0SourcecodeScholar
2025

Achieving $\widetilde{\mathcal{O}}(\sqrt{T})$ Regret in Average-Reward POMDPs with Known Observation Models

AISTATS 2025poster

We tackle average-reward infinite-horizon POMDPs with an unknown transition model but a known observation model, a setting that has been previously addressed in two limiting ways: (i) frequentist methods relying on suboptimal stochastic policies having a minimum probability of choosing each action,…

Cited by 0SourceScholar
2025

Efficient Exploitation of Hierarchical Structure in Sparse Reward Reinforcement Learning

AISTATS 2025poster

We study goal-conditioned Hierarchical Reinforcement Learning (HRL), where a high-level agent instructs sub-goals to a low-level agent. Under the assumption of a sparse reward function and known hierarchical decomposition, we propose a new algorithm to learn optimal hierarchical policies. Our algori…

Cited by 0SourceScholar
2025

Enhancing Diversity In Parallel Agents: A Maximum State Entropy Exploration Story

ICML 2025poster

Parallel data collection has redefined Reinforcement Learning (RL), unlocking unprecedented efficiency and powering breakthroughs in large-scale real-world applications. In this paradigm, $N$ identical agents operate in $N$ replicas of an environment simulator, accelerating data collection by a fact…

Cited by 0SourcePDFScholar
2024

A Retrospective on the Robot Air Hockey Challenge: Benchmarking Robust, Reliable, and Safe Learning Techniques for Real-world Robotics

NeurIPS 2024poster

Machine learning methods have a groundbreaking impact in many application domains, but their application on real robotic platforms is still limited. Despite the many challenges associated with combining machine learning technology with robotics, robot learning remains one of the most promising direc…

Cited by 0SourcePDFScholar
2024

Autoregressive Bandits

AISTATS 2024poster

Autoregressive processes naturally arise in a large variety of real-world scenarios, including stock markets, sales forecasting, weather prediction, advertising, and pricing. When facing a sequential decision-making problem in such a context, the temporal dependence between consecutive observations…

2024

Bandits with Ranking Feedback

NeurIPS 2024poster

In this paper, we introduce a novel variation of multi-armed bandits called bandits with ranking feedback. Unlike traditional bandits, this variation provides feedback to the learner that allows them to rank the arms based on previous pulls, without quantifying numerically the difference in performa…

Cited by 1SourcePDFScholar
2024

Best Arm Identification for Stochastic Rising Bandits

ICML 2024spotlight

Stochastic Rising Bandits (SRBs) model sequential decision-making problems in which the expected reward of the available options increases every time they are selected. This setting captures a wide range of scenarios in which the available options are learning entities whose performance improves (in…

2024

Exploiting Causal Graph Priors with Posterior Sampling for Reinforcement Learning

ICLR 2024poster

Posterior sampling allows exploitation of prior knowledge on the environment's transition dynamics to improve the sample efficiency of reinforcement learning. The prior is typically specified as a class of parametric distributions, the design of which can be cumbersome in practice, often resulting i…

Cited by 5SourcePDFScholar
2024

Factored-Reward Bandits with Intermediate Observations

ICML 2024poster

In several real-world sequential decision problems, at every step, the learner is required to select different actions. Every action affects a specific part of the system and generates an observable intermediate effect. In this paper, we introduce the Factored-Reward Bandits (FRBs), a novel setting…

Cited by 1SourcePDFScholar
2024

Graph-Triggered Rising Bandits

ICML 2024poster

In this paper, we propose a novel generalization of rested and restless bandits where the evolution of the arms' expected rewards is governed by a graph defined over the arms. An edge connecting a pair of arms $(i,j)$ represents the fact that a pull of arm $i$ *triggers* the evolution of arm $j$, an…

Cited by 4SourcePDFScholar
2024

How to Explore with Belief: State Entropy Maximization in POMDPs

ICML 2024poster

Recent works have studied *state entropy maximization* in reinforcement learning, in which the agent's objective is to learn a policy inducing high entropy over states visitation (Hazan et al., 2019). They typically assume full observability of the state of the system, so that the entropy of the obs…

Cited by 2SourcePDFScholar
2024

Local Linearity: the Key for No-regret Reinforcement Learning in Continuous MDPs

NeurIPS 2024poster

Achieving the no-regret property for Reinforcement Learning (RL) problems in continuous state and action-space environments is one of the major open problems in the field. Existing solutions either work under very specific assumptions or achieve bounds that are vacuous in some regimes. Furthermore,…

Cited by 0SourcePDFScholar
2024

No-Regret Reinforcement Learning in Smooth MDPs

ICML 2024poster

Obtaining no-regret guarantees for reinforcement learning (RL) in the case of problems with continuous state and/or action spaces is still one of the major open challenges in the field. Recently, a variety of solutions have been proposed, but besides very specific settings, the general problem remai…

Cited by 7SourcePDFScholar
2024

Online Markov Decision Processes Configuration with Continuous Decision Space

AAAI 2024technical

In this paper, we investigate the optimal online configuration of episodic Markov decision processes when the space of the possible configurations is continuous. Specifically, we study the interaction between a learner (referred to as the configurator) and an agent with a fixed, unknown policy, when…

Cited by 10SourcePDFScholar
2024

Optimal Multi-Fidelity Best-Arm Identification

NeurIPS 2024poster

In bandit best-arm identification, an algorithm is tasked with finding the arm with highest mean reward with a specified accuracy as fast as possible. We study multi-fidelity best-arm identification, in which the algorithm can choose to sample an arm at a lower fidelity (less accurate mean estimate)…

Cited by 4SourcePDFScholar
2024

Parameterized Projected Bellman Operator

AAAI 2024technical

Approximate value iteration (AVI) is a family of algorithms for reinforcement learning (RL) that aims to obtain an approximation of the optimal value function. Generally, AVI algorithms implement an iterated procedure where each step consists of (i) an application of the Bellman operator and (ii) a…

2024

Sub-optimal Experts mitigate Ambiguity in Inverse Reinforcement Learning

NeurIPS 2024poster

Inverse Reinforcement Learning (IRL) deals with the problem of deducing a reward function that explains the behavior of an expert agent who is assumed to act *optimally* in an underlying unknown task. Recent works have studied the IRL problem from the perspective of recovering the *feasible reward s…

Cited by 0SourcePDFScholar
2023

A Tale of Sampling and Estimation in Discounted Reinforcement Learning

AISTATS 2023poster

The most relevant problems in discounted reinforcement learning involve estimating the mean of a function under the stationary distribution of a Markov reward process, such as the expected return in policy evaluation, or the policy gradient in policy optimization. In practice, these estimates are pr…

Cited by 2SourcePDFScholar
2023

Distributional Policy Evaluation: a Maximum Entropy approach to Representation Learning

NeurIPS 2023poster

The Maximum Entropy (Max-Ent) framework has been effectively employed in a variety of Reinforcement Learning (RL) tasks. In this paper, we first propose a novel Max-Ent framework for policy evaluation in a distributional RL setting, named *Distributional Maximum Entropy Policy Evaluation* (D-Max-Ent…

Cited by 0SourcePDFScholar
2023

Dynamic Pricing with Volume Discounts in Online Settings

AAAI 2023technical

According to the main international reports, more pervasive industrial and business-process automation, thanks to machine learning and advanced analytic tools, will unlock more than 14 trillion USD worldwide annually by 2030. In the specific case of pricing problems, which constitute the class of pr…

Cited by 7SourcePDFScholar
2023

On the Relation between Policy Improvement and Off-Policy Minimum-Variance Policy Evaluation

UAI 2023poster

Off-policy methods are the basis of a large number of effective Policy Optimization (PO) algorithms. In this setting, Importance Sampling (IS) is typically employed for off-policy evaluation, with the goal of estimating the performance of a target policy, given samples collected with a different beh…

Cited by 2SourcePDFScholar
2023

Provably Efficient Causal Model-Based Reinforcement Learning for Systematic Generalization

AAAI 2023technical

In the sequential decision making setting, an agent aims to achieve systematic generalization over a large, possibly infinite, set of environments. Such environments are modeled as discrete Markov decision processes with both states and actions represented through a feature vector. The underlying st…

Cited by 19SourcePDFScholar
2023

Simultaneously Updating All Persistence Values in Reinforcement Learning

AAAI 2023technical

In Reinforcement Learning, the performance of learning agents is highly sensitive to the choice of time discretization. Agents acting at high frequencies have the best control opportunities, along with some drawbacks, such as possible inefficient exploration and vanishing of the action advantages. T…

2023

Tight Performance Guarantees of Imitator Policies with Continuous Actions

AAAI 2023technical

Behavioral Cloning (BC) aims at learning a policy that mimics the behavior demonstrated by an expert. The current theoretical understanding of BC is limited to the case of finite actions. In this paper, we study BC with the goal of providing theoretical guarantees on the performance of the imitator…

Cited by 5SourcePDFScholar
2023

Towards Theoretical Understanding of Inverse Reinforcement Learning

ICML 2023oral

Inverse reinforcement learning (IRL) denotes a powerful family of algorithms for recovering a reward function justifying the behavior demonstrated by an expert agent. A well-known limitation of IRL is the ambiguity in the choice of the reward function, due to the existence of multiple rewards that e…

Cited by 25SourcePDFScholar
2023

Truncating Trajectories in Monte Carlo Policy Evaluation: an Adaptive Approach

NeurIPS 2023poster

Policy evaluation via Monte Carlo (MC) simulation is at the core of many MC Reinforcement Learning (RL) algorithms (e.g., policy gradient methods). In this context, the designer of the learning system specifies an interaction budget that the agent usually spends by collecting trajectories of *fixed…

Cited by 1SourcePDFScholar
2023

Truncating Trajectories in Monte Carlo Reinforcement Learning

ICML 2023poster

In Reinforcement Learning (RL), an agent acts in an unknown environment to maximize the expected cumulative discounted sum of an external reward signal, i.e., the expected return. In practice, in many tasks of interest, such as policy optimization, the agent usually spends its interaction budget by…

Cited by 5SourcePDFScholar
2023

Wasserstein Actor-Critic: Directed Exploration via Optimism for Continuous-Actions Control

AAAI 2023technical

Uncertainty quantification has been extensively used as a means to achieve efficient directed exploration in Reinforcement Learning (RL). However, state-of-the-art methods for continuous actions still suffer from high sample complexity requirements. Indeed, they either completely lack strategies for…

2022

Balancing Sample Efficiency and Suboptimality in Inverse Reinforcement Learning

ICML 2022spotlight

We propose a novel formulation for the Inverse Reinforcement Learning (IRL) problem, which jointly accounts for the compatibility with the expert behavior of the identified reward and its effectiveness for the subsequent forward learning phase. Albeit quite natural, especially when the final goal is…

Cited by 9SourcePDFScholar
2022

Challenging Common Assumptions in Convex Reinforcement Learning

NeurIPS 2022accept

The classic Reinforcement Learning (RL) formulation concerns the maximization of a scalar reward function. More recently, convex RL has been introduced to extend the RL formulation to all the objectives that are convex functions of the state distribution induced by a policy. Notably, convex RL cover…

Cited by 27SourcePDFScholar
2022

Finite Sample Analysis of Mean-Volatility Actor-Critic for Risk-Averse Reinforcement Learning

AISTATS 2022poster

The goal in the standard reinforcement learning problem is to find a policy that optimizes the expected return. However, such an objective is not adequate in a lot of real-life applications, like finance, where controlling the uncertainty of the outcome is imperative. The mean-volatility objective p…

2022

Goal-Directed Planning via Hindsight Experience Replay

ICLR 2022poster

We consider the problem of goal-directed planning under a deterministic transition model. Monte Carlo Tree Search has shown remarkable performance in solving deterministic control problems. It has been extended from complex continuous domains through function approximators to bias the search of the…

Cited by 14SourcePDFScholar
2022

Lifelong Hyper-Policy Optimization with Multiple Importance Sampling Regularization

AAAI 2022technical

Learning in a lifelong setting, where the dynamics continually evolve, is a hard challenge for current reinforcement learning algorithms. Yet this would be a much needed feature for practical applications. In this paper, we propose an approach which learns a hyper-policy, whose input is time, that…

2022

Multi-Armed Bandit Problem with Temporally-Partitioned Rewards: When Partial Feedback Counts

IJCAI 2022poster

There is a rising interest in industrial online applications where data becomes available sequentially. Inspired by the recommendation of playlists to users where their preferences can be collected during the listening of the entire playlist, we study a novel bandit setting, namely Multi-Armed Bandi…

Cited by 5SourcePDFScholar
2022

Off-Policy Evaluation with Deficient Support Using Side Information

NeurIPS 2022accept

The Off-Policy Evaluation (OPE) problem consists in evaluating the performance of new policies from the data collected by another one. OPE is crucial when evaluating a new policy online is too expensive or risky. Many of the state-of-the-art OPE estimators are based on the Inverse Propensity Scoring…

Cited by 13SourcePDFScholar
2022

Reward-Free Policy Space Compression for Reinforcement Learning

AISTATS 2022poster

In reinforcement learning, we encode the potential behaviors of an agent interacting with an environment into an infinite set of policies, called policy space, typically represented by a family of parametric functions. Dealing with such a policy space is a hefty challenge, which often causes sample…

Cited by 7SourcePDFScholar
2022

Stochastic Rising Bandits

ICML 2022spotlight

This paper is in the field of stochastic Multi-Armed Bandits (MABs), i.e., those sequential selection techniques able to learn online using only the feedback given by the chosen option (a.k.a. arm). We study a particular case of the rested and restless bandits in which the arms’ expected payoff is m…

2022

The Importance of Non-Markovianity in Maximum State Entropy Exploration

ICML 2022oral

In the maximum state entropy exploration framework, an agent interacts with a reward-free environment to learn a policy that maximizes the entropy of the expected state visitations it is inducing. Hazan et al. (2019) noted that the class of Markovian stochastic policies is sufficient for the maximum…

Cited by 41SourcePDFScholar
2022

Unsupervised Reinforcement Learning in Multiple Environments

AAAI 2022technical

Several recent works have been dedicated to unsupervised reinforcement learning in a single environment, in which a policy is first pre-trained with unsupervised interactions, and then fine-tuned towards the optimal policy for several downstream supervised tasks defined over the same environment. Al…

2021

Learning in Non-Cooperative Configurable Markov Decision Processes

NeurIPS 2021poster

The Configurable Markov Decision Process framework includes two entities: a Reinforcement Learning agent and a configurator that can modify some environmental parameters to improve the agent's performance. This presupposes that the two actors have the same reward functions. What if the configurator…

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

Meta-Reinforcement Learning by Tracking Task Non-stationarity

IJCAI 2021poster

Many real-world domains are subject to a structured non-stationarity which affects the agent's goals and the environmental dynamics. Meta-reinforcement learning (RL) has been shown successful for training agents that quickly adapt to related tasks. However, most of the existing meta-RL algorithms fo…

2021

Policy Optimization as Online Learning with Mediator Feedback

AAAI 2021technical

Policy Optimization (PO) is a widely used approach to address continuous control tasks. In this paper, we introduce the notion of mediator feedback that frames PO as an online learning problem over the policy space. The additional available information, compared to the standard bandit feedback, allo…

Cited by 15SourcePDFScholar
2021

Provably Efficient Learning of Transferable Rewards

ICML 2021spotlight

The reward function is widely accepted as a succinct, robust, and transferable representation of a task. Typical approaches, at the basis of Inverse Reinforcement Learning (IRL), leverage on expert demonstrations to recover a reward function. In this paper, we study the theoretical properties of the…

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

Subgaussian and Differentiable Importance Sampling for Off-Policy Evaluation and Learning

NeurIPS 2021spotlight

Importance Sampling (IS) is a widely used building block for a large variety of off-policy estimation and learning algorithms. However, empirical and theoretical studies have progressively shown that vanilla IS leads to poor estimations whenever the behavioral and target policies are too dissimilar.…

2021

Task-Agnostic Exploration via Policy Gradient of a Non-Parametric State Entropy Estimate

AAAI 2021technical

In a reward-free environment, what is a suitable intrinsic objective for an agent to pursue so that it can learn an optimal task-agnostic exploration policy? In this paper, we argue that the entropy of the state distribution induced by finite-horizon trajectories is a sensible target. Especially, we…

2021

Time-variant variational transfer for value functions

UAI 2021poster

In most of the transfer learning approaches to reinforcement learning (RL) the distribution over the tasks is assumed to be stationary. Therefore, the target and source tasks are i.i.d. samples of the same distribution. Unfortunately, this assumption rarely holds in real-world conditions, e.g., due…

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

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

Balancing Learning Speed and Stability in Policy Gradient via Adaptive Exploration

AISTATS 2020poster

In many Reinforcement Learning (RL) applications, the goal is to find an optimal deterministic policy. However, most RL algorithms require the policy to be stochastic in order to avoid instabilities and perform a sufficient amount of exploration. Adjusting the level of stochasticity during the learn…

Cited by 21SourcePDFScholar
2020

Control Frequency Adaptation via Action Persistence in Batch Reinforcement Learning

ICML 2020poster

The choice of the control frequency of a system has a relevant impact on the ability of reinforcement learning algorithms to learn a highly performing policy. In this paper, we introduce the notion of action persistence that consists in the repetition of an action for a fixed number of decision step…

2020

Inverse Reinforcement Learning from a Gradient-based Learner

NeurIPS 2020poster

Inverse Reinforcement Learning addresses the problem of inferring an expert's reward function from demonstrations. However, in many applications, we not only have access to the expert's near-optimal behaviour, but we also observe part of her learning process. In this paper, we propose a new algorith…

Cited by 17SourcePDFScholar
2020

Risk-Averse Trust Region Optimization for Reward-Volatility Reduction

IJCAI 2020poster

The use of reinforcement learning in algorithmic trading is of growing interest, since it offers the opportunity of making profit through the development of autonomous artificial traders, that do not depend on hard-coded rules. In such a framework, keeping uncertainty under control is as important a…

Cited by 0SourcePDFScholar
2020

Sequential Transfer in Reinforcement Learning with a Generative Model

ICML 2020poster

We are interested in how to design reinforcement learning agents that provably reduce the sample complexity for learning new tasks by transferring knowledge from previously-solved ones. The availability of solutions to related problems poses a fundamental trade-off: whether to seek policies that are…

Cited by 31SourcePDFScholar
2020

Sharing Knowledge in Multi-Task Deep Reinforcement Learning

ICLR 2020poster

We study the benefit of sharing representations among tasks to enable the effective use of deep neural networks in Multi-Task Reinforcement Learning. We leverage the assumption that learning from different tasks, sharing common properties, is helpful to generalize the knowledge of them resulting in…

Cited by 159SourcecodeScholar
2020

Truly Batch Model-Free Inverse Reinforcement Learning about Multiple Intentions

AISTATS 2020poster

We consider Inverse Reinforcement Learning (IRL) about multiple intentions, \ie the problem of estimating the unknown reward functions optimized by a group of experts that demonstrate optimal behaviors. Most of the existing algorithms either require access to a model of the environment or need to re…

Cited by 42SourcePDFScholar
2019

Optimistic Policy Optimization via Multiple Importance Sampling

ICML 2019oral

Policy Search (PS) is an effective approach to Reinforcement Learning (RL) for solving control tasks with continuous state-action spaces. In this paper, we address the exploration-exploitation trade-off in PS by proposing an approach based on Optimism in the Face of Uncertainty. We cast the PS probl…

2019

Propagating Uncertainty in Reinforcement Learning via Wasserstein Barycenters

NeurIPS 2019poster

How does the uncertainty of the value function propagate when performing temporal difference learning? In this paper, we address this question by proposing a Bayesian framework in which we employ approximate posterior distributions to model the uncertainty of the value function and Wasserstein baryc…

2019

Reinforcement Learning in Configurable Continuous Environments

ICML 2019oral

Configurable Markov Decision Processes (Conf-MDPs) have been recently introduced as an extension of the usual MDP model to account for the possibility of configuring the environment to improve the agent’s performance. Currently, there is still no suitable algorithm to solve the learning problem for…

Cited by 19SourcePDFScholar
2019

Transfer of Samples in Policy Search via Multiple Importance Sampling

ICML 2019oral

We consider the transfer of experience samples in reinforcement learning. Most of the previous works in this context focused on value-based settings, where transferring instances conveniently reduces to the transfer of (s,a,s’,r) tuples. In this paper, we consider the more complex case of reusing sa…

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

Policy Optimization via Importance Sampling

NeurIPS 2018oral

Policy optimization is an effective reinforcement learning approach to solve continuous control tasks. Recent achievements have shown that alternating online and offline optimization is a successful choice for efficient trajectory reuse. However, deciding when to stop optimizing and collect new traj…

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…

2018

Transfer of Value Functions via Variational Methods

NeurIPS 2018poster

We consider the problem of transferring value functions in reinforcement learning. We propose an approach that uses the given source tasks to learn a prior distribution over optimal value functions and provide an efficient variational approximation of the corresponding posterior in a new target task…

Cited by 25SourcePDFScholar
2016

Estimating Maximum Expected Value through Gaussian Approximation

ICML 2016poster

This paper is about the estimation of the maximum expected value of a set of independent random variables. The performance of several learning algorithms (e.g., Q-learning) is affected by the accuracy of such estimation. Unfortunately, no unbiased estimator exists. The usual approach of taking the m…

Cited by 63SourcePDFScholar