Asking for Knowledge: Training RL Agents to Query External Knowledge Using Language
Not on the map. Why?
Abstract (arXiv)
To solve difficult tasks, humans ask questions to acquire knowledge from external sources. In contrast, classical reinforcement learning agents lack such an ability and often resort to exploratory behavior. This is exacerbated as few present-day environments support querying for knowledge. In order to study how agents can be taught to query external knowledge via language, we first introduce two new environments: the grid-world-based Q-BabyAI and the text-based Q-TextWorld. In addition to physical interactions, an agent can query an external knowledge source specialized for these environments to gather information. Second, we propose the "Asking for Knowledge" (AFK) agent, which learns to generate language commands to query for meaningful knowledge that helps solve the tasks. AFK leverages a non-parametric memory, a pointer mechanism and an episodic exploration bonus to tackle (1) irrelevant information, (2) a large query language space, (3) delayed reward for making meaningful queries. Extensive experiments demonstrate that the AFK agent outperforms recent baselines on the challenging Q-BabyAI and Q-TextWorld environments.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 48 of its 73 references that are in the map. First posted on arXiv 12 May 2022.
Builds on
48 of 73 references.
References · 48
- 2017Attention Is All You Needcited by 59,932
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2016Layer Normalizationcited by 5,579
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2014Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modelingcited by 2,341
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
- 2017Deep Setscited by 1,394
- 2017Get To The Point: Summarization with Pointer-Generator Networkscited by 1,281
- 2021Improving language models by retrieving from trillions of tokenscited by 1,278
36 more
- 2022Language Models as Zero-Shot Planners: Extracting Actionable Knowledge for Embodied Agentscited by 1,029
- 2017Rainbow: Combining Improvements in Deep Reinforcement Learningcited by 774
- 2017Deep Reinforcement Learning that Matterscited by 727
- 2017Seq2SQL: Generating Structured Queries from Natural Language using Reinforcement Learningcited by 679
- 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 632
- 2014Memory Networkscited by 354
- 2018TextWorld: A Learning Environment for Text-based Gamescited by 304
- 2017Count-Based Exploration with Neural Density Modelscited by 255
- 2018BabyAI: A Platform to Study the Sample Efficiency of Grounded Language Learningcited by 238
- 2017Learning to Ask: Neural Question Generation for Reading Comprehensioncited by 186
- 2017Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learningcited by 161
- 2020Answering Complex Open-Domain Questions with Multi-Hop Dense Retrievalcited by 160
- 2015Deep Reinforcement Learning in Large Discrete Action Spacescited by 158
- 2020Question and Answer Test-Train Overlap in Open-Domain Question Answering Datasetscited by 137
- 2018Episodic Curiosity through Reachabilitycited by 109
- 2019Help, Anna! Visual Navigation with Natural Multimodal Assistance via Retrospective Curiosity-Encouraging Imitation Learningcited by 100
- 2020Graph Constrained Reinforcement Learning for Natural Language Action Spacescited by 80
- 2017Learning how to Active Learn: A Deep Reinforcement Learning Approachcited by 79
- 2021Cooperative Exploration for Multi-Agent Deep Reinforcement Learningcited by 57
- 2020Learning Dynamic Belief Graphs to Generalize on Text-Based Gamescited by 49
- 2017Ask the Right Questions: Active Question Reformulation with Reinforcement Learningcited by 44
- 2019Metalearned Neural Memorycited by 44
- 2018GEP-PG: Decoupling Exploration and Exploitation in Deep Reinforcement Learning Algorithmscited by 40
- 2017Task-Oriented Query Reformulation with Reinforcement Learningcited by 39
- 2019Multi-Hop Paragraph Retrieval for Open-Domain Question Answeringcited by 34
- 2019RTFM: Generalising to Novel Environment Dynamics via Readingcited by 33
- 2017Machine Comprehension by Text-to-Text Neural Question Generationcited by 32
- 2017Learning Algorithms for Active Learningcited by 30
- 2019Ask to Learn: A Study on Curiosity-driven Question Generationcited by 15
- 2018Two can play this Game: Visual Dialog with Discriminative Question Generation and Answeringcited by 13
- 2021Reinforcement Learning with External Knowledge by using Logical Neural Networkscited by 10
- 2021SocialAI: Benchmarking Socio-Cognitive Abilities in Deep Reinforcement Learning Agentscited by 7
- 2017Reinforcement Learning with External Knowledge and Two-Stage Q-functions for Predicting Popular Reddit Threadscited by 3
- 2019Playing log(N)-Questions over Sentencescited by 2
- 2021Interactive Machine Comprehension with Dynamic Knowledge Graphscited by 2
- 2017Deep Reinforcement Learning Boosted by External Knowledgecited by 1
Built on by
Citing papers per year: 2022: 2; 2023: 3; 2024: 2; 2025: 1; 2026: 2.
10 papers.
- 2023Do the Rewards Justify the Means? Measuring Trade-Offs Between Rewards and Ethical Behavior in the MACHIAVELLI Benchmarkcited by 174
- 2022EAGER: Asking and Answering Questions for Automatic Reward Shaping in Language-guided RLcited by 23
- 2024Language-guided Skill Learning with Temporal Variational Inferencecited by 11
- 2022Behavior Cloned Transformers are Neurosymbolic Reasonerscited by 10
- 2023Architext: Language-Driven Generative Architecture Designcited by 5
- 2026MINT: Minimal Information Neuro-Symbolic Tree for Objective-Driven Knowledge-Gap Reasoning and Active Elicitationcited by 4
- 2023Improve the efficiency of deep reinforcement learning through semantic exploration guided by natural languagecited by 1
- 2024Getting By Goal Misgeneralization With a Little Help From a Mentorcited by 1
- 2025How2: How to learn from procedural How-to questionscited by 1
- 2026Knowledge Boundary Discovery for Large Language Modelscited by 1
- 2026MINT: Minimal Information Neuro-Symbolic Tree for Objective-Driven Knowledge-Gap Reasoning and Active Elicitation4 Feb 2026
- 2026Knowledge Boundary Discovery for Large Language Models14 Jan 2026
- 2025How2: How to learn from procedural How-to questions13 Oct 2025
- 2024Getting By Goal Misgeneralization With a Little Help From a Mentor28 Oct 2024
- 2024Language-guided Skill Learning with Temporal Variational Inference26 Feb 2024
- 2023Improve the efficiency of deep reinforcement learning through semantic exploration guided by natural language21 Sep 2023
- 2023Do the Rewards Justify the Means? Measuring Trade-Offs Between Rewards and Ethical Behavior in the MACHIAVELLI Benchmark6 Apr 2023
- 2023Architext: Language-Driven Generative Architecture Design13 Mar 2023
- 2022Behavior Cloned Transformers are Neurosymbolic Reasoners13 Oct 2022
- 2022EAGER: Asking and Answering Questions for Automatic Reward Shaping in Language-guided RL20 Jun 2022