FLAMBE: Structural Complexity and Representation Learning of Low Rank MDPs
Not on the map. Why?
Cited by 40 in Offline RL and imitation learning from data, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
In order to deal with the curse of dimensionality in reinforcement learning (RL), it is common practice to make parametric assumptions where values or policies are functions of some low dimensional feature space. This work focuses on the representation learning question: how can we learn such features? Under the assumption that the underlying (unknown) dynamics correspond to a low rank transition matrix, we show how the representation learning question is related to a particular non-linear matrix decomposition problem. Structurally, we make precise connections between these low rank MDPs and latent variable models, showing how they significantly generalize prior formulations for representation learning in RL. Algorithmically, we develop FLAMBE, which engages in exploration and representation learning for provably efficient RL in low rank transition models.
Key facts. Cited by 232 papers in the map (data through 7 Oct 2026). First posted on arXiv 18 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 30; 2022: 51; 2023: 53; 2024: 48; 2025: 30; 2026: 19.
232 papers.
Sort:
By area:
- 2021Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithmscited by 222
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 160
- 2022Hybrid RL: Using Both Offline and Online Data Can Make RL Efficientcited by 158
- 2021Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement Learningcited by 157
- 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2022Offline Reinforcement Learning with Realizability and Single-policy Concentrabilitycited by 122
- 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 112
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 109
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
38 more
- 2024RLHF from Heterogeneous Feedback via Personalization and Preference Aggregationcited by 71
- 2023Reinforcement Learning with Human Feedback: Learning Dynamic Choices via Pessimismcited by 69
- 2022Denoised MDPs: Learning World Models Better Than the World Itselfcited by 66
- 2024Self-Exploring Language Models: Active Preference Elicitation for Online Alignmentcited by 63
- 2022Enforcing Hard Constraints with Soft Barriers: Safe Reinforcement Learning in Unknown Stochastic Environmentscited by 54
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2022Exploration via Elliptical Episodic Bonusescited by 51
- 2022On the Principles of Parsimony and Self-Consistency for the Emergence of Intelligencecited by 50
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2021The Power of Exploiter: Provable Multi-Agent RL in Large State Spacescited by 48
- 2023Making RL with Preference-based Feedback Efficient via Randomizationcited by 46
- 2021Simple Agent, Complex Environment: Efficient Reinforcement Learning with Agent Statescited by 42
- 2022PAC Reinforcement Learning for Predictive State Representationscited by 42
- 2021MADE: Exploration via Maximizing Deviation from Explored Regionscited by 40
- 2022Provable Reinforcement Learning with a Short-Term Memorycited by 40
- 2025Towards General-Purpose Model-Free Reinforcement Learningcited by 40
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2021The Information Geometry of Unsupervised Reinforcement Learningcited by 36
- 2022Provably Efficient Reinforcement Learning in Partially Observable Dynamical Systemscited by 35
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 35
- 2021Provable Representation Learning for Imitation with Contrastive Fourier Featurescited by 34
- 2025LoRe: Personalizing LLMs via Low-Rank Reward Modelingcited by 34
- 2024Large Language Models as Markov Chainscited by 33
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2023Near-Minimax-Optimal Risk-Sensitive Reinforcement Learning with CVaRcited 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
- 2023Adversarial Model for Offline Reinforcement Learningcited by 31
- 2021Robust Policy Gradient against Strong Data Corruptioncited by 30
- 2022Provable Benefit of Multitask Representation Learning in Reinforcement Learningcited by 29
- 2022Provably Efficient Model-Free Constrained RL with Linear Function Approximationcited by 29
- 2021Provable RL with Exogenous Distractors via Multistep Inverse Dynamicscited by 28
- 2021Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisitingcited by 27
- 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
- 2021Interactive Learning from Activity Descriptioncited by 26
- 2022Sample-Efficient Reinforcement Learning in the Presence of Exogenous Informationcited by 26
50 most cited of 232 All 232 (JSON)
- 2026A Decentralized Partially Observable Team Decision Methodology with Delayed Information Sharing22 Sep 2026
- 2026Towards Scaling Reinforcement Learning to Massive Populations: Learning Mean-Field Representations26 Aug 2026
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026What a World Model Represents Is Three Questions7 Jul 2026
- 2026BIFROST: Bridging Invariant Feature Representation for Observation-space Sim2Real Transfer1 Jul 2026
- 2026Revisiting Policy Gradients for Restricted Policy Classes: Escaping Myopic Local Optima with k-step Policy Gradients11 May 2026
- 2026Breaking the Computational Barrier: Provably Efficient Actor-Critic for Low-Rank MDPs2 May 2026
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representation1 May 2026
- 2026DARLING: Detection Augmented Reinforcement Learning with Non-Stationary Guarantees17 Apr 2026
- 2026Impact of Connectivity on Laplacian Representations in Reinforcement Learning9 Mar 2026
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2026Scaling Tasks, Not Samples: Mastering Humanoid Control through Multi-Task Model-Based Reinforcement Learning2 Mar 2026
38 more
- 2026PAC Guarantees for Reinforcement Learning: Sample Complexity, Coverage, and Structure1 Mar 2026
- 2026Multi-agent imitation learning with function approximation: Linear Markov games and beyond26 Feb 2026
- 2026OPRIDE: Offline Preference-based Reinforcement Learning via In-Dataset Exploration19 Feb 2026
- 2026Unifying Model-Free Efficiency and Model-Based Representations via Latent Dynamics13 Feb 2026
- 2026The Horizon Threshold in Cooperative Multi-Agent Reward-Free Exploration1 Feb 2026
- 2026Spectral Ghost in Representation Learning: from Component Analysis to Self-Supervised Learning28 Jan 2026
- 2026Principled Fine-tuning of LLMs from User-Edits: A Medley of Preference, Supervision, and Reward27 Jan 2026
- 2025Improved Bounds for Private and Robust Alignment29 Dec 2025
- 2025Generalisation in Multitask Fitted Q-Iteration and Offline Q-learning23 Dec 2025
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025ProSh: Probabilistic Shielding for Model-free Reinforcement Learning17 Oct 2025
- 2025Demystifying the Mechanisms Behind Emergent Exploration in Goal-conditioned RL15 Oct 2025
- 2025ADARL: Adaptive Low-Rank Structures for Robust Policy Learning under Uncertainty13 Oct 2025
- 2025Representation-Based Exploration for Language Models: From Test-Time to Post-Training13 Oct 2025
- 2025Offline Reinforcement Learning in Large State Spaces: Algorithms and Guarantees5 Oct 2025
- 2025On the Limits of Tabular Hardness Metrics for Deep RL: A Study with the Pharos Benchmark21 Sep 2025
- 2025Replicable Reinforcement Learning with Linear Function Approximation10 Sep 2025
- 2025Offline Imitation Learning upon Arbitrary Demonstrations by Pre-Training Dynamics Representations20 Aug 2025
- 2025Instance-Dependent Continuous-Time Reinforcement Learning via Maximum Likelihood Estimation4 Aug 2025
- 2025Unrolling Dynamic Programming via Graph Filters29 Jul 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Spectral Bellman Method: Unifying Representation and Exploration in RL17 Jul 2025
- 2025Convergence and Sample Complexity of First-Order Methods for Agnostic Reinforcement Learning6 Jul 2025
- 2025The Sample Complexity of Online Strategic Decision Making with Information Asymmetry and Knowledge Transportability11 Jun 2025
- 2025Transformers as Multi-task Learners: Decoupling Features in Hidden Markov Models2 Jun 2025
- 2025Policy-Driven World Model Adaptation for Robust Offline Model-based Reinforcement Learning19 May 2025
- 2025LoRe: Personalizing LLMs via Low-Rank Reward Modeling20 Apr 2025
- 2025A Classification View on Meta Learning Bandits6 Apr 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
- 2025Towards Understanding the Benefit of Multitask Representation Learning in Decision Process1 Mar 2025
- 2025Towards Auto-Regressive Next-Token Prediction: In-Context Learning Emerges from Generalization24 Feb 2025
- 2025Model Selection for Off-policy Evaluation: New Algorithms and Experimental Protocol11 Feb 2025
- 2025Towards General-Purpose Model-Free Reinforcement Learning27 Jan 2025
- 2025Addressing Finite-Horizon MDPs via Low-Rank Tensor Value Approximation17 Jan 2025
- 2025Multilinear Tensor Low-Rank Approximation for Policy-Gradient Methods in Reinforcement Learning8 Jan 2025
- 2024On the Linear Speedup of Personalized Federated Reinforcement Learning with Shared Representations22 Nov 2024
50 newest of 232 All 232 (JSON)
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 160
- 2021Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement Learningcited by 157
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2022Offline Reinforcement Learning with Realizability and Single-policy Concentrabilitycited by 122
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 109
- 2021Provable Representation Learning for Imitation with Contrastive Fourier Featurescited by 34
- 2023Adversarial Model for Offline Reinforcement Learningcited by 31
- 2022Proximal Point Imitation Learningcited by 25
- 2024More Benefits of Being Distributional: Second-Order Bounds for Reinforcement Learningcited by 23
- 2022Learning Bellman Complete Representations for Offline Policy Evaluationcited by 17
- 2020DeepAveragers: Offline Reinforcement Learning by Solving Derived Non-Parametric MDPscited by 16
- 2023Model-based Offline Reinforcement Learning with Count-based Conservatismcited by 16
12 most cited of 37 All 37 (JSON)
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2026Multi-agent imitation learning with function approximation: Linear Markov games and beyond26 Feb 2026
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025Offline Reinforcement Learning in Large State Spaces: Algorithms and Guarantees5 Oct 2025
- 2025Offline Imitation Learning upon Arbitrary Demonstrations by Pre-Training Dynamics Representations20 Aug 2025
- 2025Model Selection for Off-policy Evaluation: New Algorithms and Experimental Protocol11 Feb 2025
- 2024Inverse Reinforcement Learning from Non-Stationary Learning Agents18 Oct 2024
- 2024On the Global Optimality of Policy Gradient Methods in General Utility Reinforcement Learning5 Oct 2024
- 2024Exploiting Structure in Offline Multi-Agent RL: The Benefits of Low Interaction Rank1 Oct 2024
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning10 Jul 2024
- 2024A Fast Convergence Theory for Offline Decision Making3 Jun 2024
- 2024Provably Efficient Off-Policy Adversarial Imitation Learning with Convergence Guarantees26 May 2024
12 newest of 37 All 37 (JSON)
- 2024RLHF from Heterogeneous Feedback via Personalization and Preference Aggregationcited by 71
- 2023Reinforcement Learning with Human Feedback: Learning Dynamic Choices via Pessimismcited by 69
- 2024Self-Exploring Language Models: Active Preference Elicitation for Online Alignmentcited by 63
- 2023Making RL with Preference-based Feedback Efficient via Randomizationcited by 46
- 2025LoRe: Personalizing LLMs via Low-Rank Reward Modelingcited by 34
- 2023Provable Reward-Agnostic Preference-Based Reinforcement Learningcited by 25
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Modelscited by 16
- 2024Accelerated Preference Optimization for Large Language Model Alignmentcited by 5
- 2025Improved Bounds for Private and Robust Alignmentcited by 3
- 2025Improved Bounds for Private and Robust Alignment29 Dec 2025
- 2025LoRe: Personalizing LLMs via Low-Rank Reward Modeling20 Apr 2025
- 2024Accelerated Preference Optimization for Large Language Model Alignment8 Oct 2024
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Models6 Jun 2024
- 2024Self-Exploring Language Models: Active Preference Elicitation for Online Alignment29 May 2024
- 2024RLHF from Heterogeneous Feedback via Personalization and Preference Aggregation30 Apr 2024
- 2023Making RL with Preference-based Feedback Efficient via Randomization23 Oct 2023
- 2023Provable Reward-Agnostic Preference-Based Reinforcement Learning29 May 2023
- 2023Reinforcement Learning with Human Feedback: Learning Dynamic Choices via Pessimism29 May 2023
- 2022Denoised MDPs: Learning World Models Better Than the World Itselfcited by 66
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 20
- 2022Learning General World Models in a Handful of Reward-Free Deploymentscited by 12
- 2025Policy-Driven World Model Adaptation for Robust Offline Model-based Reinforcement Learningcited by 2
- 2026What a World Model Represents Is Three Questionscited by 0
- 2026What a World Model Represents Is Three Questions7 Jul 2026
- 2025Policy-Driven World Model Adaptation for Robust Offline Model-based Reinforcement Learning19 May 2025
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2022Learning General World Models in a Handful of Reward-Free Deployments23 Oct 2022
- 2022Denoised MDPs: Learning World Models Better Than the World Itself30 Jun 2022
- 2021Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithmscited by 222
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2022Hybrid RL: Using Both Offline and Online Data Can Make RL Efficientcited by 158
- 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
- 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 112
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Enforcing Hard Constraints with Soft Barriers: Safe Reinforcement Learning in Unknown Stochastic Environmentscited by 54
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2022Exploration via Elliptical Episodic Bonusescited by 51
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
12 most cited of 178 All 178 (JSON)
- 2026A Decentralized Partially Observable Team Decision Methodology with Delayed Information Sharing22 Sep 2026
- 2026Towards Scaling Reinforcement Learning to Massive Populations: Learning Mean-Field Representations26 Aug 2026
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026BIFROST: Bridging Invariant Feature Representation for Observation-space Sim2Real Transfer1 Jul 2026
- 2026Revisiting Policy Gradients for Restricted Policy Classes: Escaping Myopic Local Optima with k-step Policy Gradients11 May 2026
- 2026Breaking the Computational Barrier: Provably Efficient Actor-Critic for Low-Rank MDPs2 May 2026
- 2026Near-optimal and Efficient First-Order Algorithm for Multi-Task Learning with Shared Linear Representation1 May 2026
- 2026DARLING: Detection Augmented Reinforcement Learning with Non-Stationary Guarantees17 Apr 2026
- 2026Impact of Connectivity on Laplacian Representations in Reinforcement Learning9 Mar 2026
- 2026Scaling Tasks, Not Samples: Mastering Humanoid Control through Multi-Task Model-Based Reinforcement Learning2 Mar 2026
- 2026PAC Guarantees for Reinforcement Learning: Sample Complexity, Coverage, and Structure1 Mar 2026
- 2026OPRIDE: Offline Preference-based Reinforcement Learning via In-Dataset Exploration19 Feb 2026
12 newest of 178 All 178 (JSON)