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

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

Ryan Li, Yigit Korkmaz, Erdem B{\i}y{\i}k · N/A · 2026

vision reinforcement-learning control learning-from-demonstration

Framework

N/A

License

N/A

Stars

N/A

Summary

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.

Abstract Summary

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.

Key Points

  • 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.

Method

Reward-DAgger leverages dense progress signals obtained from general-purpose reward models to determine when a robot policy should transfer control to the expert. In this section, we first introduce how we construct a gating criterion from general-purpose reward models, and then discuss how we incorporate the criterion into an interactive imitation learning pipeline for fine-tuning the base policy.

A general-purpose reward model is typically constructed by prompting or fine-tuning a VLM to produce dense progress signals.

Method — Reward-DAgger: Robot-Gated Interactive Imitation Learning with General-Purpose Progress-Based Reward Models
Fig. 2 : Detection frontier on LIBERO tasks. Each curve shows the Pareto frontier between balanced accuracy and mean normalized trigger time. S and F denote the numbers of successful and failed trajectories; the dashed gray line indicates the mean normalized length of successful trajectories.

Experiments

Our experiments evaluate the effectiveness of Reward-DAgger as a robot-gated interactive imitation learning framework across different tasks and policy architectures. Specifically, we seek to answer the following research questions: Does the gating mechanism of Reward-DAgger accurately detect failures with low detection latency?

Does the gating configuration of Reward-DAgger transfer across tasks without additional calibration? Does Reward-DAgger efficiently improve downstream policy performance while minimizing human burden? Baselines.

Experiments — Reward-DAgger: Robot-Gated Interactive Imitation Learning with General-Purpose Progress-Based Reward Models
Fig. 3 : Overview of simulated and real-world tasks. Left and right images show each task’s initial and final frames.

Sources and Demonstrations

Figures are reproduced from Li et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.

Share

Related Papers

A Sim-to-Real Integration Pipeline for Training and Deployment of Chunk-Based VLA Manipulation Policies
arXiv preprint

A Sim-to-Real Integration Pipeline for Training and Deployment of Chunk-Based VLA Manipulation Policies

Mathilde Kappel, Clémence Grislain, Mohamed Chetouani et al.

Vision-Language-Action (VLA) models have become a prominent paradigm for mapping multimodal inputs, including semantic instructions, visual observations of the scene, and proprioceptive observations, to robot actions.

manipulation vision vla reinforcement-learning sim-to-real control learning-from-demonstration benchmark
PDF Advanced
Capability-Aware Arbitration for Semantic Intent-Based Shared Control
arXiv preprint

Capability-Aware Arbitration for Semantic Intent-Based Shared Control

Zhaoda Du, Michael Bowman, Xiaoli Zhang

Shared control often allocates robot authority based on confidence in inferred human intent, assuming reliable autonomous execution. When this assumption fails, high intent confidence can cause over-helping.

manipulation vision vla reinforcement-learning control learning-from-demonstration benchmark
Code PDF Intermediate
GitHub ★ —