Kinematic State Abstraction and Provably Efficient Rich-Observation Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
We present an algorithm, HOMER, for exploration and reinforcement learning in rich observation environments that are summarizable by an unknown latent state space. The algorithm interleaves representation learning to identify a new notion of kinematic state abstraction with strategic exploration to reach new states using the learned abstraction. The algorithm provably explores the environment with sample complexity scaling polynomially in the number of latent states and the time horizon, and, crucially, with no dependence on the size of the observation space, which could be infinitely large. This exploration guarantee further enables sample-efficient global policy optimization for any reward function. On the computational side, we show that the algorithm can be implemented efficiently whenever certain supervised learning problems are tractable. Empirically, we evaluate HOMER on a challenging exploration problem, where we show that the algorithm is exponentially more sample efficient than standard reinforcement learning baselines.
Key facts. Cited by 114 papers in the map (data through 14 Oct 2026). First posted on arXiv 13 Nov 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 27; 2022: 41; 2023: 22; 2024: 17; 2025: 4; 2026: 3.
114 papers.
Sort:
By area:
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 178
- 2022Hybrid RL: Using Both Offline and Online Data Can Make RL Efficientcited by 161
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 129
- 2021Rapid Exploration for Open-World Navigation with Latent Goal Modelscited by 121
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2022The Role of Coverage in Online Reinforcement Learningcited by 83
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 52
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2022Guaranteed Discovery of Control-Endogenous Latent States with Multi-Step Inverse Modelscited by 45
- 2021Learning Markov State Abstractions for Deep Reinforcement Learningcited by 44
38 more
- 2021Improved Regret Analysis for Variance-Adaptive Linear Bandits and Horizon-Free Linear Mixture MDPscited by 44
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2021MADE: Exploration via Maximizing Deviation from Explored Regionscited by 40
- 2022Provable Reinforcement Learning with a Short-Term Memorycited by 40
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2021The Information Geometry of Unsupervised Reinforcement Learningcited by 37
- 2021Return-Based Contrastive Representation Learning for Reinforcement Learningcited by 36
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 36
- 2022Provably Efficient Reinforcement Learning in Partially Observable Dynamical Systemscited by 35
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022Model-based RL with Optimistic Posterior Sampling: Structural Conditions and Sample Complexitycited by 32
- 2023Near-Minimax-Optimal Risk-Sensitive Reinforcement Learning with CVaRcited by 32
- 2022Learning in Observable POMDPs, without Computationally Intractable Oraclescited by 30
- 2021Which Mutual-Information Representation Learning Objectives are Sufficient for Control?cited by 29
- 2022Provable Benefit of Multitask Representation Learning in Reinforcement Learningcited by 29
- 2021Provable RL with Exogenous Distractors via Multistep Inverse Dynamicscited by 28
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022On the Statistical Efficiency of Reward-Free Exploration in Non-Linear RLcited by 27
- 2023The Benefits of Being Distributional: Small-Loss Bounds for Reinforcement Learningcited by 27
- 2022Sample-Efficient Reinforcement Learning in the Presence of Exogenous Informationcited by 26
- 2022Contrastive UCB: Provably Efficient Contrastive Self-Supervised Learning in Online Reinforcement Learningcited by 26
- 2021MURAL: Meta-Learning Uncertainty-Aware Rewards for Outcome-Driven Reinforcement Learningcited by 24
- 2021A Free Lunch from the Noise: Provable and Practical Exploration for Representation Learningcited by 24
- 2022Towards Deployment-Efficient Reinforcement Learning: Lower Bound and Optimalitycited by 24
- 2023Representation Learning with Multi-Step Inverse Kinematics: An Efficient and Optimal Approach to Rich-Observation RLcited by 22
- 2021Cross-Trajectory Representation Learning for Zero-Shot Generalization in RLcited by 20
- 2023Distributional Offline Policy Evaluation with Predictive Error Guaranteescited by 20
- 2022Deciding What to Model: Value-Equivalent Sampling for Reinforcement Learningcited by 19
- 2021Model-Invariant State Abstractions for Model-Based Reinforcement Learningcited by 18
- 2022Pretraining in Deep Reinforcement Learning: A Surveycited by 18
- 2021Improving Zero-shot Generalization in Offline Reinforcement Learning using Generalized Similarity Functionscited by 17
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 17
- 2022Agent-Controller Representations: Principled Offline RL with Rich Exogenous Informationcited by 16
- 2023Efficient Model-Free Exploration in Low-Rank MDPscited by 16
- 2022Latent Variable Representation for Reinforcement Learningcited by 15
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 15
- 2023Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPscited by 14
- 2024Offline Multi-task Transfer RL with Representational Penalizationcited by 14
50 most cited of 114 All 114 (JSON)
- 2026Towards Scaling Reinforcement Learning to Massive Populations: Learning Mean-Field Representations26 Aug 2026
- 2026End-to-End Efficient RL for Linear Bellman Complete MDPs with Deterministic Transitions24 Mar 2026
- 2026Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search for Hard Exploration23 Mar 2026
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learning3 Oct 2025
- 2025Where to Intervene: Action Selection in Deep Reinforcement Learning5 Jul 2025
- 2025Offline Action-Free Learning of Ex-BMDPs by Comparing Diverse Datasets26 Mar 2025
- 2025Accelerating Multi-Task Temporal Difference Learning under Low-Rank Representation3 Mar 2025
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimation30 Oct 2024
- 2024Learning a Fast Mixing Exogenous Block MDP using a Single Trajectory3 Oct 2024
- 2024The Central Role of the Loss Function in Reinforcement Learning19 Sep 2024
- 2024Rethinking State Disentanglement in Causal Reinforcement Learning24 Aug 2024
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actions17 Jun 2024
38 more
- 2024Hybrid Reinforcement Learning from Offline Observation Alone11 Jun 2024
- 2024Spatio-temporal Value Semantics-based Abstraction for Dense Deep Reinforcement Learning24 May 2024
- 2024Generalizing Multi-Step Inverse Models for Representation Learning to Finite-Memory POMDPs22 Apr 2024
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learning15 Apr 2024
- 2024Provable Interactive Learning with Hindsight Instruction Feedback14 Apr 2024
- 2024Exploration is Harder than Prediction: Cryptographically Separating Reinforcement Learning from Supervised Learning4 Apr 2024
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learning20 Mar 2024
- 2024Multistep Inverse Is Not All You Need18 Mar 2024
- 2024A Natural Extension To Online Algorithms For Hybrid RL With Limited Coverage7 Mar 2024
- 2024Offline Multi-task Transfer RL with Representational Penalization19 Feb 2024
- 2024Premier-TACO is a Few-Shot Policy Learner: Pretraining Multitask Representation via Temporal Action-Driven Contrastive Loss9 Feb 2024
- 2024Learning mirror maps in policy mirror descent7 Feb 2024
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guarantees14 Nov 2023
- 2023Low-Rank MDPs with Continuous Action Spaces6 Nov 2023
- 2023PcLast: Discovering Plannable Continuous Latent States6 Nov 2023
- 2023Understanding and Addressing the Pitfalls of Bisimulation-based Representations in Offline Reinforcement Learning26 Oct 2023
- 2023Spectral Entry-wise Matrix Estimation for Low-Rank Reinforcement Learning10 Oct 2023
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?9 Oct 2023
- 2023Representation Learning in Low-rank Slate-based Recommender Systems10 Sep 2023
- 2023Policy Finetuning in Reinforcement Learning via Design of Experiments using Offline Data10 Jul 2023
- 2023Efficient Model-Free Exploration in Low-Rank MDPs8 Jul 2023
- 2023Sample-Efficient Learning of POMDPs with Multiple Observations In Hindsight6 Jul 2023
- 2023Context-lumpable stochastic bandits22 Jun 2023
- 2023Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP21 Jun 2023
- 2023Sample Efficient Reinforcement Learning in Mixed Systems through Augmented Samples and Its Applications to Queueing Networks25 May 2023
- 2023The Benefits of Being Distributional: Small-Loss Bounds for Reinforcement Learning25 May 2023
- 2023Representations and Exploration for Deep Reinforcement Learning using Singular Value Decomposition1 May 2023
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learning14 Apr 2023
- 2023Representation Learning with Multi-Step Inverse Kinematics: An Efficient and Optimal Approach to Rich-Observation RL12 Apr 2023
- 2023Accelerating exploration and representation learning with offline pre-training31 Mar 2023
- 2023Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPs20 Mar 2023
- 2023Fast exploration and learning of latent graphs with aliased observations13 Mar 2023
- 2023Distributional Offline Policy Evaluation with Predictive Error Guarantees19 Feb 2023
- 2023Near-Minimax-Optimal Risk-Sensitive Reinforcement Learning with CVaR7 Feb 2023
- 2022Representation Learning in Deep RL via Discrete Information Bottleneck28 Dec 2022
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculum24 Dec 2022
- 2022Latent Variable Representation for Reinforcement Learning17 Dec 2022
- 2022Exploring through Random Curiosity with General Value Functions18 Nov 2022
50 newest of 114 All 114 (JSON)
- 2024Towards Principled Representation Learning from Videos for Reinforcement Learningcited by 10
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learningcited by 8
- 2024The Central Role of the Loss Function in Reinforcement Learningcited by 8
- 2024Exploration is Harder than Prediction: Cryptographically Separating Reinforcement Learning from Supervised Learningcited by 6
- 2024A Natural Extension To Online Algorithms For Hybrid RL With Limited Coveragecited by 5
- 2024Hybrid Reinforcement Learning from Offline Observation Alonecited by 5
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actionscited by 4
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learningcited by 3
- 2024Generalizing Multi-Step Inverse Models for Representation Learning to Finite-Memory POMDPscited by 2
- 2024Learning mirror maps in policy mirror descentcited by 1
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimationcited by 1
- 2025Accelerating Multi-Task Temporal Difference Learning under Low-Rank Representationcited by 1
12 most cited of 18 All 18 (JSON)
- 2026Towards Scaling Reinforcement Learning to Massive Populations: Learning Mean-Field Representations26 Aug 2026
- 2026End-to-End Efficient RL for Linear Bellman Complete MDPs with Deterministic Transitions24 Mar 2026
- 2026Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search for Hard Exploration23 Mar 2026
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learning3 Oct 2025
- 2025Where to Intervene: Action Selection in Deep Reinforcement Learning5 Jul 2025
- 2025Accelerating Multi-Task Temporal Difference Learning under Low-Rank Representation3 Mar 2025
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimation30 Oct 2024
- 2024The Central Role of the Loss Function in Reinforcement Learning19 Sep 2024
- 2024Rethinking State Disentanglement in Causal Reinforcement Learning24 Aug 2024
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actions17 Jun 2024
- 2024Hybrid Reinforcement Learning from Offline Observation Alone11 Jun 2024
- 2024Spatio-temporal Value Semantics-based Abstraction for Dense Deep Reinforcement Learning24 May 2024
12 newest of 18 All 18 (JSON)
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 129
- 2022The Role of Coverage in Online Reinforcement Learningcited by 83
- 2021Improving Zero-shot Generalization in Offline Reinforcement Learning using Generalized Similarity Functionscited by 17
- 2022Agent-Controller Representations: Principled Offline RL with Rich Exogenous Informationcited by 16
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 15
- 2024Offline Multi-task Transfer RL with Representational Penalizationcited by 14
- 2021Provably Efficient Representation Selection in Low-rank Markov Decision Processes: From Online to Offline RLcited by 12
- 2022Behavior Prior Representation learning for Offline Reinforcement Learningcited by 12
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guaranteescited by 11
- 2023Understanding and Addressing the Pitfalls of Bisimulation-based Representations in Offline Reinforcement Learningcited by 10
- 2024Offline Multi-task Transfer RL with Representational Penalization19 Feb 2024
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guarantees14 Nov 2023
- 2023Understanding and Addressing the Pitfalls of Bisimulation-based Representations in Offline Reinforcement Learning26 Oct 2023
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learning14 Apr 2023
- 2022Behavior Prior Representation learning for Offline Reinforcement Learning2 Nov 2022
- 2022Agent-Controller Representations: Principled Offline RL with Rich Exogenous Information31 Oct 2022
- 2022The Role of Coverage in Online Reinforcement Learning9 Oct 2022
- 2021Improving Zero-shot Generalization in Offline Reinforcement Learning using Generalized Similarity Functions29 Nov 2021
- 2021Representation Learning for Online and Offline RL in Low-rank MDPs9 Oct 2021
- 2021Provably Efficient Representation Selection in Low-rank Markov Decision Processes: From Online to Offline RL22 Jun 2021
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 178
- 2022Hybrid RL: Using Both Offline and Online Data Can Make RL Efficientcited by 161
- 2021Rapid Exploration for Open-World Navigation with Latent Goal Modelscited by 121
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 52
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2022Guaranteed Discovery of Control-Endogenous Latent States with Multi-Step Inverse Modelscited by 45
- 2021Learning Markov State Abstractions for Deep Reinforcement Learningcited by 44
- 2021Improved Regret Analysis for Variance-Adaptive Linear Bandits and Horizon-Free Linear Mixture MDPscited by 44
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
12 most cited of 85 All 85 (JSON)
- 2025Offline Action-Free Learning of Ex-BMDPs by Comparing Diverse Datasets26 Mar 2025
- 2024Learning a Fast Mixing Exogenous Block MDP using a Single Trajectory3 Oct 2024
- 2024Provable Interactive Learning with Hindsight Instruction Feedback14 Apr 2024
- 2024Multistep Inverse Is Not All You Need18 Mar 2024
- 2024Premier-TACO is a Few-Shot Policy Learner: Pretraining Multitask Representation via Temporal Action-Driven Contrastive Loss9 Feb 2024
- 2023Low-Rank MDPs with Continuous Action Spaces6 Nov 2023
- 2023PcLast: Discovering Plannable Continuous Latent States6 Nov 2023
- 2023Spectral Entry-wise Matrix Estimation for Low-Rank Reinforcement Learning10 Oct 2023
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?9 Oct 2023
- 2023Representation Learning in Low-rank Slate-based Recommender Systems10 Sep 2023
- 2023Policy Finetuning in Reinforcement Learning via Design of Experiments using Offline Data10 Jul 2023
- 2023Efficient Model-Free Exploration in Low-Rank MDPs8 Jul 2023
12 newest of 85 All 85 (JSON)