English

Belief-Based Offline Reinforcement Learning for Delay-Robust Policy Optimization

Machine Learning 2026-02-12 v2 Artificial Intelligence

Abstract

Offline-to-online deployment of reinforcement-learning (RL) agents must bridge two gaps: (1) the sim-to-real gap, where real systems add latency and other imperfections not present in simulation, and (2) the interaction gap, where policies trained purely offline face out-of-distribution states during online execution because gathering new interaction data is costly or risky. Agents therefore have to generalize from static, delay-free datasets to dynamic, delay-prone environments. Standard offline RL learns from delay-free logs yet must act under delays that break the Markov assumption and hurt performance. We introduce DT-CORL (Delay-Transformer belief policy Constrained Offline RL), an offline-RL framework built to cope with delayed dynamics at deployment. DT-CORL (i) produces delay-robust actions with a transformer-based belief predictor even though it never sees delayed observations during training, and (ii) is markedly more sample-efficient than na\"ive history-augmentation baselines. Experiments on D4RL benchmarks with several delay settings show that DT-CORL consistently outperforms both history-augmentation and vanilla belief-based methods, narrowing the sim-to-real latency gap while preserving data efficiency.

Keywords

Cite

@article{arxiv.2506.00131,
  title  = {Belief-Based Offline Reinforcement Learning for Delay-Robust Policy Optimization},
  author = {Simon Sinong Zhan and Qingyuan Wu and Philip Wang and Frank Yang and Xiangyu Shi and Chao Huang and Qi Zhu},
  journal= {arXiv preprint arXiv:2506.00131},
  year   = {2026}
}
R2 v1 2026-07-01T02:51:33.270Z