about
DiffuserLite: Towards Real-Time Diffusion Planning (arxiv.org)
2 points by PaulHoule on Feb 5, 2024 | hide | past | pdf | discuss on HN

In plain words: Instead of slowly drawing a detailed plan step by step, this system sketches a rough path first, then fills in fine details, cutting wasted work. It decides 122 times a second, far faster than leading planners, while matching their best results.

Abstract · DiffuserLite: Towards Real-time Diffusion Planning

Diffusion planning has been recognized as an effective decision-making paradigm in various domains. The capability of generating high-quality long-horizon trajectories makes it a promising research direction. However, existing diffusion planning methods suffer from low decision-making frequencies due to the expensive iterative sampling cost. To alleviate this, we introduce DiffuserLite, a super fast and lightweight diffusion planning framework, which employs a planning refinement process (PRP) to generate coarse-to-fine-grained trajectories, significantly reducing the modeling of redundant information and leading to notable increases in decision-making frequency. Our experimental results demonstrate that DiffuserLite achieves a decision-making frequency of 122.2Hz (112.7x faster than predominant frameworks) and reaches state-of-the-art performance on D4RL, Robomimic, and FinRL benchmarks. In addition, DiffuserLite can also serve as a flexible plugin to increase the decision-making frequency of other diffusion planning algorithms, providing a structural design reference for future works. More details and visualizations are available at https://diffuserlite.github.io/.

Zibin Dong, Jianye Hao, Yifu Yuan, Fei Ni, Yitian Wang, Pengyi Li, Yan Zheng
arXiv:2401.15443 · cs.AI · submitted Jan 27, 2024 · updated Oct 25, 2024
abstract · pdf · html

add comment on HN