cs.ROOct 2, 2026

Reward-DAgger: Robot-Gated Interactive Imitation Learning with General-Purpose Progress-Based Reward Models

Authors: Ryan Li, Yigit Korkmaz, Erdem Bıyık

Organizations: Department of Mathematics, University of Southern California · Thomas Lord Department of Computer Science, University of Southern California

Abstract

Recent advances in robot learning have enabled generalist control policies capable of completing a wide range of tasks. However, their performance degrades when deployed in unseen environments, making it critical to detect failures and teach recovery behaviors. Existing runtime monitoring methods often require task- and policy-specific training or hyperparameter tuning, limiting cross-task deployment and introducing additional overhead during iterative policy updates. We present Reward-DAgger, a robot-gated interactive imitation learning framework that uses dense progress signals from a general-purpose reward model to determine when human intervention is needed. Our approach is agnostic to the underlying policy architecture, requires no access to policy internals, and can be applied across tasks without retuning the gating mechanism. Our results show that Reward-DAgger achieves a better failure-detection accuracy-latency tradeoff than existing runtime monitoring baselines. Across eight simulated and real-world tasks, Reward-DAgger consistently improves the downstream policy's autonomous success rate throughout interactive learning and achieves strong return on human effort, outperforming the baselines in most settings. Importantly, the same gating configuration is used across tasks without task-specific hyperparameter tuning, demonstrating transfer across tasks, environments, and policy architectures. Code and videos are available at https://liralab.usc.edu/reward-dagger.

Figures & tables

Explore similar work

CardsList
  1. TimelyDAgger: Timing-Aware Expert Querying for VLA Policy Improvement

    Sep 27, 2026Zhixuan Zhao, Peiyan Li, Enhao Zhang +10Robot Policy LearningRobot Imitation Learning

  2. WM-DAgger: Enabling Efficient Data Aggregation for Imitation Learning with World Models

    Apr 13, 2026Anlan Yu, Zaishu Chen, Peili Song +6Robot Imitation LearningData-Efficient Robot Learning

  3. BlenDAgger: Blended Shared Control for Interactive Imitation Learning

    Sep 29, 2026Cailyn Smith, Geoffrey Sun, Henny Admoni +1Robotic ControlRobotic Manipulation