English

When Does Non-Uniform Replay Matter in Reinforcement Learning?

Machine Learning 2026-05-19 v3 Artificial Intelligence

Abstract

Modern off-policy reinforcement learning algorithms often rely on simple uniform replay sampling and it remains unclear when and why non-uniform replay improves over this strong baseline. Across diverse RL settings, we show that the effectiveness of non-uniform replay is governed by three factors: replay volume, the number of replayed transitions per environment step; expected recency, how recent sampled transitions are; and the entropy of the replay sampling distribution. Our main contribution is clarifying when non-uniform replay is beneficial and providing practical guidance for replay design in modern off-policy RL. Namely, we find that non-uniform replay is most beneficial when replay volume is low, and that high-entropy sampling is important even at comparable expected recency. Motivated by these findings, we adopt a simple Truncated Geometric replay that biases sampling toward recent experience while preserving high entropy and incurring negligible computational overhead. Across large-scale parallel simulation, single-task, and multi-task settings, including three modern algorithms evaluated on five RL benchmark suites, this replay sampling strategy improves sample efficiency in low-volume regimes while remaining competitive when replay volume is high.

Keywords

Cite

@article{arxiv.2605.10236,
  title  = {When Does Non-Uniform Replay Matter in Reinforcement Learning?},
  author = {Michal Korniak and Mikołaj Czarnecki and Yarden As and Piotr Miłoś and Pieter Abbeel and Michal Nauman},
  journal= {arXiv preprint arXiv:2605.10236},
  year   = {2026}
}
R2 v1 2026-07-22T07:03:48.831Z