Instance-Dependent Auto-Exploration for Tabular Online Discounted Reinforcement Learning
Abstract
In online discounted reinforcement learning from a single trajectory, a central cost is exploration: the time spent reaching states whose optimal action is still unknown. Existing auto-exploration guarantees for stochastic policy mirror descent charge every iteration the cost of reaching the hardest state, even once its optimal action is known, through a constant exponential in the mixing time of the optimal policy. We prove that the exploration cost is controlled instead by a certified hitting cost , which is equivalent up to constants to the worst-case expected time to reach under policies that keep mass at least on optimal actions. For the algorithm's , every iterate is such a policy with high probability. Each rollout stops once it covers the states where the policy keeps mass at least on two or more actions. Resolved states leave this set, so every iteration is charged only the largest over unresolved states. For accuracy and advantage gaps , the sample complexity is then at most an baseline plus the exploration cost , up to logarithmic and instance-independent factors. The only structural assumption is that some deterministic optimal policy induces an irreducible chain; the mixing regime and action-independent transitions are special cases with explicit bounds on . On an action-dependent family of states whose small gaps arise from transitions, the upper bound matches a lower bound with deterministic costs up to logarithmic factors, for a fixed discount factor. Charging the hardest state at every iteration gives .
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.