Research · Offline RL

Offline reinforcement learning

Policy learning and evaluation from fixed logs of prior decisions, without further interaction.

Overview

Offline reinforcement learning derives a policy from a fixed log of prior decisions, with no opportunity to query the environment about actions the log does not contain. Coverage in such a dataset follows the behavior policy that produced it, so the policies with the highest estimated value are frequently the ones supported by the least evidence, and unconstrained value backups extrapolate into that gap. Work in the area therefore divides between restricting learning to the region the data supports and quantifying what the data licenses — which policies can be distinguished, at what confidence, and under which assumptions about how the log was generated.

Themes

  • Behavior-policy support and distributional shift — separating high estimated value from thin coverage; pessimism under uncertainty as a penalty on unsupported actions; locating the point at which conservatism binds tightly enough to discard policies worth considering
  • Confounding in observational logs — clinical and scientific records are produced by decision-makers conditioning on covariates the log never stores, so the identifying assumptions are usually the load-bearing part of the argument; grounding transfer in counterfactual estimation rather than distributional similarity, as in Counterfactually Guided Policy Transfer in Clinical Settings
  • Off-policy evaluation — importance-weighted and doubly-robust estimators, their variance under long horizons and limited overlap, and interval estimates and safe-policy-improvement bounds in place of point estimates
  • State representation under partial observability — an offline policy acts on a summary of an incomplete history rather than on the state itself, and the choice of summary changes the learned policy measurably, as quantified in An Empirical Study of Representation Learning for Reinforcement Learning in Healthcare
  • Data-constrained regimes — when a dataset is too small to identify an optimal policy, weaker claims remain recoverable: Medical Dead-ends and Learning to Identify High-Risk States and Treatments uses negative outcomes to learn which behaviors to avoid rather than which to select
  • Tractability under action structure — methods that score every action stop being computable well before the statistics degrade; see structured action spaces

Open questions

Established offline RL benchmarks hold fixed full observability, dense immediate reward, retryable actions, and a characterizable logging policy. The lab works on the regime in which several of those assumptions fail at once, which is the regime that clinical and scientific records produce.

  • Whether method rankings established at benchmark dataset sizes hold as datasets shrink toward the sizes available in practice
  • How pessimism should be allocated when uncertainty originates in unobserved confounders rather than in sampling error
  • What off-policy evaluation can establish when the outcome is delayed, proxied, or observed at irregular intervals; see delayed and unreliable feedback
  • Which value estimates remain usable near states from which an acceptable outcome is no longer reachable; see irreversibility and risk

Also relevant: BraVE: Offline Reinforcement Learning for Discrete Combinatorial Action Spaces, which addresses offline value estimation when the joint action space cannot be enumerated.

  • RL
  • Offline RL

Publications

Related papers

2026

Improving and Accelerating Offline RL in Large Discrete Action Spaces with Structured Policy Initialization

Matt Landers, Taylor W. Killian, Tom Hartvigsen, Afsaneh Doryab

ICLR 2026

SPIN is a two-stage framework for reinforcement learning in combinatorial action spaces: first pre-train an Action Structure Model that learns which joint actions are valid, then train lightweight control heads on top. It outperforms prior methods by up to 39% in reward while converging up to 12.8x faster.

2025

BraVE: Offline Reinforcement Learning for Discrete Combinatorial Action Spaces

Matt Landers, Taylor W. Killian, Hugo Barnes, Tom Hartvigsen, Afsaneh Doryab

NeurIPS 2025

Offline RL in high-dimensional discrete action spaces, using tree-structured traversal to capture sub-action dependencies. Only a linear number of joint actions need to be evaluated, outperforming existing methods by up to 20x in complex environments.

2024

Clinically Motivated Sequential Decision Making Under Uncertainty in Offline Settings

Taylor W. Killian

PhD Thesis, University of Toronto, Department of Computer Science

The modeling decisions that let you draw actionable conclusions from sequentially observed healthcare data — anchoring method development in the intended real-world use case rather than the benchmark.

2023

Risk Sensitive Dead-end Identification in Safety-Critical Offline Reinforcement Learning

Taylor W. Killian, Sonali Parbhoo, Marzyeh Ghassemi

Transactions on Machine Learning Research (TMLR)

A risk-sensitive treatment of dead-end discovery that uses distributional RL for value estimation, flagging unrecoverable states earlier and tunably, according to the risk tolerance of the task.

2022

Counterfactually Guided Policy Transfer in Clinical Settings

Taylor W. Killian, Marzyeh Ghassemi, Shalmali Joshi

Conference on Health, Inference and Learning (CHIL) 2022

2021

Medical Dead-ends and Learning to Identify High-Risk States and Treatments

Mehdi Fatemi, Taylor W. Killian, Jayakumar Subramanian, Marzyeh Ghassemi

NeurIPS 2021

In data-constrained offline settings an optimal policy may simply not be recoverable. Negative outcomes in the data can still be used to identify behaviors to avoid, guarding against the overoptimistic decisions that reduced data availability invites.

2020

An Empirical Study of Representation Learning for Reinforcement Learning in Healthcare

Taylor W. Killian, Haoran Zhang, Jayakumar Subramanian, Mehdi Fatemi, Marzyeh Ghassemi

ML4H: Machine Learning for Health Workshop at NeurIPS

How the choice of state representation changes what an offline RL agent learns from clinical time series, evaluated across a range of representation learning approaches.

Counterfactual Transfer via Inductive Bias in Clinical Settings

Taylor W. Killian, Marzyeh Ghassemi, Shalmali Joshi

Inductive Biases, Invariances and Generalization in RL (BIG) Workshop, ICML

All publications