← Search

Marek Petrik

25 accepted papers

2026

Dynamic Programming for Epistemic Uncertainty in Markov Decision Processes

ICML 2026spotlight

In this paper, we propose a general theory of ambiguity-averse MDPs, which treats the uncertain transition probabilities as random variables and evaluates a policy via a risk measure applied to its random return. This ambiguity-averse MDP framework unifies several models of MDPs with epistemic uncer…

Cited by 0SourceScholar
2025

Provable Policy Gradient for Robust Average-Reward MDPs Beyond Rectangularity

ICML 2025poster

Robust Markov Decision Processes (MDPs) offer a promising framework for computing reliable policies under model uncertainty. While policy gradient methods have gained increasing popularity in robust discounted MDPs, their application to the average-reward criterion remains largely unexplored. This p…

Cited by 0SourcePDFScholar
2025

Q-learning for Quantile MDPs: A Decomposition, Performance, and Convergence Analysis

AISTATS 2025poster

In Markov decision processes (MDPs), quantile risk measures such as Value-at-Risk are a standard metric for modeling RL agents' preferences for certain outcomes. This paper proposes a new Q-learning algorithm for quantile optimization in MDPs with strong convergence and performance guarantees. The a…

Cited by 0SourcecodeScholar
2023

On Dynamic Programming Decompositions of Static Risk Measures in Markov Decision Processes

NeurIPS 2023poster

Optimizing static risk-averse objectives in Markov decision processes is difficult because they do not admit standard dynamic programming equations common in Reinforcement Learning (RL) algorithms. Dynamic programming decompositions that augment the state space with discrete risk levels have recentl…

Cited by 10SourcePDFScholar
2023

Percentile Criterion Optimization in Offline Reinforcement Learning

NeurIPS 2023poster

In reinforcement learning, robust policies for high-stakes decision-making problems with limited data are usually computed by optimizing the percentile criterion. The percentile criterion is optimized by constructing an uncertainty set that contains the true model with high probability and optimizin…

2023

Policy Gradient in Robust MDPs with Global Convergence Guarantee

ICML 2023poster

Robust Markov decision processes (RMDPs) provide a promising framework for computing reliable policies in the face of model errors. Many successful reinforcement learning algorithms build on variations of policy-gradient methods, but adapting these methods to RMDPs has been challenging. As a result,…

2023

Reducing Blackwell and Average Optimality to Discounted MDPs via the Blackwell Discount Factor

NeurIPS 2023poster

We introduce the Blackwell discount factor for Markov Decision Processes (MDPs). Classical objectives for MDPs include discounted, average, and Blackwell optimality. Many existing approaches to computing average-optimal policies solve for discount-optimal policies with a discount factor close to $1$…

Cited by 18SourcePDFScholar
2022

Data poisoning attacks on off-policy policy evaluation methods

UAI 2022poster

Off-policy Evaluation (OPE) methods are a crucial tool for evaluating policies in high-stakes domains such as healthcare, where exploration is often infeasible, unethical, or expensive. However, the extent to which such methods can be trusted under adversarial threats to data quality is largely unex…

Cited by 2SourcePDFScholar
2021

Fast Algorithms for $L_\infty$-constrained S-rectangular Robust MDPs

NeurIPS 2021poster

Robust Markov decision processes (RMDPs) are a useful building block of robust reinforcement learning algorithms but can be hard to solve. This paper proposes a fast, exact algorithm for computing the Bellman operator for S-rectangular robust Markov decision processes with $L_\infty$-constrained rec…

Cited by 34SourcePDFScholar
2021

Optimizing Percentile Criterion using Robust MDPs

AISTATS 2021poster

We address the problem of computing reliable policies in reinforcement learning problems with limited data. In particular, we compute policies that achieve good returns with high confidence when deployed. This objective, known as the percentile criterion, can be optimized using Robust MDPs (RMDPs).…

Cited by 22SourcePDFScholar
2021

Policy Gradient Bayesian Robust Optimization for Imitation Learning

ICML 2021spotlight

The difficulty in specifying rewards for many real-world problems has led to an increased focus on learning rewards from human feedback, such as demonstrations. However, there are often many different reward functions that explain the human feedback, leaving agents with uncertainty over what the tru…

Cited by 26SourcePDFScholar
2021

Robust Behavior Cloning with Adversarial Demonstration Detection

IROS 2021poster

Imitation learning (IL) frameworks in robotics typically assume that a domain expert's demonstration always contains a correct way of doing the task. Despite its theoretical convenience, this assumption has limited practical values for an IL-powered robot in real world. There are many reasons for an…

Cited by 5SourceScholar
2019

Beyond Confidence Regions: Tight Bayesian Ambiguity Sets for Robust MDPs

NeurIPS 2019poster

Robust MDPs (RMDPs) can be used to compute policies with provable worst-case guarantees in reinforcement learning. The quality and robustness of an RMDP solution are determined by the ambiguity set---the set of plausible transition probabilities---which is usually constructed as a multi-dimensional…

2018

Policy-Conditioned Uncertainty Sets for Robust Markov Decision Processes

NeurIPS 2018spotlight

What policy should be employed in a Markov decision process with uncertain parameters? Robust optimization answer to this question is to use rectangular uncertainty sets, which independently reflect available knowledge about each state, and then obtains a decision policy that maximizes expected rewa…

Cited by 23SourcePDFScholar