Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret Bound
Not on the map. Why?
Cited by 40 in Deep reinforcement learning methods, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Exploration in reinforcement learning (RL) suffers from the curse of dimensionality when the state-action space is large. A common practice is to parameterize the high-dimensional value and policy functions using given features. However existing methods either have no theoretical guarantee or suffer a regret that is exponential in the planning horizon H. In this paper, we propose an online RL algorithm, namely the MatrixRL, that leverages ideas from linear bandit to learn a low-dimensional representation of the probability transition model while carefully balancing the exploitation-exploration tradeoff. We show that MatrixRL achieves a regret bound Obig(H2dlog T√Tbig) where d is the number of features. MatrixRL has an equivalent kernelized version, which is able to work with an arbitrary kernel Hilbert space without using explicit features. In this case, the kernelized MatrixRL satisfies a regret bound Obig(H2widetildedlog T√Tbig), where widetilded is the effective dimension of the kernel space. To our best knowledge, for RL using features or kernels, our results are the first regret bounds that are near-optimal in time T and dimension d (or widetilded) and polynomial in the planning horizon H.
Key facts. Cited by 201 papers in the map (data through 14 Oct 2026). First posted on arXiv 24 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 55; 2022: 42; 2023: 45; 2024: 31; 2025: 16; 2026: 10.
201 papers.
Sort:
By area:
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 164
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 129
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 110
- 2021Recent Advances in Reinforcement Learning in Financecited by 110
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 65
- 2022Unpacking Reward Shaping: Understanding the Benefits of Reward Engineering on Sample Complexitycited by 62
- 2022Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision Processescited by 57
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learningcited by 55
- 2021Cautiously Optimistic Policy Optimization and Exploration with Linear Function Approximationcited by 53
- 2021Policy Optimization in Adversarial MDPs: Improved Exploration via Dilated Bonusescited by 52
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
38 more
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 52
- 2021Breaking the Deadly Triad with a Target Networkcited by 47
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 46
- 2023Reason for Future, Act for Now: A Principled Framework for Autonomous LLM Agents with Provable Sample Efficiencycited by 46
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2021Near-optimal Representation Learning for Linear Bandits and Linear RLcited by 45
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2021Improved Regret Analysis for Variance-Adaptive Linear Bandits and Horizon-Free Linear Mixture MDPscited by 44
- 2021Towards General Function Approximation in Zero-Sum Markov Gamescited by 43
- 2022PAC Reinforcement Learning for Predictive State Representationscited by 43
- 2022Leveraging Offline Data in Online Reinforcement Learningcited by 42
- 2021A Provably Efficient Algorithm for Linear Markov Decision Process with Low Switching Costcited by 40
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2023Variance-Dependent Regret Bounds for Linear Bandits and Reinforcement Learning: Adaptivity and Computational Efficiencycited by 40
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2022Provably Efficient Reinforcement Learning in Partially Observable Dynamical Systemscited by 35
- 2021Provable Representation Learning for Imitation with Contrastive Fourier Featurescited by 34
- 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
- 2022Target Network and Truncation Overcome The Deadly Triad in Q-Learningcited by 31
- 2022A General Framework for Sample-Efficient Function Approximation in Reinforcement Learningcited by 31
- 2021Robust Policy Gradient against Strong Data Corruptioncited by 30
- 2022Nearly Minimax Optimal Reinforcement Learning with Linear Function Approximationcited by 30
- 2021Reward-Free Model-Based Reinforcement Learning with Linear Function Approximationcited by 29
- 2022Provable Benefit of Multitask Representation Learning in Reinforcement Learningcited by 29
- 2023Provable and Practical: Efficient Exploration in Reinforcement Learning via Langevin Monte Carlocited by 29
- 2021Near-optimal Policy Optimization Algorithms for Learning Adversarial Linear Mixture MDPscited by 28
- 2021Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisitingcited by 28
- 2023Breaking the Curse of Multiagents in a Large State Space: RL in Markov Games with Independent Linear Function Approximationcited by 27
- 2022Contrastive UCB: Provably Efficient Contrastive Self-Supervised Learning in Online Reinforcement Learningcited by 26
- 2021Almost Optimal Algorithms for Two-player Zero-Sum Linear Mixture Markov Gamescited by 25
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
- 2024Distributionally Robust Off-Dynamics Reinforcement Learning: Provable Efficiency with Linear Function Approximationcited by 25
- 2020Sparse Feature Selection Makes Batch Reinforcement Learning More Sample Efficientcited by 24
- 2021A Free Lunch from the Noise: Provable and Practical Exploration for Representation Learningcited by 24
- 2021Can Reinforcement Learning Find Stackelberg-Nash Equilibria in General-Sum Markov Games with Myopic Followers?cited by 24
- 2021Provably Efficient Cooperative Multi-Agent Reinforcement Learning with Function Approximationcited by 23
- 2021Efficient Local Planning with Linear Function Approximationcited by 23
50 most cited of 201 All 201 (JSON)
- 2026Mini-Batch Risk-Averse Deep Q-Learning: A Robot Navigation Case Study7 Sep 2026
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost31 Aug 2026
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026On Generalizing Preference-based Reinforcement Learning to Multi-Objective Problems13 Jul 2026
- 2026Learning to Adapt: Representation-Based Reinforcement Learning for Multi-Task Skill Transfer11 Jun 2026
- 2026Learning Kernel-Based MDPs from Episodic Preferential Feedback22 May 2026
- 2026Spectral Souping: A Unified Framework for Online Preference Alignment19 May 2026
- 2026Taming the Curses of Multiagency in Robust Markov Games with Large State Space through Linear Function Approximation4 May 2026
- 2026Reinforcement Learning with Markov Risk Measures and Multipattern Risk Approximation1 May 2026
- 2026Laplacian Kernelized Bandit1 Jan 2026
- 2025Fitted Q-Evaluation without Bellman Completeness via Occupancy Weighting29 Dec 2025
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
38 more
- 2025Distributionally Robust Online Markov Game with Linear Function Approximation11 Nov 2025
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes23 Oct 2025
- 2025Linear Dynamics meets Linear MDPs: Closed-Form Optimal Policies via Reinforcement Learning24 Aug 2025
- 2025Generalized Kernelized Bandits: A Novel Self-Normalized Bernstein-Like Dimension-Free Inequality and Regret Bounds3 Aug 2025
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Model30 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
- 2025Generalized Linear Markov Decision Process1 Jun 2025
- 2025A Provable Approach for End-to-End Safe Reinforcement Learning28 May 2025
- 2025Near-Optimal Sample Complexity in Reward-Free Kernel-Based Reinforcement Learning11 Feb 2025
- 2025The Sample Complexity of Online Reinforcement Learning: A Multi-model Perspective27 Jan 2025
- 2024Hyper: Hyperparameter Robust Efficient Exploration in Reinforcement Learning4 Dec 2024
- 2024Efficient, Low-Regret, Online Reinforcement Learning for Linear MDPs16 Nov 2024
- 2024Performative Reinforcement Learning with Linear Markov Decision Process7 Nov 2024
- 2024Demystifying Linear MDPs and Novel Dynamics Aggregation Framework31 Oct 2024
- 2024The Limits of Transfer Reinforcement Learning with Latent Low-rank Structure28 Oct 2024
- 2024Primal-Dual Spectral Representation for Off-policy Evaluation23 Oct 2024
- 2024Learning Infinite-Horizon Average-Reward Linear Mixture MDPs of Bounded Span19 Oct 2024
- 2024Neural Combinatorial Clustered Bandits for Recommendation Systems18 Oct 2024
- 2024Upper and Lower Bounds for Distributionally Robust Off-Dynamics Reinforcement Learning30 Sep 2024
- 2024BECAUSE: Bilinear Causal Representation for Generalizable Offline Model-based Reinforcement Learning15 Jul 2024
- 2024Spectral Representation for Causal Estimation with Hidden Confounders15 Jul 2024
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2024More Efficient Randomized Exploration for Reinforcement Learning via Approximate Sampling18 Jun 2024
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actions17 Jun 2024
- 2024Graph Neural Thompson Sampling15 Jun 2024
- 2024Pretraining Decision Transformers with Reward Prediction for In-Context Multi-task Structured Bandit Learning7 Jun 2024
- 2024Bayesian Design Principles for Offline-to-Online Reinforcement Learning31 May 2024
- 2024Randomized Exploration for Reinforcement Learning with Multinomial Logistic Function Approximation30 May 2024
- 2024Provably Efficient Reinforcement Learning with Multinomial Logit Function Approximation27 May 2024
- 2024Achieving Constant Regret in Linear Markov Decision Processes16 Apr 2024
- 2024Efficient Duple Perturbation Robustness in Low-rank MDPs11 Apr 2024
- 2024Skill Transfer and Discovery for Sim-to-Real Learning: A Representation-Based Viewpoint7 Apr 2024
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes19 Mar 2024
- 2024RL in Markov Games with Independent Function Approximation: Improved Sample Complexity Bound under the Local Access Model18 Mar 2024
50 newest of 201 All 201 (JSON)
- 2024Distributionally Robust Off-Dynamics Reinforcement Learning: Provable Efficiency with Linear Function Approximationcited by 25
- 2024Upper and Lower Bounds for Distributionally Robust Off-Dynamics Reinforcement Learningcited by 16
- 2024Towards Robust Model-Based Reinforcement Learning Against Adversarial Corruptioncited by 12
- 2024More Efficient Randomized Exploration for Reinforcement Learning via Approximate Samplingcited by 12
- 2024Provably Efficient Reinforcement Learning with Multinomial Logit Function Approximationcited by 10
- 2024Performative Reinforcement Learning with Linear Markov Decision Processcited by 9
- 2024Randomized Exploration for Reinforcement Learning with Multinomial Logistic Function Approximationcited by 7
- 2025The Sample Complexity of Online Reinforcement Learning: A Multi-model Perspectivecited by 6
- 2024Long-term Safe Reinforcement Learning with Binary Feedbackcited by 4
- 2024Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actionscited by 4
- 2024Primal-Dual Spectral Representation for Off-policy Evaluationcited by 4
- 2025Spectral Bellman Method: Unifying Representation and Exploration in RLcited by 4
12 most cited of 37 All 37 (JSON)
- 2026Mini-Batch Risk-Averse Deep Q-Learning: A Robot Navigation Case Study7 Sep 2026
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost31 Aug 2026
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026On Generalizing Preference-based Reinforcement Learning to Multi-Objective Problems13 Jul 2026
- 2026Learning to Adapt: Representation-Based Reinforcement Learning for Multi-Task Skill Transfer11 Jun 2026
- 2026Taming the Curses of Multiagency in Robust Markov Games with Large State Space through Linear Function Approximation4 May 2026
- 2026Reinforcement Learning with Markov Risk Measures and Multipattern Risk Approximation1 May 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Distributionally Robust Online Markov Game with Linear Function Approximation11 Nov 2025
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes23 Oct 2025
- 2025Linear Dynamics meets Linear MDPs: Closed-Form Optimal Policies via Reinforcement Learning24 Aug 2025
12 newest of 37 All 37 (JSON)
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 164
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 129
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 110
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 65
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2022Leveraging Offline Data in Online Reinforcement Learningcited by 42
- 2021Provable Representation Learning for Imitation with Contrastive Fourier Featurescited by 34
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
- 2020Sparse Feature Selection Makes Batch Reinforcement Learning More Sample Efficientcited by 24
- 2023Model-based Offline Reinforcement Learning with Count-based Conservatismcited by 16
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processescited by 14
- 2024Bayesian Design Principles for Offline-to-Online Reinforcement Learningcited by 14
12 most cited of 26 All 26 (JSON)
- 2025Generalized Linear Markov Decision Process1 Jun 2025
- 2024BECAUSE: Bilinear Causal Representation for Generalizable Offline Model-based Reinforcement Learning15 Jul 2024
- 2024Pretraining Decision Transformers with Reward Prediction for In-Context Multi-task Structured Bandit Learning7 Jun 2024
- 2024Bayesian Design Principles for Offline-to-Online Reinforcement Learning31 May 2024
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes19 Mar 2024
- 2023Unsupervised Behavior Extraction via Random Intent Priors28 Oct 2023
- 2023Bi-Level Offline Policy Optimization with Limited Exploration10 Oct 2023
- 2023Pessimistic Nonlinear Least-Squares Value Iteration for Offline Reinforcement Learning2 Oct 2023
- 2023Stackelberg Batch Policy Learning28 Sep 2023
- 2023Model-based Offline Reinforcement Learning with Count-based Conservatism21 Jul 2023
- 2023Matrix Estimation for Offline Reinforcement Learning with Low-Rank Structure24 May 2023
- 2023What can online reinforcement learning with function approximation benefit from general coverage conditions?25 Apr 2023
12 newest of 26 All 26 (JSON)
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2021Recent Advances in Reinforcement Learning in Financecited by 110
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Unpacking Reward Shaping: Understanding the Benefits of Reward Engineering on Sample Complexitycited by 62
- 2022Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision Processescited by 57
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learningcited by 55
- 2021Cautiously Optimistic Policy Optimization and Exploration with Linear Function Approximationcited by 53
- 2021Policy Optimization in Adversarial MDPs: Improved Exploration via Dilated Bonusescited by 52
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 52
- 2021Breaking the Deadly Triad with a Target Networkcited by 47
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 46
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
12 most cited of 133 All 133 (JSON)
- 2026Laplacian Kernelized Bandit1 Jan 2026
- 2025Fitted Q-Evaluation without Bellman Completeness via Occupancy Weighting29 Dec 2025
- 2025Generalized Kernelized Bandits: A Novel Self-Normalized Bernstein-Like Dimension-Free Inequality and Regret Bounds3 Aug 2025
- 2025The Sample Complexity of Online Strategic Decision Making with Information Asymmetry and Knowledge Transportability11 Jun 2025
- 2024Demystifying Linear MDPs and Novel Dynamics Aggregation Framework31 Oct 2024
- 2024Learning Infinite-Horizon Average-Reward Linear Mixture MDPs of Bounded Span19 Oct 2024
- 2024Neural Combinatorial Clustered Bandits for Recommendation Systems18 Oct 2024
- 2024Spectral Representation for Causal Estimation with Hidden Confounders15 Jul 2024
- 2024Graph Neural Thompson Sampling15 Jun 2024
- 2024Skill Transfer and Discovery for Sim-to-Real Learning: A Representation-Based Viewpoint7 Apr 2024
- 2024RL in Markov Games with Independent Function Approximation: Improved Sample Complexity Bound under the Local Access Model18 Mar 2024
- 2024Tree Search-Based Evolutionary Bandits for Protein Sequence Optimization8 Jan 2024
12 newest of 133 All 133 (JSON)