English

State-Action Inpainting Diffuser for Continuous Control with Delay

Artificial Intelligence 2026-03-03 v1 Machine Learning

Abstract

Signal delay poses a fundamental challenge in continuous control and reinforcement learning (RL) by introducing a temporal gap between interaction and perception. Current solutions have largely evolved along two distinct paradigms: model-free approaches which utilize state augmentation to preserve Markovian properties, and model-based methods which focus on inferring latent beliefs via dynamics modeling. In this paper, we bridge these perspectives by introducing State-Action Inpainting Diffuser (SAID), a framework that integrates the inductive bias of dynamics learning with the direct decision-making capability of policy optimization. By formulating the problem as a joint sequence inpainting task, SAID implicitly captures environmental dynamics while directly generating consistent plans, effectively operating at the intersection of model-based and model-free paradigms. Crucially, this generative formulation allows SAID to be seamlessly applied to both online and offline RL. Extensive experiments on delayed continuous control benchmarks demonstrate that SAID achieves state-of-the-art and robust performance. Our study suggests a new methodology to advance the field of RL with delay.

Keywords

Cite

@article{arxiv.2603.01553,
  title  = {State-Action Inpainting Diffuser for Continuous Control with Delay},
  author = {Dongqi Han and Wei Wang and Enze Zhang and Dongsheng Li},
  journal= {arXiv preprint arXiv:2603.01553},
  year   = {2026}
}
R2 v1 2026-07-01T10:58:40.853Z