Minimal Ingredients for Reward Assignment from Expert Demonstrations
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Reward assignment from scarce demonstrations is a key challenge in both offline and online imitation learning. A common and intuitive strategy assigns rewards according to how closely learner trajectories match expert demonstrations. Although this principle underlies many existing methods, the core ingredients that drive performance remain systematically underexplored. We therefore ask: what is the minimal structure that reward assignment must encode to achieve effective downstream RL performance across settings? We approach this question along two design axes: proximity approximation and temporal alignment. Across 32 benchmarks spanning offline and online settings, and with three downstream RL algorithms, our empirical findings suggest: (1) In offline regimes, proximity alone captures the reward structure necessary for effective offline RL, while (2) lightweight temporal correspondence provides consistent gains that are modest offline but essential online or in the presence of multiple demonstrations. We further complement our offline results with a lightweight theory characterizing when simple proximity approximation suffices. Overall, these findings advocate algorithmic minimalism in reward design before introducing complex schemes in both offline and online imitation learning.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. Builds on 29 of its 34 references that are in the map. First posted on arXiv 7 Jun 2025.
Builds on
29 of 34 references.
Closest ancestors · 10
Same direction, newest first
- 2024Align Your Intents: Offline Imitation Learning via Optimal Transport20 Feb
- 2024SEABO: A Simple Search-Based Method for Offline Imitation Learning6 Feb
- 2023Imitation Learning from Observation with Automatic Discount Scheduling11 Oct
- 2022Watch and Match: Supercharging Imitation with Regularized Optimal Transport30 Jun
- 2022Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching4 Feb
- 2021Cross-Domain Imitation Learning via Optimal Transport7 Oct
- 2021IQ-Learn: Inverse soft-Q Learning for Imitation23 Jun
- 2020Offline Learning from Demonstrations and Unlabeled Experience27 Nov
- 2020Primal Wasserstein Imitation Learning8 Jun
- 2019Wasserstein Adversarial Imitation Learning19 Jun
Other references · 19
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,920
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,809
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,427
- 2021Offline Reinforcement Learning with Implicit Q-Learningcited by 1,293
- 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,204
- 2021A Minimalist Approach to Offline Reinforcement Learningcited by 899
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 372
- 2018A Survey of Inverse Reinforcement Learning: Challenges, Methods and Progresscited by 298
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Librarycited by 119
- 2023Teach a Robot to FISH: Versatile Imitation from One Minute of Demonstrationscited by 75
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learningcited by 64
7 more
- 2022How to Leverage Unlabeled Data in Offline Reinforcement Learningcited by 62
- 2023Optimal Transport for Offline Imitation Learningcited by 42
- 2023CLUE: Calibrated Latent Guidance for Offline Reinforcement Learningcited by 11
- 2023Pre-training with Synthetic Data Helps Offline Reinforcement Learningcited by 9
- 2024Rethinking Optimal Transport in Offline Reinforcement Learningcited by 9
- 2024Robot Policy Learning with Temporal Optimal Transport Rewardcited by 8
- 2025Imitation Learning from a Single Temporally Misaligned Videocited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025PROF: An LLM-based Reward Code Preference Optimization Framework for Offline Imitation Learning
- 2025Reward Compatibility: A Framework for Inverse RL
- 2025Towards Improving Reward Design in RL: A Reward Alignment Metric for RL Practitioners
- 2026Generalize and Guide: Decomposing Rewards for Few-Shot Inverse Reinforcement Learning
- 2026Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning
- 2021Of Moments and Matching: A Game-Theoretic Framework for Closing the Imitation Gap
- 2020DERAIL: Diagnostic Environments for Reward And Imitation Learning
- 2019SQIL: Imitation Learning via Reinforcement Learning with Sparse Rewards