Provable Representation with Efficient Planning for Partially Observable Reinforcement Learning
Hongming Zhang, Tongzheng Ren, Chenjun Xiao, Dale Schuurmans, Bo Dai
Abstract
In most real-world reinforcement learning applications, state information is only partially observable, which breaks the Markov decision process assumption and leads to inferior performance for algorithms that conflate observations with state. Partially Observable Markov Decision Processes (POMDPs), on the other hand, provide a general framework that allows for partial observability to be accounted for in learning, exploration and planning, but presents significant computational and statistical challenges. To address these difficulties, we develop a representation-based perspective that leads to a coherent framework and tractable algorithmic approach for practical reinforcement learning from partial observations. We provide a theoretical analysis for justifying the statistical efficiency of the proposed algorithm, and also empirically demonstrate the proposed algorithm can surpass state-of-the-art performance with partial observations across various benchmarks, advancing reliable reinforcement learning towards more practical applications.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 484c45e2-6de5-47d2-b79f-d41057fc8937Cited by top-tier papers6
- Diffusion Spectral Representation for Reinforcement LearningDmitry Shribak, Chen-Xiao Gao, Yitong Li, Chenjun Xiao et al.NeurIPS 2024 · 18 citations
- Mean-Field Sampling for Cooperative Multi-Agent Reinforcement LearningEmile Anand, Ishani Karmarkar, Guannan QuNeurIPS 2025 · 10 citations
- Exploiting the Replay Memory Before Exploring the Environment: Enhancing Reinforcement Learning Through Empirical MDP IterationHongming Zhang, Chenjun Xiao, Chao Gao, Han Wang et al.NeurIPS 2024 · 7 citations
- On the Role of Information Structure in Reinforcement Learning for Partially-Observable Sequential Teams and GamesAwni Altabaa, Zhuoran YangNeurIPS 2024 · 5 citations
- Principled Fast and Meta Knowledge Learners for Continual Reinforcement LearningKe Sun, Hongming Zhang, Jun Jin, Chao Gao et al.ICLR 2026 · 1 citation
Builds on25
- Dream to Control: Learning Behaviors by Latent ImaginationDanijar Hafner, Timothy P. Lillicrap, Jimmy Ba, Mohammad NorouziICLR 2020 · 1,852 citations
- Mastering Atari with Discrete World ModelsDanijar Hafner, Timothy P. Lillicrap, Mohammad Norouzi, Jimmy BaICLR 2021 · 1,170 citations
- Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from PixelsDenis Yarats, Ilya Kostrikov, Rob FergusICLR 2021 · 911 citations
- Improving Sample Efficiency in Model-Free Reinforcement Learning from ImagesDenis Yarats, Amy Zhang, Ilya Kostrikov, Brandon Amos et al.AAAI 2021 · 506 citations
- Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement LearningDenis Yarats, Rob Fergus, Alessandro Lazaric, Lerrel PintoICLR 2022 · 457 citations
Related papers
- Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDPJiacheng Guo, Zihao Li, Huazheng Wang, Mengdi Wang et al.ICML 2023 · 8 citations
- Learning Belief Representations for Partially Observable Deep RLAndrew Wang, Andrew C. Li, Toryn Q. Klassen, Rodrigo Toro Icarte et al.ICML 2023 · 21 citations
- Sample-Efficient Reinforcement Learning of Undercomplete POMDPsChi Jin, Sham M. Kakade, Akshay Krishnamurthy, Qinghua LiuNeurIPS 2020 · 88 citations
- Partially Observable RL with B-Stability: Unified Structural Condition and Sharp Sample-Efficient AlgorithmsFan Chen, Yu Bai, Song MeiICLR 2023 · 2 citations
- Represent to Control Partially Observed Systems: Representation Learning with Provable Sample EfficiencyLingxiao Wang, Qi Cai, Zhuoran Yang, Zhaoran WangICLR 2023
