English

D3P: Dynamic Denoising Diffusion Policy via Reinforcement Learning

Robotics 2025-08-12 v1

Abstract

Diffusion policies excel at learning complex action distributions for robotic visuomotor tasks, yet their iterative denoising process poses a major bottleneck for real-time deployment. Existing acceleration methods apply a fixed number of denoising steps per action, implicitly treating all actions as equally important. However, our experiments reveal that robotic tasks often contain a mix of \emph{crucial} and \emph{routine} actions, which differ in their impact on task success. Motivated by this finding, we propose \textbf{D}ynamic \textbf{D}enoising \textbf{D}iffusion \textbf{P}olicy \textbf{(D3P)}, a diffusion-based policy that adaptively allocates denoising steps across actions at test time. D3P uses a lightweight, state-aware adaptor to allocate the optimal number of denoising steps for each action. We jointly optimize the adaptor and base diffusion policy via reinforcement learning to balance task performance and inference efficiency. On simulated tasks, D3P achieves an averaged 2.2×\times inference speed-up over baselines without degrading success. Furthermore, we demonstrate D3P's effectiveness on a physical robot, achieving a 1.9×\times acceleration over the baseline.

Keywords

Cite

@article{arxiv.2508.06804,
  title  = {D3P: Dynamic Denoising Diffusion Policy via Reinforcement Learning},
  author = {Shu-Ang Yu and Feng Gao and Yi Wu and Chao Yu and Yu Wang},
  journal= {arXiv preprint arXiv:2508.06804},
  year   = {2025}
}
R2 v1 2026-07-01T04:42:11.214Z