English

Q-Star Meets Scalable Posterior Sampling: Bridging Theory and Practice via HyperAgent

Machine Learning 2024-06-17 v5 Artificial Intelligence Machine Learning

Abstract

We propose HyperAgent, a reinforcement learning (RL) algorithm based on the hypermodel framework for exploration in RL. HyperAgent allows for the efficient incremental approximation of posteriors associated with an optimal action-value function (QQ^\star) without the need for conjugacy and follows the greedy policies w.r.t. these approximate posterior samples. We demonstrate that HyperAgent offers robust performance in large-scale deep RL benchmarks. It can solve Deep Sea hard exploration problems with episodes that optimally scale with problem size and exhibits significant efficiency gains in the Atari suite. Implementing HyperAgent requires minimal code addition to well-established deep RL frameworks like DQN. We theoretically prove that, under tabular assumptions, HyperAgent achieves logarithmic per-step computational complexity while attaining sublinear regret, matching the best known randomized tabular RL algorithm.

Keywords

Cite

@article{arxiv.2402.10228,
  title  = {Q-Star Meets Scalable Posterior Sampling: Bridging Theory and Practice via HyperAgent},
  author = {Yingru Li and Jiawei Xu and Lei Han and Zhi-Quan Luo},
  journal= {arXiv preprint arXiv:2402.10228},
  year   = {2024}
}

Comments

Proceedings of the $\mathit{41}^{st}$ International Conference on Machine Learning, Vienna, Austria. PMLR 235, 2024. Copyright 2024 by the author(s). Invited talk in Informs Optimization Conference 2024 and International Symposium on Mathematical Programming 2024

R2 v1 2026-06-28T14:50:01.000Z