Processing math: 100%

2507.10347

Total: 1

#1 Parallel Sampling of Diffusion Models on SO(3) [PDF] [Copy] [Kimi1] [REL]

Authors: Yan-Ting Chen, Hao-Wei Chen, Tsu-Ching Hsiao, Chun-Yi Lee

In this paper, we design an algorithm to accelerate the diffusion process on the SO(3) manifold. The inherently sequential nature of diffusion models necessitates substantial time for denoising perturbed data. To overcome this limitation, we proposed to adapt the numerical Picard iteration for the SO(3) space. We demonstrate our algorithm on an existing method that employs diffusion models to address the pose ambiguity problem. Moreover, we show that this acceleration advantage occurs without any measurable degradation in task reward. The experiments reveal that our algorithm achieves a speed-up of up to 4.9×, significantly reducing the latency for generating a single sample.

Subject: Machine Learning

Publish: 2025-07-14 14:51:02 UTC