Flexible Attention-Based Multi-Policy Fusion for Efficient Deep Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning (RL) agents have long sought to approach the efficiency of human learning. Humans are great observers who can learn by aggregating external knowledge from various sources, including observations from others' policies of attempting a task. Prior studies in RL have incorporated external knowledge policies to help agents improve sample efficiency. However, it remains non-trivial to perform arbitrary combinations and replacements of those policies, an essential feature for generalization and transferability. In this work, we present Knowledge-Grounded RL (KGRL), an RL paradigm fusing multiple knowledge policies and aiming for human-like efficiency and flexibility. We propose a new actor architecture for KGRL, Knowledge-Inclusive Attention Network (KIAN), which allows free knowledge rearrangement due to embedding-based attentive action prediction. KIAN also addresses entropy imbalance, a problem arising in maximum entropy KGRL that hinders an agent from efficiently exploring the environment, through a new design of policy distributions. The experimental results demonstrate that KIAN outperforms alternative methods incorporating external knowledge policies and achieves efficient and flexible learning. Our implementation is available at https://github.com/Pascalson/KGRL.git
Key facts. Cited by 5 papers in the map (data through 13 Oct 2026). Builds on 28 of its 48 references that are in the map. First posted on arXiv 7 Oct 2022.
Builds on
28 of 48 references.
References · 28
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,223
- 2017Proximal Policy Optimization Algorithmscited by 14,592
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,656
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,726
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 731
- 2017Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrationscited by 687
- 2017Reinforcement Learning with Deep Energy-Based Policiescited by 660
- 2016The Option-Critic Architecturecited by 439
- 2017Overcoming Exploration in Reinforcement Learning with Demonstrationscited by 325
- 2017Leveraging Demonstrations for Deep Reinforcement Learning on Robotics Problems with Sparse Rewardscited by 304
- 2016Hierarchical Deep Reinforcement Learning: Integrating Temporal Abstraction and Intrinsic Motivationcited by 282
- 2018A Lyapunov-based Approach to Safe Reinforcement Learningcited by 279
16 more
- 2019End-to-End Safe Reinforcement Learning through Barrier Functions for Safety-Critical Continuous Control Taskscited by 257
- 2018Multi-Goal Reinforcement Learning: Challenging Robotics Environments and Request for Researchcited by 256
- 2019Language as an Abstraction for Hierarchical Deep Reinforcement Learningcited by 138
- 2020Provably Efficient Safe Exploration via Primal-Dual Policy Optimizationcited by 132
- 2018Near-Optimal Representation Learning for Hierarchical Reinforcement Learningcited by 108
- 2022Reincarnating Reinforcement Learning: Reusing Prior Computation to Accelerate Progresscited by 64
- 2014Safe Exploration of State and Action Spaces in Reinforcement Learningcited by 37
- 2018Reinforced Imitation: Sample Efficient Deep Reinforcement Learning for Map-less Navigation by Leveraging Prior Demonstrationscited by 35
- 2021Unsupervised Skill Discovery with Bottleneck Option Learningcited by 28
- 2017Deep Q-learning from Demonstrationscited by 27
- 2019Composing Task-Agnostic Policies with Deep Reinforcement Learningcited by 20
- 2015Attend, Adapt and Transfer: Attentive Deep Architecture for Adaptive Transfer from multiple sources in the same domaincited by 19
- 2017A Low-Cost Ethics Shaping Approach for Designing Reinforcement Learning Agentscited by 18
- 2020Reinforcement Learning Under Moral Uncertaintycited by 16
- 2020KoGuN: Accelerating Deep Reinforcement Learning via Integrating Human Suboptimal Knowledgecited by 13
- 2021Toward Robust Long Range Policy Transfercited by 6
Built on by
Citing papers per year: 2023: 1; 2024: 3; 2025: 1.
5 papers.
- 2024Improving Sample Efficiency of Reinforcement Learning with Background Knowledge from Large Language Modelscited by 4
- 2023Heterogeneous Knowledge for Augmented Modular Reinforcement Learningcited by 1
- 2025CARoL: Context-aware Adaptation for Robot Learningcited by 1
- 2024SurgIRL: Towards Life-Long Learning for Surgical Automation by Incremental Reinforcement Learningcited by 0
- 2024Conceptual Belief-Informed Reinforcement Learningcited by 0
- 2025CARoL: Context-aware Adaptation for Robot Learning8 Jun 2025
- 2024Conceptual Belief-Informed Reinforcement Learning2 Oct 2024
- 2024SurgIRL: Towards Life-Long Learning for Surgical Automation by Incremental Reinforcement Learning24 Sep 2024
- 2024Improving Sample Efficiency of Reinforcement Learning with Background Knowledge from Large Language Models4 Jul 2024
- 2023Heterogeneous Knowledge for Augmented Modular Reinforcement Learning1 Jun 2023