Online RL in Linearly qπ-Realizable MDPs Is as Easy as in Linear MDPs If You Learn What to Ignore
Gellért Weisz, András György, Csaba Szepesvári
Abstract
We consider online reinforcement learning (RL) in episodic Markov decision processes (MDPs) under the linear -realizability assumption, where it is assumed that the action-values of all policies can be expressed as linear functions of stateaction features. This class is known to be more general than linear MDPs, where the transition kernel and the reward function are assumed to be linear functions of the feature vectors. As our first contribution, we show that the difference between the two classes is the presence of states in linearly -realizable MDPs where for any policy, all the actions have approximately equal values, and skipping over these states by following an arbitrarily fixed policy in those states transforms the problem to a linear MDP. Based on this observation, we derive a novel (computationally inefficient) learning algorithm for linearly -realizable MDPs that simultaneously learns what states should be skipped over and runs another learning algorithm on the linear MDP hidden in the problem. The method returns an -optimal policy after polylog( , )/ 2 interactions with the MDP, where is the time horizon and is the dimension of the feature vectors, giving the first polynomial-sample-complexity online RL algorithm for this setting. The results are proved for the misspecified case, where the sample complexity is shown to degrade gracefully with the misspecification error. any ∈ [A], P , , is the distribution over the histories when first action is used in state , after which policy is followed. E• is the expectation operator corresponding to a distribution P• (e.g., E , is the expectation with respect to P , ). The state-and action-value functions and are defined as the expected total reward within the first episode while is used: Let ★ ∈ Π be an optimal policy, satisfying ★ ( , ) = sup ∈Π ( , ) = sup ∈all policies ( , ) for all ( , ) ∈ S × [A]. Let ★ ( , ) = ★ ( , ) and ★ ( ) = sup ′ ∈ [ A ] ★ ( , ) for all ( , ).
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.
Cited by top-tier papers6
- Provable and Practical: Efficient Exploration in Reinforcement Learning via Langevin Monte CarloHaque Ishfaq, Qingfeng Lan, Pan Xu, A. Rupam Mahmood et al.ICLR 2024 · 33 citations
- The Power of Resets in Online Reinforcement LearningZakaria Mhammedi, Dylan J. Foster, Alexander RakhlinNeurIPS 2024 · 15 citations
- Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPsAntoine Moulin, Gergely Neu, Luca VianoNeurIPS 2025 · 6 citations
- Confident Natural Policy Gradient for Local Planning in qπ-realizable Constrained MDPsTian Tian, Lin Yang, Csaba SzepesváriNeurIPS 2024 · 6 citations
- Eluder dimension: localise it!Alireza Bakhtiari, Alex Ayoub, Samuel Robertson, David Janz et al.NeurIPS 2025 · 3 citations
Builds on5
- Is Pessimism Provably Efficient for Offline RL?Ying Jin, Zhuoran Yang, Zhaoran WangICML 2021 · 419 citations
- Learning Near Optimal Policies with Low Inherent Bellman ErrorAndrea Zanette, Alessandro Lazaric, Mykel J. Kochenderfer, Emma BrunskillICML 2020 · 238 citations
- Is a Good Representation Sufficient for Sample Efficient Reinforcement Learning?Simon S. Du, Sham M. Kakade, Ruosong Wang, Lin F. YangICLR 2020 · 213 citations
- Learning with Good Feature Representations in Bandits and in RL with a Generative ModelTor Lattimore, Csaba Szepesvári, Gellért WeiszICML 2020 · 181 citations
- Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision ProcessesAndrew J. Wagenmaker, Yifang Chen, Max Simchowitz, Simon S. Du et al.ICML 2022 · 61 citations
Related papers
- Frozen Policy Iteration: Computationally Efficient RL under Linear Qπ Realizability for Deterministic DynamicsYijing Ke, Zihan Zhang, Ruosong WangICLR 2026
- Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision ProcessesJiafan He, Heyang Zhao, Dongruo Zhou, Quanquan GuICML 2023 · 68 citations
- Trajectory Data Suffices for Statistically Efficient Learning in Offline RL with Linear qπ-Realizability and ConcentrabilityVolodymyr Tkachuk, Gellért Weisz, Csaba SzepesváriNeurIPS 2024 · 3 citations
- A Primal-Dual Algorithm for Offline Constrained Reinforcement Learning with Linear MDPsKihyuk Hong, Ambuj TewariICML 2024 · 5 citations
- Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPsJunkai Zhang, Weitong Zhang, Quanquan GuICML 2023 · 6 citations
