Learning Finite State Representations of Recurrent Policy Networks
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Recurrent neural networks (RNNs) are an effective representation of control policies for a wide range of reinforcement and imitation learning problems. RNN policies, however, are particularly difficult to explain, understand, and analyze due to their use of continuous-valued memory vectors and observation features. In this paper, we introduce a new technique, Quantized Bottleneck Insertion, to learn finite representations of these vectors and features. The result is a quantized representation of the RNN that can be analyzed to improve our understanding of memory use and general behavior. We present results of this approach on synthetic environments and six Atari games. The resulting finite representations are surprisingly small in some cases, using as few as 3 discrete memory states and 10 observations for a perfect Pong policy. We also show that these finite policy representations lead to improved interpretability.
Key facts. Cited by 37 papers in the map (data through 14 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. First posted on arXiv 29 Nov 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 11; 2022: 6; 2023: 8; 2024: 3; 2025: 2; 2026: 6.
37 papers.
Sort:
By area:
- 2021Acquisition of Chess Knowledge in AlphaZerocited by 123
- 2021A Survey on Interpretable Reinforcement Learningcited by 73
- 2023Bridging the Human-AI Knowledge Gap: Concept Discovery and Transfer in AlphaZerocited by 42
- 2021Counterfactual State Explanations for Reinforcement Learning Agents via Generative Deep Learningcited by 39
- 2021Iterative Bounding MDPs: Learning Interpretable Policies via Non-Interpretable Methodscited by 31
- 2022A Survey of Explainable Reinforcement Learningcited by 31
- 2021Classical Planning in Deep Latent Spacecited by 22
- 2023Explainable Reinforcement Learning via a Causal World Modelcited by 22
- 2022SoK: Explainable Machine Learning for Computer Security Applicationscited by 20
- 2021Explainable Autonomous Robots: A Survey and Perspectivecited by 15
- 2024Synthesizing Programmatic Reinforcement Learning Policies with Large Language Model Guided Searchcited by 13
- 2019Mo"ET: Mixture of Expert Trees and its Application to Verifiable Reinforcement Learningcited by 12
25 more
- 2023Explaining RL Decisions with Trajectoriescited by 7
- 2024Pessimistic Iterative Planning with RNNs for Robust POMDPscited by 7
- 2025Translate Policy to Language: Flow Matching Generated Rewards for LLM Explanationscited by 6
- 2022Local Explanations for Reinforcement Learningcited by 5
- 2021Generating Probabilistic Safety Guarantees for Neural Network Controllerscited by 4
- 2023Learning Symbolic Representations for Reinforcement Learning of Non-Markovian Behaviorcited by 4
- 2023Synthesizing Programmatic Policies with Actor-Critic Algorithms and ReLU Networkscited by 3
- 2021Learning-Aided Heuristics Design for Storage Systemcited by 2
- 2021Identifying Reasoning Flaws in Planning-Based RL Using Tree Explanationscited by 2
- 2022A Framework for Understanding and Visualizing Strategies of RL Agentscited by 2
- 2021Knowledge-based XAI through CBR: There is more to explanations than models can tellcited by 1
- 2023Program Machine Policy: Addressing Long-Horizon Tasks by Integrating Program Synthesis and State Machinescited by 1
- 2023Training Neural Networks with Internal State, Unconstrained Connectivity, and Discrete Activationscited by 1
- 2026Finite-State Controllers for (Hidden-Model) POMDPs using Deep Reinforcement Learningcited by 1
- 2021Longitudinal Distance: Towards Accountable Instance Attributioncited by 0
- 2022Beyond Value: CHECKLIST for Testing Inferences in Planning-Based RLcited by 0
- 2022Global and Local Analysis of Interestingness for Competency-Aware Deep Reinforcement Learningcited by 0
- 2023IxDRL: A Novel Explainable Deep Reinforcement Learning Toolkit based on Analyses of Interestingnesscited by 0
- 2024Explainable Representation of Finite-Memory Policies for POMDPs using Decision Treescited by 0
- 2025Efficiently Vectorized MCMC on Modern Acceleratorscited by 0
- 2026Principal Prototype Analysis on Manifold for Interpretable Reinforcement Learningcited by 0
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learningcited by 0
- 2026ORCAID: Oblique Rule-Based Continuous-Action Interpretation for Deep RL Policiescited by 0
- 2026Explaining Reinforcement Learning Agents via Inductive Logic Programmingcited by 0
- 2026From Black Box to Executable Logic: Explainable Reinforcement Learning through Prolog Expert Systemscited by 0
- 2026From Black Box to Executable Logic: Explainable Reinforcement Learning through Prolog Expert Systems16 Jul 2026
- 2026Explaining Reinforcement Learning Agents via Inductive Logic Programming15 Jul 2026
- 2026ORCAID: Oblique Rule-Based Continuous-Action Interpretation for Deep RL Policies8 Jul 2026
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learning24 Jun 2026
- 2026Principal Prototype Analysis on Manifold for Interpretable Reinforcement Learning30 Mar 2026
- 2026Finite-State Controllers for (Hidden-Model) POMDPs using Deep Reinforcement Learning9 Feb 2026
- 2025Efficiently Vectorized MCMC on Modern Accelerators20 Mar 2025
- 2025Translate Policy to Language: Flow Matching Generated Rewards for LLM Explanations18 Feb 2025
- 2024Explainable Representation of Finite-Memory Policies for POMDPs using Decision Trees20 Nov 2024
- 2024Pessimistic Iterative Planning with RNNs for Robust POMDPs16 Aug 2024
- 2024Synthesizing Programmatic Reinforcement Learning Policies with Large Language Model Guided Search26 May 2024
- 2023Training Neural Networks with Internal State, Unconstrained Connectivity, and Discrete Activations22 Dec 2023
25 more
- 2023Program Machine Policy: Addressing Long-Horizon Tasks by Integrating Program Synthesis and State Machines27 Nov 2023
- 2023Bridging the Human-AI Knowledge Gap: Concept Discovery and Transfer in AlphaZero25 Oct 2023
- 2023Synthesizing Programmatic Policies with Actor-Critic Algorithms and ReLU Networks4 Aug 2023
- 2023IxDRL: A Novel Explainable Deep Reinforcement Learning Toolkit based on Analyses of Interestingness18 Jul 2023
- 2023Explaining RL Decisions with Trajectories6 May 2023
- 2023Explainable Reinforcement Learning via a Causal World Model4 May 2023
- 2023Learning Symbolic Representations for Reinforcement Learning of Non-Markovian Behavior8 Jan 2023
- 2022Global and Local Analysis of Interestingness for Competency-Aware Deep Reinforcement Learning11 Nov 2022
- 2022SoK: Explainable Machine Learning for Computer Security Applications22 Aug 2022
- 2022A Framework for Understanding and Visualizing Strategies of RL Agents17 Aug 2022
- 2022Beyond Value: CHECKLIST for Testing Inferences in Planning-Based RL4 Jun 2022
- 2022A Survey of Explainable Reinforcement Learning17 Feb 2022
- 2022Local Explanations for Reinforcement Learning8 Feb 2022
- 2021A Survey on Interpretable Reinforcement Learning24 Dec 2021
- 2021Acquisition of Chess Knowledge in AlphaZero17 Nov 2021
- 2021Identifying Reasoning Flaws in Planning-Based RL Using Tree Explanations28 Sep 2021
- 2021Longitudinal Distance: Towards Accountable Instance Attribution23 Aug 2021
- 2021Knowledge-based XAI through CBR: There is more to explanations than models can tell23 Aug 2021
- 2021Classical Planning in Deep Latent Space30 Jun 2021
- 2021Learning-Aided Heuristics Design for Storage System14 Jun 2021
- 2021Explainable Autonomous Robots: A Survey and Perspective6 May 2021
- 2021Generating Probabilistic Safety Guarantees for Neural Network Controllers1 Mar 2021
- 2021Iterative Bounding MDPs: Learning Interpretable Policies via Non-Interpretable Methods25 Feb 2021
- 2021Counterfactual State Explanations for Reinforcement Learning Agents via Generative Deep Learning29 Jan 2021
- 2019Mo"ET: Mixture of Expert Trees and its Application to Verifiable Reinforcement Learning16 Jun 2019
- 2026Finite-State Controllers for (Hidden-Model) POMDPs using Deep Reinforcement Learningcited by 1
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learningcited by 0
- 2026ORCAID: Oblique Rule-Based Continuous-Action Interpretation for Deep RL Policiescited by 0
- 2026Explaining Reinforcement Learning Agents via Inductive Logic Programmingcited by 0
- 2026From Black Box to Executable Logic: Explainable Reinforcement Learning through Prolog Expert Systemscited by 0
- 2026From Black Box to Executable Logic: Explainable Reinforcement Learning through Prolog Expert Systems16 Jul 2026
- 2026Explaining Reinforcement Learning Agents via Inductive Logic Programming15 Jul 2026
- 2026ORCAID: Oblique Rule-Based Continuous-Action Interpretation for Deep RL Policies8 Jul 2026
- 2026Compositional Behavioral Semantics for State Abstraction in Reinforcement Learning24 Jun 2026
- 2026Finite-State Controllers for (Hidden-Model) POMDPs using Deep Reinforcement Learning9 Feb 2026
- 2023Explainable Reinforcement Learning via a Causal World Modelcited by 22
- 2021Acquisition of Chess Knowledge in AlphaZerocited by 123
- 2021A Survey on Interpretable Reinforcement Learningcited by 73
- 2023Bridging the Human-AI Knowledge Gap: Concept Discovery and Transfer in AlphaZerocited by 42
- 2021Counterfactual State Explanations for Reinforcement Learning Agents via Generative Deep Learningcited by 39
- 2021Iterative Bounding MDPs: Learning Interpretable Policies via Non-Interpretable Methodscited by 31
- 2022A Survey of Explainable Reinforcement Learningcited by 31
- 2021Classical Planning in Deep Latent Spacecited by 22
- 2022SoK: Explainable Machine Learning for Computer Security Applicationscited by 20
- 2021Explainable Autonomous Robots: A Survey and Perspectivecited by 15
- 2024Synthesizing Programmatic Reinforcement Learning Policies with Large Language Model Guided Searchcited by 13
- 2023Explaining RL Decisions with Trajectoriescited by 7
- 2024Pessimistic Iterative Planning with RNNs for Robust POMDPscited by 7
12 most cited of 28 All 28 (JSON)
- 2025Efficiently Vectorized MCMC on Modern Accelerators20 Mar 2025
- 2024Explainable Representation of Finite-Memory Policies for POMDPs using Decision Trees20 Nov 2024
- 2024Pessimistic Iterative Planning with RNNs for Robust POMDPs16 Aug 2024
- 2024Synthesizing Programmatic Reinforcement Learning Policies with Large Language Model Guided Search26 May 2024
- 2023Training Neural Networks with Internal State, Unconstrained Connectivity, and Discrete Activations22 Dec 2023
- 2023Program Machine Policy: Addressing Long-Horizon Tasks by Integrating Program Synthesis and State Machines27 Nov 2023
- 2023Bridging the Human-AI Knowledge Gap: Concept Discovery and Transfer in AlphaZero25 Oct 2023
- 2023Synthesizing Programmatic Policies with Actor-Critic Algorithms and ReLU Networks4 Aug 2023
- 2023IxDRL: A Novel Explainable Deep Reinforcement Learning Toolkit based on Analyses of Interestingness18 Jul 2023
- 2023Explaining RL Decisions with Trajectories6 May 2023
- 2023Learning Symbolic Representations for Reinforcement Learning of Non-Markovian Behavior8 Jan 2023
- 2022Global and Local Analysis of Interestingness for Competency-Aware Deep Reinforcement Learning11 Nov 2022
12 newest of 28 All 28 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026Investigating Action Encodings in Recurrent Neural Networks in Reinforcement Learning
- 2026Why Linear Recurrent Memory Works in Partially Observable Reinforcement Learning
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning
- 2026Probabilistic Verification of Recurrent Neural Networks for Single and Multi-Agent Reinforcement Learning
- 2026Reinforcement Learning with Complex (valued) Memories
- 2024Bridging State and History Representations: Understanding Self-Predictive RL
- 2026Does Learning Recurrent Networks Matter in Partially Observable RL?
- 2015On Learning to Think: Algorithmic Information Theory for Novel Combinations of Reinforcement Learning Controllers and Recurrent Neural World Models