stat.MLOct 3, 2026

Exact Fast Batch Simulation for Tabular Reinforcement Learning

Authors: Haochen Zhang, Lingzhou Xue, Zhong Zheng

Organizations: Department of Statistics, The Pennsylvania State University · Department of Applied and Computational Mathematics and Statistics, University of Notre Dame

Abstract

Simulation is a fundamental computational primitive in reinforcement learning (RL), yet conventional simulation explicitly generates individual trajectories even when downstream procedures use only aggregate statistics. To address this, we develop an exact fast-simulation framework for finite-horizon tabular Markov decision processes. Our framework has two complementary modes. In direct batch simulation, a batch is represented by its aggregate Markov flow. With sufficient parallel simulation resources, this flow can be obtained by trajectory aggregation; when such simulation is unavailable or costly but the initial state and transition distributions are directly accessible, we instead generate an identically distributed flow through forward Markov-flow sampling without materializing individual trajectories. The latter reduces the simulator-side computational dependence on batch size mm from O(m)O(m) to O(1)O(1). In adaptive batch simulation, when batch length is determined by a data-dependent condition, exact multivariate-hypergeometric splitting recursively refines a candidate Markov flow while preserving the conditional law, reducing the cost dependence on mm from O(m)O(m) to O(log⁡m)O(\log m). Together, these modes accelerate simulation by keeping trajectories aggregated whenever possible and refining flows only when required to locate data-dependent boundaries. The framework applies broadly across simulator-based, offline, and online batch or stage-based RL, as illustrated with representative algorithms from each setting.

Explore similar work

CardsList
  1. Accelerating Reinforcement Learning Training Using Simulation Surrogate Models

    May 26, 2026Mohammadmahdi Ghasemloo, David J. Eckman, Yaxian LiSurrogate ModelingReinforcement Learning

  2. Hybrid-Adaptive Thread Tuning to Mitigate Simulation Execution Bottlenecks in High-Performance Reinforcement Learning Inference

    Aug 6, 2026Jiming Su, Hantao Hua, Lujia Yin +2