A Ranking Game for Imitation Learning
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
We propose a new framework for imitation learning -- treating imitation as a two-player ranking-based game between a policy and a reward. In this game, the reward agent learns to satisfy pairwise performance rankings between behaviors, while the policy agent learns to maximize this reward. In imitation learning, near-optimal expert data can be difficult to obtain, and even in the limit of infinite data cannot imply a total ordering over trajectories as preferences can. On the other hand, learning from preferences alone is challenging as a large number of preferences are required to infer a high-dimensional reward function, though preference data is typically much easier to collect than expert demonstrations. The classical inverse reinforcement learning (IRL) formulation learns from expert demonstrations but provides no mechanism to incorporate learning from offline preferences and vice versa. We instantiate the proposed ranking-game framework with a novel ranking loss giving an algorithm that can simultaneously learn from expert demonstrations and preferences, gaining the advantages of both modalities. Our experiments show that the proposed method achieves state-of-the-art sample efficiency and can solve previously unsolvable tasks in the Learning from Observation (LfO) setting. Project video and code can be found at https://hari-sikchi.github.io/rank-game/
Key facts. Cited by 18 papers in the map (data through 7 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. Builds on 45 of its 90 references that are in the map. First posted on arXiv 7 Feb 2022.
Builds on
45 of 90 references.
Closest ancestors · 18
Same direction, newest first
- 2021IQ-Learn: Inverse soft-Q Learning for Imitation23 Jun
- 2021What Matters for Adversarial Imitation Learning?1 Jun
- 2021Of Moments and Matching: A Game-Theoretic Framework for Closing the Imitation Gap4 Mar
- 2021Off-Policy Imitation Learning from Observations25 Feb
- 2021MobILE: Model-Based Imitation Learning From Observation Alone22 Feb
- 2020Error Bounds of Imitating Policies and Environments22 Oct
- 2020Non-Adversarial Imitation Learning and its Connections to Adversarial Methods8 Aug
- 2020Safe Imitation Learning via Fast Bayesian Reward Inference from Preferences21 Feb
- 2019Imitation Learning via Off-Policy Distribution Matching10 Dec
- 2019State Alignment-based Imitation Learning21 Nov
- 2019A Divergence Minimization Perspective on Imitation Learning Methods6 Nov
- 2019Imitation Learning from Observations by Minimizing Inverse Dynamics Disagreement10 Oct
6 more
- 2019Recent Advances in Imitation Learning from Observation30 May
- 2019Imitation Learning as f-Divergence Minimization30 May
- 2019Provably Efficient Imitation Learning from Observation Alone27 May
- 2018Generative Adversarial Imitation from Observation17 Jul
- 2018Imitating Latent Policies from Observation21 May
- 2018Behavioral Cloning from Observation4 May
Other references · 27
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
- 2016Least Squares Generative Adversarial Networkscited by 1,000
- 2017Deep Reinforcement Learning that Matterscited by 727
- 2017Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrationscited by 685
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 485
- 2020robosuite: A Modular Simulation Framework and Benchmark for Robot Learningcited by 484
- 2016Guided Cost Learning: Deep Inverse Optimal Control via Policy Optimizationcited by 400
15 more
- 2018Reward learning from human preferences and demonstrations in Ataricited by 314
- 2019Extrapolating Beyond Suboptimal Demonstrations via Inverse Reinforcement Learning from Observationscited by 256
- 2018MixUp as Locally Linear Out-Of-Manifold Regularizationcited by 158
- 2020Reward-rational (implicit) choice: A unifying formalism for reward learningcited by 119
- 2020Learning from Suboptimal Demonstration via Self-Supervised Reward Regressioncited by 84
- 2020A Game Theoretic Framework for Model Based Reinforcement Learningcited by 72
- 2020Learning Reward Functions from Diverse Sources of Human Feedback: Optimally Integrating Demonstrations and Preferencescited by 71
- 2020f-IRL: Inverse Reinforcement Learning via State Marginal Matchingcited by 67
- 2019Learning Reward Functions by Integrating Human Demonstrations and Preferencescited by 64
- 2015Increasing the Action Gap: New Operators for Reinforcement Learningcited by 49
- 2020When Humans Aren't Optimal: Robots that Collaborate with Risk-Aware Humanscited by 34
- 2019Competitive Gradient Descentcited by 32
- 2019Convergence of Learning Dynamics in Stackelberg Gamescited by 28
- 2019Reinforcement Learning from Imperfect Demonstrations under Soft Expert Guidancecited by 25
- 2021Stackelberg Actor-Critic: Game-Theoretic Reinforcement Learning Algorithmscited by 18
Built on by
Citing papers per year: 2023: 4; 2024: 9; 2025: 4; 2026: 1.
18 papers.
Sort:
By area:
- 2024A Minimaximalist Approach to Reinforcement Learning from Human Feedbackcited by 141
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
- 2023Contrastive Preference Learning: Learning from Human Feedback without RLcited by 77
- 2023Dual RL: Unification and New Methods for Reinforcement and Imitation Learningcited by 48
- 2024Aligning Language Models with Demonstrated Feedbackcited by 32
- 2023LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learningcited by 28
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learningcited by 21
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learningcited by 10
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasetscited by 10
- 2024Sample-Efficient Preference-based Reinforcement Learning with Dynamics Aware Rewardscited by 9
- 2025CREStE: Scalable Mapless Navigation with Internet Scale Priors and Counterfactual Guidancecited by 9
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
6 more
- 2024Robot Air Hockey: A Manipulation Testbed for Robot Learning with Reinforcement Learningcited by 4
- 2024A Unified Linear Programming Framework for Offline Reward Learning from Human Demonstrations and Feedbackcited by 4
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learningcited by 2
- 2025On Generalization and Distributional Update for Mimicking Observations with Adequate Explorationcited by 0
- 2025Learning from Observation: A Survey of Recent Advancescited by 0
- 2026Inverse Reinforcement Learning without an Optimal Demonstrator: A Feasible Reward Set Approachcited by 0
- 2026Inverse Reinforcement Learning without an Optimal Demonstrator: A Feasible Reward Set Approach29 May 2026
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025CREStE: Scalable Mapless Navigation with Internet Scale Priors and Counterfactual Guidance5 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025On Generalization and Distributional Update for Mimicking Observations with Adequate Exploration22 Jan 2025
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasets13 Jun 2024
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithms5 Jun 2024
- 2024Aligning Language Models with Demonstrated Feedback2 Jun 2024
- 2024A Unified Linear Programming Framework for Offline Reward Learning from Human Demonstrations and Feedback20 May 2024
- 2024Robot Air Hockey: A Manipulation Testbed for Robot Learning with Reinforcement Learning6 May 2024
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learning20 Mar 2024
- 2024Sample-Efficient Preference-based Reinforcement Learning with Dynamics Aware Rewards28 Feb 2024
6 more
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learning1 Feb 2024
- 2024A Minimaximalist Approach to Reinforcement Learning from Human Feedback8 Jan 2024
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learning3 Nov 2023
- 2023Contrastive Preference Learning: Learning from Human Feedback without RL20 Oct 2023
- 2023LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learning1 Mar 2023
- 2023Dual RL: Unification and New Methods for Reinforcement and Imitation Learning16 Feb 2023
- 2023Dual RL: Unification and New Methods for Reinforcement and Imitation Learningcited by 48
- 2023LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learningcited by 28
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learningcited by 21
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasetscited by 10
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learningcited by 2
- 2025On Generalization and Distributional Update for Mimicking Observations with Adequate Explorationcited by 0
- 2025Learning from Observation: A Survey of Recent Advancescited by 0
- 2026Inverse Reinforcement Learning without an Optimal Demonstrator: A Feasible Reward Set Approachcited by 0
- 2026Inverse Reinforcement Learning without an Optimal Demonstrator: A Feasible Reward Set Approach29 May 2026
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025On Generalization and Distributional Update for Mimicking Observations with Adequate Exploration22 Jan 2025
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasets13 Jun 2024
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learning1 Feb 2024
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learning3 Nov 2023
- 2023LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learning1 Mar 2023
- 2023Dual RL: Unification and New Methods for Reinforcement and Imitation Learning16 Feb 2023
- 2024A Minimaximalist Approach to Reinforcement Learning from Human Feedbackcited by 141
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
- 2023Contrastive Preference Learning: Learning from Human Feedback without RLcited by 77
- 2024Aligning Language Models with Demonstrated Feedbackcited by 32
- 2024Sample-Efficient Preference-based Reinforcement Learning with Dynamics Aware Rewardscited by 9
- 2024A Unified Linear Programming Framework for Offline Reward Learning from Human Demonstrations and Feedbackcited by 4
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithms5 Jun 2024
- 2024Aligning Language Models with Demonstrated Feedback2 Jun 2024
- 2024A Unified Linear Programming Framework for Offline Reward Learning from Human Demonstrations and Feedback20 May 2024
- 2024Sample-Efficient Preference-based Reinforcement Learning with Dynamics Aware Rewards28 Feb 2024
- 2024A Minimaximalist Approach to Reinforcement Learning from Human Feedback8 Jan 2024
- 2023Contrastive Preference Learning: Learning from Human Feedback without RL20 Oct 2023
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learningcited by 10
- 2025CREStE: Scalable Mapless Navigation with Internet Scale Priors and Counterfactual Guidancecited by 9
- 2024Robot Air Hockey: A Manipulation Testbed for Robot Learning with Reinforcement Learningcited by 4
- 2025CREStE: Scalable Mapless Navigation with Internet Scale Priors and Counterfactual Guidance5 Mar 2025
- 2024Robot Air Hockey: A Manipulation Testbed for Robot Learning with Reinforcement Learning6 May 2024
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learning20 Mar 2024
Similar, not cited
Close in wording, no citation either way
- 2019Better-than-Demonstrator Imitation Learning via Automatically-Ranked Demonstrations
- 2021Imitation Learning by Reinforcement Learning
- 2024Imitation Learning from Suboptimal Demonstrations via Meta-Learning An Action Ranker
- 2021Hyperparameter Selection for Imitation Learning
- 2025PROF: An LLM-based Reward Code Preference Optimization Framework for Offline Imitation Learning
- 2026Matching Multiple Experts: On the Exploitability of Multi-Agent Imitation Learning
- 2022Imitation Learning by Estimating Expertise of Demonstrators
- 2019Reinforced Imitation in Heterogeneous Action Space