Self-Supervised Online Reward Shaping in Sparse-Reward Environments
Not on the map. Why?
Abstract (arXiv)
We introduce Self-supervised Online Reward Shaping (SORS), which aims to improve the sample efficiency of any RL algorithm in sparse-reward environments by automatically densifying rewards. The proposed framework alternates between classification-based reward inference and policy update steps -- the original sparse reward provides a self-supervisory signal for reward inference by ranking trajectories that the agent observes, while the policy update is performed with the newly inferred, typically dense reward function. We introduce theory that shows that, under certain conditions, this alteration of the reward function will not change the optimal policy of the original MDP, while potentially increasing learning speed significantly. Experimental results on several sparse-reward environments demonstrate that, across multiple domains, the proposed algorithm is not only significantly more sample efficient than a standard RL baseline using sparse rewards, but, at times, also achieves similar sample efficiency compared to when hand-designed dense reward functions are used.
Key facts. Cited by 34 papers in the map (data through 7 Oct 2026). Builds on 19 of its 38 references that are in the map. First posted on arXiv 8 Mar 2021.
Builds on
19 of 38 references.
References · 19
- 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
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,704
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2015End-to-End Training of Deep Visuomotor Policiescited by 847
- 2016Deep Reinforcement Learning for Robotic Manipulation with Asynchronous Off-Policy Updatescited by 342
- 2018Reward learning from human preferences and demonstrations in Ataricited by 314
- 2019Extrapolating Beyond Suboptimal Demonstrations via Inverse Reinforcement Learning from Observationscited by 256
- 2017Inverse Reward Designcited by 246
- 2020Fast Differentiable Sorting and Rankingcited by 196
- 2018Learning by Playing - Solving Sparse Reward Tasks from Scratchcited by 164
7 more
- 2019Keeping Your Distance: Solving Sparse Reward Tasks Using Self-Balancing Shaped Rewardscited by 76
- 2020Safe Imitation Learning via Fast Bayesian Reward Inference from Preferencescited by 73
- 2021Modular Deep Reinforcement Learning for Continuous Motion Planning with Temporal Logiccited by 38
- 2019Reward Shaping via Meta-Learningcited by 30
- 2018Machine Teaching for Inverse Reinforcement Learning: Algorithms and Applicationscited by 28
- 2018Generative Adversarial Self-Imitation Learningcited by 25
- 2020Active Task-Inference-Guided Deep Inverse Reinforcement Learningcited by 17
Built on by
Citing papers per year: 2021: 2; 2022: 0; 2023: 8; 2024: 9; 2025: 8; 2026: 7.
34 papers.
Sort:
By area:
- 2023Benchmarks and Algorithms for Offline Preference-Based Reward Learningcited by 55
- 2023Safe Reinforcement Learning via Probabilistic Logic Shieldscited by 41
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewardscited by 29
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learningcited by 21
- 2021Offline Preference-Based Apprenticeship Learningcited by 17
- 2024Offline Model-Based Optimization by Learning to Rankcited by 13
- 2023Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysiscited by 10
- 2025Multi-Stage Manipulation with Demonstration-Augmented Reward, Policy, and World Model Learningcited by 10
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Taskscited by 9
- 2023Towards Safe Autonomous Driving Policies using a Neuro-Symbolic Deep Reinforcement Learning Approachcited by 7
- 2025Preference-based Multi-Objective Reinforcement Learningcited by 7
- 2025Attention-Based Reward Shaping for Sparse and Delayed Rewardscited by 5
22 more
- 2023Intrinsically motivated graph exploration using network theories of human curiositycited by 3
- 2024Informativeness of Reward Functions in Reinforcement Learningcited by 3
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shapingcited by 3
- 2021On the Benefits of Inducing Local Lipschitzness for Robust Generative Adversarial Imitation Learningcited by 2
- 2024ETGL-DDPG: A Deep Deterministic Policy Gradient Algorithm for Sparse Reward Continuous Controlcited by 2
- 2025Shaping Sparse Rewards in Reinforcement Learning: A Semi-supervised Approachcited by 2
- 2025Beyond expected value: geometric mean optimization for long-term policy performance in reinforcement learningcited by 2
- 2026E2HiL: Entropy-Guided Sample Selection for Efficient Real-World Human-in-the-Loop Reinforcement Learningcited by 2
- 2024On the Sample Efficiency of Abstractions and Potential-Based Reward Shaping in Reinforcement Learningcited by 1
- 2024Appraisal-Guided Proximal Policy Optimization: Modeling Psychological Disorders in Dynamic Grid Worldcited by 1
- 2026Automating Potential-based Reward Shaping with Vision Language Model Guidancecited by 1
- 2023A reinforcement learning path planning approach for range-only underwater target localization with autonomous vehiclescited by 0
- 2023Embedding Contextual Information through Reward Shaping in Multi-Agent Learning: A Case Study from Google Footballcited by 0
- 2023Mining of Single-Class by Active Learning for Semantic Segmentationcited by 0
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learningcited by 0
- 2025Finite-time Convergence Analysis of Actor-Critic with Evolving Rewardcited by 0
- 2025Learning to Compress Graphs via Dual Agents for Consistent Topological Robustness Evaluationcited by 0
- 2026STO-RL: Offline RL under Sparse Rewards via LLM-Guided Subgoal Temporal Ordercited by 0
- 2026PRISM: Parallel Reward Integration with Symmetry for MORLcited by 0
- 2026ARMS: Automatic Reward Shaping for Sparse-Reward Multi-Agent Reinforcement Learningcited by 0
- 2026Learning Process Rewards via Success Visitation Matching for Efficient RLcited by 0
- 2026iADD: Improving Alignment and Diversity in Diffusion Policy Optimizationcited by 0
- 2026iADD: Improving Alignment and Diversity in Diffusion Policy Optimization1 Oct 2026
- 2026Automating Potential-based Reward Shaping with Vision Language Model Guidance25 Jun 2026
- 2026Learning Process Rewards via Success Visitation Matching for Efficient RL22 Jun 2026
- 2026ARMS: Automatic Reward Shaping for Sparse-Reward Multi-Agent Reinforcement Learning22 May 2026
- 2026PRISM: Parallel Reward Integration with Symmetry for MORL20 Feb 2026
- 2026E2HiL: Entropy-Guided Sample Selection for Efficient Real-World Human-in-the-Loop Reinforcement Learning27 Jan 2026
- 2026STO-RL: Offline RL under Sparse Rewards via LLM-Guided Subgoal Temporal Order13 Jan 2026
- 2025Learning to Compress Graphs via Dual Agents for Consistent Topological Robustness Evaluation24 Nov 2025
- 2025Finite-time Convergence Analysis of Actor-Critic with Evolving Reward14 Oct 2025
- 2025Beyond expected value: geometric mean optimization for long-term policy performance in reinforcement learning29 Aug 2025
- 2025Preference-based Multi-Objective Reinforcement Learning18 Jul 2025
- 2025Attention-Based Reward Shaping for Sparse and Delayed Rewards16 May 2025
22 more
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards14 Mar 2025
- 2025Multi-Stage Manipulation with Demonstration-Augmented Reward, Policy, and World Model Learning3 Mar 2025
- 2025Shaping Sparse Rewards in Reinforcement Learning: A Semi-supervised Approach31 Jan 2025
- 2024Offline Model-Based Optimization by Learning to Rank15 Oct 2024
- 2024ETGL-DDPG: A Deep Deterministic Policy Gradient Algorithm for Sparse Reward Continuous Control7 Oct 2024
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learning20 Aug 2024
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learning6 Aug 2024
- 2024Appraisal-Guided Proximal Policy Optimization: Modeling Psychological Disorders in Dynamic Grid World29 Jul 2024
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Tasks25 Apr 2024
- 2024On the Sample Efficiency of Abstractions and Potential-Based Reward Shaping in Reinforcement Learning11 Apr 2024
- 2024Informativeness of Reward Functions in Reinforcement Learning10 Feb 2024
- 2023Mining of Single-Class by Active Learning for Semantic Segmentation18 Jul 2023
- 2023Intrinsically motivated graph exploration using network theories of human curiosity11 Jul 2023
- 2023Towards Safe Autonomous Driving Policies using a Neuro-Symbolic Deep Reinforcement Learning Approach3 Jul 2023
- 2023Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis29 Jun 2023
- 2023Embedding Contextual Information through Reward Shaping in Multi-Agent Learning: A Case Study from Google Football25 Mar 2023
- 2023Safe Reinforcement Learning via Probabilistic Logic Shields6 Mar 2023
- 2023A reinforcement learning path planning approach for range-only underwater target localization with autonomous vehicles17 Jan 2023
- 2023Benchmarks and Algorithms for Offline Preference-Based Reward Learning3 Jan 2023
- 2021Offline Preference-Based Apprenticeship Learning20 Jul 2021
- 2021On the Benefits of Inducing Local Lipschitzness for Robust Generative Adversarial Imitation Learning30 Jun 2021
- 2021Offline Preference-Based Apprenticeship Learningcited by 17
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shapingcited by 3
- 2021On the Benefits of Inducing Local Lipschitzness for Robust Generative Adversarial Imitation Learningcited by 2
- 2023Embedding Contextual Information through Reward Shaping in Multi-Agent Learning: A Case Study from Google Footballcited by 0
- 2026STO-RL: Offline RL under Sparse Rewards via LLM-Guided Subgoal Temporal Ordercited by 0
- 2026STO-RL: Offline RL under Sparse Rewards via LLM-Guided Subgoal Temporal Order13 Jan 2026
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2023Embedding Contextual Information through Reward Shaping in Multi-Agent Learning: A Case Study from Google Football25 Mar 2023
- 2021Offline Preference-Based Apprenticeship Learning20 Jul 2021
- 2021On the Benefits of Inducing Local Lipschitzness for Robust Generative Adversarial Imitation Learning30 Jun 2021
- 2023Safe Reinforcement Learning via Probabilistic Logic Shieldscited by 41
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learningcited by 21
- 2024Offline Model-Based Optimization by Learning to Rankcited by 13
- 2023Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysiscited by 10
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Taskscited by 9
- 2025Preference-based Multi-Objective Reinforcement Learningcited by 7
- 2025Attention-Based Reward Shaping for Sparse and Delayed Rewardscited by 5
- 2023Intrinsically motivated graph exploration using network theories of human curiositycited by 3
- 2024Informativeness of Reward Functions in Reinforcement Learningcited by 3
- 2024ETGL-DDPG: A Deep Deterministic Policy Gradient Algorithm for Sparse Reward Continuous Controlcited by 2
- 2025Shaping Sparse Rewards in Reinforcement Learning: A Semi-supervised Approachcited by 2
- 2025Beyond expected value: geometric mean optimization for long-term policy performance in reinforcement learningcited by 2
12 most cited of 23 All 23 (JSON)
- 2026Learning Process Rewards via Success Visitation Matching for Efficient RL22 Jun 2026
- 2026ARMS: Automatic Reward Shaping for Sparse-Reward Multi-Agent Reinforcement Learning22 May 2026
- 2026PRISM: Parallel Reward Integration with Symmetry for MORL20 Feb 2026
- 2026E2HiL: Entropy-Guided Sample Selection for Efficient Real-World Human-in-the-Loop Reinforcement Learning27 Jan 2026
- 2025Learning to Compress Graphs via Dual Agents for Consistent Topological Robustness Evaluation24 Nov 2025
- 2025Finite-time Convergence Analysis of Actor-Critic with Evolving Reward14 Oct 2025
- 2025Beyond expected value: geometric mean optimization for long-term policy performance in reinforcement learning29 Aug 2025
- 2025Preference-based Multi-Objective Reinforcement Learning18 Jul 2025
- 2025Attention-Based Reward Shaping for Sparse and Delayed Rewards16 May 2025
- 2025Shaping Sparse Rewards in Reinforcement Learning: A Semi-supervised Approach31 Jan 2025
- 2024Offline Model-Based Optimization by Learning to Rank15 Oct 2024
- 2024ETGL-DDPG: A Deep Deterministic Policy Gradient Algorithm for Sparse Reward Continuous Control7 Oct 2024
12 newest of 23 All 23 (JSON)