paper-with-me

홈 › Papers

PegasusFlow: Parallel Rolling-Denoising Score Sampling for Robot Diffusion Planner Flow Matching

2025-09-10 · Lei Ye, Haibo Gao, Peng Xu, Zhelin Zhang, Junqi Shan, Ao Zhang, Wei Zhang, Ruyi Zhou, Zongquan Deng, Liang Ding arxiv

Diffusion models offer powerful generative capabilities for robot trajectory planning, yet their practical deployment on robots is hindered by a critical bottleneck: a reliance on imitation learning from expert demonstrations. This paradigm is often impractical for specialized robots where data is scarce and creates an inefficient, theoretically suboptimal training pipeline. To overcome this, we introduce PegasusFlow, a hierarchical rolling-denoising framework that enables direct and parallel sampling of trajectory score gradients from environmental interaction, completely bypassing the need for expert data. Our core innovation is a novel sampling algorithm, Weighted Basis Function Optimization (WBFO), which leverages spline basis representations to achieve superior sample efficiency and faster convergence compared to traditional methods like MPPI. The framework is embedded within a scalable, asynchronous parallel simulation architecture that supports massively parallel rollouts for efficient data collection. Extensive experiments on trajectory optimization and robotic navigation tasks demonstrate that our approach, particularly Action-Value WBFO (AVWBFO) combined with a reinforcement learning warm-start, significantly outperforms baselines. In a challenging barrier-crossing task, our method achieved a 100% success rate and was 18% faster than the next-best method, validating its effectiveness for complex terrain locomotion planning. https://masteryip.github.io/pegasusflow.github.io/

📄 PDF Abstract BibTeX arXiv:2509.08435

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement LearningTrajectory Planning

Similar Papers 제목 키워드 기반

Parallel Sampling of Diffusion Models

2023-05-25 · NeurIPS 2023 11 · Andy Shih, Suneel Belkhale, Stefano Ermon, Dorsa Sadigh 외

Diffusion models are powerful generative models but suffer from slow sampling, often taking 1000 sequential denoising steps for one sample. As a result, considerable efforts have been directed toward reducing the number …

DenoisingImage Generation

TSGM: Regular and Irregular Time-series Generation using Score-based Generative Models

2025-11-26 · Haksoo Lim, Jaehoon Lee, Sewon Park, Minjung Kim 외 arxiv

Score-based generative models (SGMs) have demonstrated unparalleled sampling quality and diversity in numerous fields, such as image generation, voice synthesis, and tabular data synthesis, etc. Inspired by those outstan…

Image Generation

Faster Diffusion Sampling with Randomized Midpoints: Sequential and Parallel

2024-06-03 · Shivam Gupta, Linda Cai, Sitan Chen

Sampling algorithms play an important role in controlling the quality and runtime of diffusion model inference. In recent years, a number of works~\cite{chen2023sampling,chen2023ode,benton2023error,lee2022convergence} ha…

Efficient Diffusion Training through Parallelization with Truncated Karhunen-Loève Expansion

2025-03-22 · Yumeng Ren, Yaofang Liu, Aitor Artola, Laurent Mertz 외

Diffusion denoising models have become a popular approach for image generation, but they often suffer from slow convergence during training. In this paper, we identify that this slow convergence is partly due to the comp…

DenoisingImage Generation

Convergence of Deterministic and Stochastic Diffusion-Model Samplers: A Simple Analysis in Wasserstein Distance

2025-08-05 · Eliot Beyler, Francis Bach arxiv

We provide new convergence guarantees in Wasserstein distance for diffusion-based generative models, covering both stochastic (DDPM-like) and deterministic (DDIM-like) sampling methods. We introduce a simple framework to…