Spectral Entry-wise Matrix Estimation for Low-Rank Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
We study matrix estimation problems arising in reinforcement learning (RL) with low-rank structure. In low-rank bandits, the matrix to be recovered specifies the expected arm rewards, and for low-rank Markov Decision Processes (MDPs), it may for example characterize the transition kernel of the MDP. In both cases, each entry of the matrix carries important information, and we seek estimation methods with low entry-wise error. Importantly, these methods further need to accommodate for inherent correlations in the available data (e.g. for MDPs, the data consists of system trajectories). We investigate the performance of simple spectral-based matrix estimation approaches: we show that they efficiently recover the singular subspaces of the matrix and exhibit nearly-minimal entry-wise error. These new results on low-rank matrix estimation make it possible to devise reinforcement learning algorithms that fully exploit the underlying low-rank structure. We provide two examples of such algorithms: a regret minimization algorithm for low-rank bandit problems, and a best policy identification algorithm for reward-free RL in low-rank MDPs. Both algorithms yield state-of-the-art performance guarantees.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 42 of its 64 references that are in the map. First posted on arXiv 10 Oct 2023.
Builds on
42 of 64 references.
References · 42
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 760
- 2015An Introduction to Matrix Concentration Inequalitiescited by 397
- 2016Contextual Decision Processes with Low Bellman Rank are PAC-Learnablecited by 292
- 2020Decoupling Representation Learning from Reinforcement Learningcited by 269
- 2020FLAMBE: Structural Complexity and Representation Learning of Low Rank MDPscited by 233
- 2020Reward-Free Exploration for Reinforcement Learningcited by 177
- 2019Provably efficient RL with Rich Observations via Latent State Decodingcited by 169
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 129
- 2019Kinematic State Abstraction and Provably Efficient Rich-Observation Reinforcement Learningcited by 114
- 2020Spectral Methods for Data Science: A Statistical Perspectivecited by 112
- 2020Fast active learning for pure exploration in reinforcement learningcited by 92
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
30 more
- 2020Instance-Dependent Complexity of Contextual Bandits and Reinforcement Learning: A Disagreement-Based Perspectivecited by 78
- 2020Adaptive Reward-Free Explorationcited by 77
- 2018On Oracle-Efficient PAC RL with Rich Observationscited by 71
- 2019Noisy Matrix Completion: Understanding Statistical Guarantees for Convex Relaxation via Nonconvex Optimizationcited by 58
- 2019Bilinear Bandits with Low-rank Structurecited by 55
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 52
- 2022Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approachcited by 51
- 2020Low-Rank Generalized Linear Bandit Problemscited by 49
- 2017Spectral Method and Regularized MLE Are Both Optimal for Top-K Rankingcited by 47
- 2016Rate-Optimal Perturbation Bounds for Singular Subspaces with Applications to High-Dimensional Statisticscited by 45
- 2015Fast spectral algorithms from sum-of-squares proofs: tensor decomposition and planted sparse vectorscited by 38
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 36
- 2022Learning in Observable POMDPs, without Computationally Intractable Oraclescited by 30
- 2017Stochastic Low-Rank Banditscited by 28
- 2019Estimating the Mixing Time of Ergodic Markov Chainscited by 25
- 2020Sample Efficient Reinforcement Learning via Low-Rank Matrix Estimationcited by 25
- 2016Stochastic Rank-1 Banditscited by 23
- 2019Solving Bernoulli Rank-One Bandits with Unimodal Thompson Samplingcited by 23
- 2018Spectral State Compression of Markov Processescited by 18
- 2017Unperturbed: spectral analysis beyond Davis-Kahancited by 16
- 2021Optimal Uniform OPE and Model-based Offline Reinforcement Learning in Time-Homogeneous, Reward-Free and Task-Agnostic Settingscited by 14
- 2019Learning Markov models via low-rank optimizationcited by 12
- 2021Entrywise Estimation of Singular Vectors of Low-Rank Matrices with Heteroskedasticity and Dependencecited by 12
- 2022Online Low Rank Matrix Completioncited by 9
- 2022Overcoming the Long Horizon Barrier for Sample-Efficient Reinforcement Learning with Latent Low-Rank Structurecited by 7
- 2019Low-rank matrix completion and denoising under Poisson noisecited by 6
- 2022Nearly Optimal Latent State Decoding in Block MDPscited by 5
- 2022Speed Up the Cold-Start Learning in Two-Sided Bandits with Many Armscited by 4
- 2018Estimation of Markov Chain via Rank-Constrained Likelihoodcited by 3
- 2023Representations and Exploration for Deep Reinforcement Learning using Singular Value Decompositioncited by 2
Built on by
Citing papers per year: 2024: 6; 2025: 1; 2026: 3.
10 papers.
- 2024From Self-Attention to Markov Models: Unveiling the Dynamics of Generative Transformerscited by 34
- 2024Low-Rank Bandits via Tight Two-to-Infinity Singular Subspace Recoverycited by 6
- 2024The Limits of Transfer Reinforcement Learning with Latent Low-rank Structurecited by 1
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimationcited by 1
- 2025Addressing Finite-Horizon MDPs via Low-Rank Tensor Value Approximationcited by 1
- 2026Catching a Moving Subspace: Low-Rank Bandits Beyond Stationaritycited by 1
- 2024Entrywise error bounds for low-rank approximations of kernel matricescited by 0
- 2024Online Matrix Completion: A Collaborative Approach with Hott Itemscited by 0
- 2026Factorized Spectral Representations for Reinforcement Learningcited by 0
- 2026Optimal Online Regret for Context-Lumpable Banditscited by 0
- 2026Optimal Online Regret for Context-Lumpable Bandits13 Oct 2026
- 2026Factorized Spectral Representations for Reinforcement Learning15 Jul 2026
- 2026Catching a Moving Subspace: Low-Rank Bandits Beyond Stationarity18 May 2026
- 2025Addressing Finite-Horizon MDPs via Low-Rank Tensor Value Approximation17 Jan 2025
- 2024Model-free Low-Rank Reinforcement Learning via Leveraged Entry-wise Matrix Estimation30 Oct 2024
- 2024The Limits of Transfer Reinforcement Learning with Latent Low-rank Structure28 Oct 2024
- 2024Online Matrix Completion: A Collaborative Approach with Hott Items11 Aug 2024
- 2024Entrywise error bounds for low-rank approximations of kernel matrices23 May 2024
- 2024Low-Rank Bandits via Tight Two-to-Infinity Singular Subspace Recovery24 Feb 2024
- 2024From Self-Attention to Markov Models: Unveiling the Dynamics of Generative Transformers21 Feb 2024