LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learning
On the map: Offline RL and imitation learning from data → Inverse reinforcement learning
Abstract (arXiv)
Recent methods for imitation learning directly learn a Q-function using an implicit reward formulation rather than an explicit reward function. However, these methods generally require implicit reward regularization to improve stability and often mistreat absorbing states. Previous works show that a squared norm regularization on the implicit reward function is effective, but do not provide a theoretical analysis of the resulting properties of the algorithms. In this work, we show that using this regularizer under a mixture distribution of the policy and the expert provides a particularly illuminating perspective: the original objective can be understood as squared Bellman error minimization, and the corresponding optimization problem minimizes a bounded chi2-Divergence between the expert and the mixture distribution. This perspective allows us to address instabilities and properly treat absorbing states. We show that our method, Least Squares Inverse Q-Learning (LS-IQ), outperforms state-of-the-art algorithms, particularly in environments with absorbing states. Finally, we propose to use an inverse dynamics model to learn from observations only. Using this approach, we retain performance in settings where no expert actions are available.
Key facts. Cited by 29 papers in the map (data through 14 Oct 2026). Placed in Inverse reinforcement learning, one of 13 directions in Offline RL and imitation learning from data. Builds on 24 of its 35 references that are in the map. First posted on arXiv 1 Mar 2023.
Builds on
24 of 35 references.
Closest ancestors · 1
Same direction, newest first
- 2019RIDM: Reinforced Inverse Dynamics Modeling for Learning from a Single Observed Demonstration18 Jun
Other references · 23
- 2022Generative Adversarial Networkscited by 5,627
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2016Generative Adversarial Imitation Learningcited by 1,597
- 2016Least Squares Generative Adversarial Networkscited by 1,001
- 2016f-GAN: Training Generative Neural Samplers using Variational Divergence Minimizationcited by 506
- 2018Behavioral Cloning from Observationcited by 492
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 490
- 2019Deep Evidential Regressioncited by 355
- 2019DualDICE: Behavior-Agnostic Estimation of Discounted Stationary Distribution Correctionscited by 254
- 2018Discriminator-Actor-Critic: Addressing Sample Inefficiency and Reward Bias in Adversarial Imitation Learningcited by 204
- 2021IQ-Learn: Inverse soft-Q Learning for Imitationcited by 193
- 2019A Divergence Minimization Perspective on Imitation Learning Methodscited by 190
11 more
- 2018Generative Adversarial Imitation from Observationcited by 184
- 2019Imitation Learning via Off-Policy Distribution Matchingcited by 183
- 2018Variational Discriminator Bottleneck: Improving Imitation Learning, Inverse RL, and GANs by Constraining Information Flowcited by 105
- 2019Recent Advances in Imitation Learning from Observationcited by 93
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 91
- 2017Combining Self-Supervised Learning and Imitation for Vision-Based Rope Manipulationcited by 90
- 2021Of Moments and Matching: A Game-Theoretic Framework for Closing the Imitation Gapcited by 81
- 2020MushroomRL: Simplifying Reinforcement Learning Researchcited by 50
- 2022A Ranking Game for Imitation Learningcited by 18
- 2020Non-Adversarial Imitation Learning and its Connections to Adversarial Methodscited by 17
- 2019Hybrid Reinforcement Learning with Expert State Sequencescited by 14
Built on by
Citing papers per year: 2023: 7; 2024: 9; 2025: 6; 2026: 7.
29 papers.
Sort:
By area:
- 2023Inverse Preference Learning: Preference-based RL without a Reward Functioncited by 68
- 2023LocoMuJoCo: A Comprehensive Imitation Learning Benchmark for Locomotioncited by 33
- 2023Coherent Soft Imitation Learningcited by 22
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learningcited by 21
- 2024Imitating Language via Scalable Inverse Reinforcement Learningcited by 21
- 2024Diffusing States and Matching Scores: A New Framework for Imitation Learningcited by 16
- 2023SequenceMatch: Imitation Learning for Autoregressive Sequence Modelling with Backtrackingcited by 13
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasetscited by 10
- 2023Learning Robot Manipulation from Cross-Morphology Demonstrationcited by 8
- 2024Reward-free World Models for Online Imitation Learningcited by 7
- 2024OLLIE: Imitation Learning from Offline Pretraining to Online Finetuningcited by 6
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigationcited by 6
17 more
- 2023Mimicking Better by Matching the Approximate Action Distributioncited by 5
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learningcited by 2
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspectivecited by 2
- 2024Exciting Action: Investigating Efficient Exploration for Learning Musculoskeletal Humanoid Locomotioncited by 1
- 2024The Role of Domain Randomization in Training Diffusion Policies for Whole-Body Humanoid Controlcited by 1
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learningcited by 1
- 2024UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrationscited by 0
- 2025RIZE: Adaptive Regularization for Imitation Learningcited by 0
- 2025DO-IQS: Dynamics-Aware Offline Inverse Q-Learning for Optimal Stopping with Unknown Gain Functionscited by 0
- 2025Learning What to Do and What Not To Do: Offline Imitation from Expert and Undesirable Demonstrationscited by 0
- 2025Value from Observations: Towards Large-Scale Imitation Learning via Self-Improvementcited by 0
- 2026Latent Wasserstein Adversarial Imitation Learningcited by 0
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanismcited by 0
- 2026Trust Region Inverse Reinforcement Learning: Explicit Dual Ascent using Local Policy Updatescited by 0
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learningcited by 0
- 2026Loop-Free Inverse Reinforcement Learning via Sequential Value Recovery with Q-Score Matchingcited by 0
- 2026A Geometric Perspective on Reward Function Updates in Inverse Reinforcement Learningcited by 0
- 2026A Geometric Perspective on Reward Function Updates in Inverse Reinforcement Learning2 Oct 2026
- 2026Loop-Free Inverse Reinforcement Learning via Sequential Value Recovery with Q-Score Matching30 Sep 2026
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learning28 Sep 2026
- 2026Trust Region Inverse Reinforcement Learning: Explicit Dual Ascent using Local Policy Updates10 May 2026
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanism24 Mar 2026
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learning1 Feb 2026
- 2025Value from Observations: Towards Large-Scale Imitation Learning via Self-Improvement9 Jul 2025
- 2025Learning What to Do and What Not To Do: Offline Imitation from Expert and Undesirable Demonstrations27 May 2025
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspective24 May 2025
- 2025DO-IQS: Dynamics-Aware Offline Inverse Q-Learning for Optimal Stopping with Unknown Gain Functions5 Mar 2025
- 2025RIZE: Adaptive Regularization for Imitation Learning27 Feb 2025
17 more
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigation4 Feb 2025
- 2024The Role of Domain Randomization in Training Diffusion Policies for Whole-Body Humanoid Control2 Nov 2024
- 2024Reward-free World Models for Online Imitation Learning17 Oct 2024
- 2024Diffusing States and Matching Scores: A New Framework for Imitation Learning17 Oct 2024
- 2024UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrations10 Oct 2024
- 2024Imitating Language via Scalable Inverse Reinforcement Learning2 Sep 2024
- 2024Exciting Action: Investigating Efficient Exploration for Learning Musculoskeletal Humanoid Locomotion16 Jul 2024
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasets13 Jun 2024
- 2024OLLIE: Imitation Learning from Offline Pretraining to Online Finetuning24 May 2024
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learning1 Feb 2024
- 2023LocoMuJoCo: A Comprehensive Imitation Learning Benchmark for Locomotion4 Nov 2023
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learning3 Nov 2023
- 2023Mimicking Better by Matching the Approximate Action Distribution16 Jun 2023
- 2023SequenceMatch: Imitation Learning for Autoregressive Sequence Modelling with Backtracking8 Jun 2023
- 2023Coherent Soft Imitation Learning25 May 2023
- 2023Inverse Preference Learning: Preference-based RL without a Reward Function24 May 2023
- 2023Learning Robot Manipulation from Cross-Morphology Demonstration7 Apr 2023
- 2023LocoMuJoCo: A Comprehensive Imitation Learning Benchmark for Locomotioncited by 33
- 2023Coherent Soft Imitation Learningcited by 22
- 2023SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learningcited by 21
- 2024Imitating Language via Scalable Inverse Reinforcement Learningcited by 21
- 2023SequenceMatch: Imitation Learning for Autoregressive Sequence Modelling with Backtrackingcited by 13
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasetscited by 10
- 2023Learning Robot Manipulation from Cross-Morphology Demonstrationcited by 8
- 2024OLLIE: Imitation Learning from Offline Pretraining to Online Finetuningcited by 6
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigationcited by 6
- 2023Mimicking Better by Matching the Approximate Action Distributioncited by 5
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learningcited by 2
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspectivecited by 2
12 most cited of 21 All 21 (JSON)
- 2026A Geometric Perspective on Reward Function Updates in Inverse Reinforcement Learning2 Oct 2026
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learning28 Sep 2026
- 2026Trust Region Inverse Reinforcement Learning: Explicit Dual Ascent using Local Policy Updates10 May 2026
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanism24 Mar 2026
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2025Value from Observations: Towards Large-Scale Imitation Learning via Self-Improvement9 Jul 2025
- 2025Learning What to Do and What Not To Do: Offline Imitation from Expert and Undesirable Demonstrations27 May 2025
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspective24 May 2025
- 2025RIZE: Adaptive Regularization for Imitation Learning27 Feb 2025
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigation4 Feb 2025
- 2024UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrations10 Oct 2024
- 2024Imitating Language via Scalable Inverse Reinforcement Learning2 Sep 2024
12 newest of 21 All 21 (JSON)
- 2024Diffusing States and Matching Scores: A New Framework for Imitation Learningcited by 16
- 2024The Role of Domain Randomization in Training Diffusion Policies for Whole-Body Humanoid Controlcited by 1
- 2026Loop-Free Inverse Reinforcement Learning via Sequential Value Recovery with Q-Score Matchingcited by 0
- 2026Loop-Free Inverse Reinforcement Learning via Sequential Value Recovery with Q-Score Matching30 Sep 2026
- 2024The Role of Domain Randomization in Training Diffusion Policies for Whole-Body Humanoid Control2 Nov 2024
- 2024Diffusing States and Matching Scores: A New Framework for Imitation Learning17 Oct 2024
- 2024Reward-free World Models for Online Imitation Learningcited by 7
- 2024Reward-free World Models for Online Imitation Learning17 Oct 2024
Similar, not cited
Close in wording, no citation either way
- 2025Statistical analysis of Inverse Entropy-regularized Reinforcement Learning
- 2020Regularized Inverse Reinforcement Learning
- 2024Learning Causally Invariant Reward Functions from Diverse Demonstrations
- 2023A Novel Variational Lower Bound for Inverse Reinforcement Learning
- 2026Mixing Regularization for Average-Reward Q-Learning and Q-Planning
- 2026Decoding Rewards in Competitive Games: Inverse Game Theory with Entropy Regularization
- 2021Identifiability in inverse reinforcement learning
- 2025Inverse Reinforcement Learning with Just Classification and a Few Regressions