English

Diffusion Modulation via Environment Mechanism Modeling for Planning

Artificial Intelligence 2026-02-25 v1 Machine Learning

Abstract

Diffusion models have shown promising capabilities in trajectory generation for planning in offline reinforcement learning (RL). However, conventional diffusion-based planning methods often fail to account for the fact that generating trajectories in RL requires unique consistency between transitions to ensure coherence in real environments. This oversight can result in considerable discrepancies between the generated trajectories and the underlying mechanisms of a real environment. To address this problem, we propose a novel diffusion-based planning method, termed as Diffusion Modulation via Environment Mechanism Modeling (DMEMM). DMEMM modulates diffusion model training by incorporating key RL environment mechanisms, particularly transition dynamics and reward functions. Experimental results demonstrate that DMEMM achieves state-of-the-art performance for planning with offline reinforcement learning.

Keywords

Cite

@article{arxiv.2602.20422,
  title  = {Diffusion Modulation via Environment Mechanism Modeling for Planning},
  author = {Hanping Zhang and Yuhong Guo},
  journal= {arXiv preprint arXiv:2602.20422},
  year   = {2026}
}
R2 v1 2026-07-01T10:48:58.520Z