cs.LGMay 11, 2026

When Does Non-Uniform Replay Matter in Reinforcement Learning?

Authors: Michal KorniakMikołaj CzarneckiYarden AsPiotr MiłośPieter AbbeelMichal Nauman

Organizations: 1ETH Zurich · University of Warsaw · 3UC Berkeley · 4Amazon FAR

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.

Explore similar work

CardsList
  1. Uncertainty-Driven Replay Memory for Reinforcement Learning

    Aug 30, 2026Sheeraja Rajakrishnan, Alexander G. Ororbia, Travis Desell +1Transitions