← Search

Drew Bagnell

12 accepted papers

2026

All Roads Lead to Likelihood: The Value of Reinforcement Learning in Fine-Tuning

ICLR 2026poster

From a first-principles perspective, it may seem odd that the strongest results in foundation model fine-tuning (FT) are achieved via a relatively complex, two-stage training procedure. Specifically, one first trains a reward model (RM) on some dataset (e.g., human preferences) before using it to pr…

Cited by 62SourceScholar
2025

To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable RL

NeurIPS 2025spotlight

Partial observability is a notorious challenge in reinforcement learning (RL), due to the need to learn complex, history-dependent policies. Recent empirical successes have used *privileged expert distillation* -- which leverages availability of latent state information during training (e.g., from…

Cited by 0SourceScholar
2024

Hybrid Inverse Reinforcement Learning

ICML 2024poster

The inverse reinforcement learning approach to imitation learning is a double-edged sword. On the one hand, it can enable learning from a smaller number of expert demonstrations with more robustness to error compounding than behavioral cloning approaches. On the other hand, it requires that the lear…

2024

The Importance of Online Data: Understanding Preference Fine-tuning via Coverage

NeurIPS 2024poster

Learning from human preference data has emerged as the dominant paradigm for fine-tuning large language models (LLMs). The two most common families of techniques -- online reinforcement learning (RL) such as Proximal Policy Optimization (PPO) and offline contrastive methods such as Direct Preference…

Cited by 13SourcePDFScholar
2023

Hybrid RL: Using both offline and online data can make RL efficient

ICLR 2023poster

We consider a hybrid reinforcement learning setting (Hybrid RL), in which an agent has access to an offline dataset and the ability to collect experience via real-world online interaction. The framework mitigates the challenges that arise in both pure offline and online RL settings, allowing for the…

2023

Inverse Reinforcement Learning without Reinforcement Learning

ICML 2023poster

Inverse Reinforcement Learning (IRL) is a powerful set of techniques for imitation learning that aims to learn a reward function that rationalizes expert demonstrations. Unfortunately, traditional IRL methods suffer from a computational weakness: they require repeatedly solving a hard reinforcement…

2023

The Virtues of Laziness in Model-based RL: A Unified Objective and Algorithms

ICML 2023poster

We propose a novel approach to addressing two fundamental challenges in Model-based Reinforcement Learning (MBRL): the computational expense of repeatedly finding a good policy in the learned model, and the objective mismatch between model fitting and policy computation. Our "lazy" method leverages…

2022

Causal Imitation Learning under Temporally Correlated Noise

ICML 2022oral

We develop algorithms for imitation learning from policy data that was corrupted by temporally correlated noise in expert actions. When noise affects multiple timesteps of recorded data, it can manifest as spurious correlations between states and actions that a learner might latch on to, leading to…

2022

Minimax Optimal Online Imitation Learning via Replay Estimation

NeurIPS 2022accept

Online imitation learning is the problem of how best to mimic expert demonstrations, given access to the environment or an accurate simulator. Prior work has shown that in the \textit{infinite} sample regime, exact moment matching achieves value equivalence to the expert policy. However, in the \tex…

2022

Sequence Model Imitation Learning with Unobserved Contexts

NeurIPS 2022accept

We consider imitation learning problems where the learner's ability to mimic the expert increases throughout the course of an episode as more information is revealed. One example of this is when the expert has access to privileged information: while the learner might not be able to accurately reprod…

2019

Provably Efficient Imitation Learning from Observation Alone

ICML 2019oral

We study Imitation Learning (IL) from Observations alone (ILFO) in large-scale MDPs. While most IL algorithms rely on an expert to directly provide actions to the learner, in this setting the expert only supplies sequences of observations. We design a new model-free algorithm for ILFO, Forward Adver…