Provably efficient RL with Rich Observations via Latent State Decoding
Not on the map. Why?
Abstract (arXiv)
We study the exploration problem in episodic MDPs with rich observations generated from a small number of latent states. Under certain identifiability assumptions, we demonstrate how to estimate a mapping from the observations to latent states inductively through a sequence of regression and clustering steps -- where previously decoded latent states provide labels for later regression problems -- and use it to construct good exploration policies. We provide finite-sample guarantees on the quality of the learned state decoding function and exploration policies, and complement our theory with an empirical evaluation on a class of hard exploration problems. Our method exponentially improves over Q-learning with na"ive exploration, even when Q-learning has cheating access to latent states.
Key facts. Cited by 169 papers in the map (data through 7 Oct 2026). First posted on arXiv 25 Jan 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 37; 2022: 52; 2023: 35; 2024: 21; 2025: 15; 2026: 9.
169 papers.
Sort:
By area:
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022The Role of Coverage in Online Reinforcement Learningcited by 82
- 2022Can Foundation Models Perform Zero-Shot Task Specification For Robot Manipulation?cited by 70
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 69
- 2022Denoised MDPs: Learning World Models Better Than the World Itselfcited by 66
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 63
- 2021Learning Task Informed Abstractionscited by 58
38 more
- 2022Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learningcited by 58
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2021Learning Markov State Abstractions for Deep Reinforcement Learningcited by 42
- 2021Towards General Function Approximation in Zero-Sum Markov Gamescited by 42
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 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
- 2021A Provably Efficient Algorithm for Linear Markov Decision Process with Low Switching Costcited by 38
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2022On the Power of Foundation Modelscited by 37
- 2022Provably Efficient Reinforcement Learning in Partially Observable Dynamical Systemscited by 35
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 35
- 2022Offline Reinforcement Learning Under Value and Density-Ratio Realizability: The Power of Gapscited by 34
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022On the Complexity of Adversarial Decision Makingcited by 32
- 2022Model-based RL with Optimistic Posterior Sampling: Structural Conditions and Sample Complexitycited by 31
- 2022A General Framework for Sample-Efficient Function Approximation in Reinforcement Learningcited by 31
- 2022Learning in Observable POMDPs, without Computationally Intractable Oraclescited by 30
- 2022Provable Benefit of Multitask Representation Learning in Reinforcement Learningcited by 29
- 2023Tight Guarantees for Interactive Decision Making with the Decision-Estimation Coefficientcited by 29
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergencecited by 29
- 2021Near-optimal Policy Optimization Algorithms for Learning Adversarial Linear Mixture MDPscited by 28
- 2021Provable RL with Exogenous Distractors via Multistep Inverse Dynamicscited 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
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learningcited by 26
- 2021Block Contextual MDPs for Continual Learningcited by 25
- 2021Learning Representations for Pixel-based Control: What Matters and Why?cited 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
- 2022Partially Observable RL with B-Stability: Unified Structural Condition and Sharp Sample-Efficient Algorithmscited by 24
- 2022A Comprehensive Survey of Data Augmentation in Visual Reinforcement Learningcited by 23
- 2023Representation Learning with Multi-Step Inverse Kinematics: An Efficient and Optimal Approach to Rich-Observation RLcited by 22
50 most cited of 169 All 169 (JSON)
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026What Must Generalist Agents Remember?17 Jun 2026
- 2026Learning to Perceive the World Through Control: Empowerment-Based Representation Learning28 May 2026
- 2026Compositional Transduction with Latent Analogies for Offline Goal-Conditioned Reinforcement Learning20 May 2026
- 2026Smaller Abstract State Spaces Enable Cross-Scale Generalization in Reinforcement Learning19 May 2026
- 2026Revisiting Policy Gradients for Restricted Policy Classes: Escaping Myopic Local Optima with k-step Policy Gradients11 May 2026
- 2026End-to-End Efficient RL for Linear Bellman Complete MDPs with Deterministic Transitions24 Mar 2026
- 2026PAC Guarantees for Reinforcement Learning: Sample Complexity, Coverage, and Structure1 Mar 2026
- 2026Perception-Based Beliefs for POMDPs with Visual Observations5 Feb 2026
- 2025Generalisation in Multitask Fitted Q-Iteration and Offline Q-learning23 Dec 2025
- 2025Dual Goal Representations8 Oct 2025
- 2025Scalable Policy-Based RL Algorithms for POMDPs8 Oct 2025
38 more
- 2025Multi-Task Reinforcement Learning with Language-Encoded Gated Policy Networks7 Oct 2025
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learning3 Oct 2025
- 2025Reinforcement Learning from Adversarial Preferences in Tabular MDPs15 Jul 2025
- 2025Learning Task Belief Similarity with Latent Dynamics for Meta-Reinforcement Learning24 Jun 2025
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning20 Jun 2025
- 2025Distances for Markov chains from sample streams23 May 2025
- 2025DSADF: Thinking Fast and Slow for Decision Making13 May 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
- 2025Statistical Tractability of Off-policy Evaluation of History-dependent Policies in POMDPs3 Mar 2025
- 2025On Generalization Across Environments In Multi-Objective Reinforcement Learning2 Mar 2025
- 2025Towards Understanding the Benefit of Multitask Representation Learning in Decision Process1 Mar 2025
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimation30 Oct 2024
- 2024Guarantees for Nonlinear Representation Learning: Non-identical Covariates, Dependent Data, Fewer Samples15 Oct 2024
- 2024Learning a Fast Mixing Exogenous Block MDP using a Single Trajectory3 Oct 2024
- 2024DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractors26 Sep 2024
- 2024Rethinking State Disentanglement in Causal Reinforcement Learning24 Aug 2024
- 2024Satisficing Exploration for Deep Reinforcement Learning16 Jul 2024
- 2024Three Dogmas of Reinforcement Learning15 Jul 2024
- 2024Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy10 Jul 2024
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actions17 Jun 2024
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learning2 Jun 2024
- 2024Preparing for Black Swans: The Antifragility Imperative for Machine Learning18 May 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
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalization16 Mar 2024
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractors15 Mar 2024
- 2024A Natural Extension To Online Algorithms For Hybrid RL With Limited Coverage7 Mar 2024
- 2024Improved Algorithm for Adversarial Linear Mixture MDPs with Bandit Feedback and Unknown Transition7 Mar 2024
- 2024On the Curses of Future and History in Future-dependent Value Functions for Off-policy Evaluation22 Feb 2024
- 2024Offline Multi-task Transfer RL with Representational Penalization19 Feb 2024
- 2023Provable Representation with Efficient Planning for Partial Observable Reinforcement Learning20 Nov 2023
- 2023Provably Efficient CVaR RL in Low-rank MDPs20 Nov 2023
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guarantees14 Nov 2023
- 2023Learning Adversarial Low-rank Markov Decision Processes with Unknown Transition and Full-information Feedback14 Nov 2023
- 2023Low-Rank MDPs with Continuous Action Spaces6 Nov 2023
50 newest of 169 All 169 (JSON)
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2022The Role of Coverage in Online Reinforcement Learningcited by 82
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 63
- 2022Offline Reinforcement Learning Under Value and Density-Ratio Realizability: The Power of Gapscited by 34
- 2024Offline Multi-task Transfer RL with Representational Penalizationcited by 13
- 2021Provably Efficient Representation Selection in Low-rank Markov Decision Processes: From Online to Offline RLcited by 12
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guaranteescited by 11
- 2023SeMAIL: Eliminating Distractors in Visual Imitation via Separated Modelscited by 10
- 2024DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractorscited by 5
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasetscited by 1
- 2026Compositional Transduction with Latent Analogies for Offline Goal-Conditioned Reinforcement Learningcited by 1
- 2026Compositional Transduction with Latent Analogies for Offline Goal-Conditioned Reinforcement Learning20 May 2026
- 2024DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractors26 Sep 2024
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024Offline Multi-task Transfer RL with Representational Penalization19 Feb 2024
- 2023Offline Data Enhanced On-Policy Policy Gradient with Provable Guarantees14 Nov 2023
- 2023SeMAIL: Eliminating Distractors in Visual Imitation via Separated Models19 Jun 2023
- 2022The Role of Coverage in Online Reinforcement Learning9 Oct 2022
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient3 Oct 2022
- 2022Offline Reinforcement Learning Under Value and Density-Ratio Realizability: The Power of Gaps25 Mar 2022
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimism11 Mar 2022
- 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
- 2021Reinforcement Learning in Reward-Mixing MDPscited by 15
- 2021Reinforcement Learning in Reward-Mixing MDPs7 Oct 2021
- 2025DSADF: Thinking Fast and Slow for Decision Makingcited by 6
- 2025DSADF: Thinking Fast and Slow for Decision Making13 May 2025
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Can Foundation Models Perform Zero-Shot Task Specification For Robot Manipulation?cited by 70
- 2021RL for Latent MDPs: Regret Guarantees and a Lower Boundcited by 69
- 2021Learning Task Informed Abstractionscited by 58
- 2022Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learningcited by 58
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
12 most cited of 151 All 151 (JSON)
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026What Must Generalist Agents Remember?17 Jun 2026
- 2026Learning to Perceive the World Through Control: Empowerment-Based Representation Learning28 May 2026
- 2026Smaller Abstract State Spaces Enable Cross-Scale Generalization in Reinforcement Learning19 May 2026
- 2026Revisiting Policy Gradients for Restricted Policy Classes: Escaping Myopic Local Optima with k-step Policy Gradients11 May 2026
- 2026End-to-End Efficient RL for Linear Bellman Complete MDPs with Deterministic Transitions24 Mar 2026
- 2026PAC Guarantees for Reinforcement Learning: Sample Complexity, Coverage, and Structure1 Mar 2026
- 2026Perception-Based Beliefs for POMDPs with Visual Observations5 Feb 2026
- 2025Generalisation in Multitask Fitted Q-Iteration and Offline Q-learning23 Dec 2025
- 2025Dual Goal Representations8 Oct 2025
- 2025Scalable Policy-Based RL Algorithms for POMDPs8 Oct 2025
- 2025Multi-Task Reinforcement Learning with Language-Encoded Gated Policy Networks7 Oct 2025
12 newest of 151 All 151 (JSON)