让扩散模型轨迹规划快100倍,还能保持甚至提升效果。
Fast Monte Carlo Tree Diffusion: 100x Speedup via Parallel Sparse Planning
- 用并行化和稀疏规划减少重复计算,加速树搜索过程。
- 实验显示速度提升最高达100倍,且在部分任务上比Diffuser还快。
- 适合需要高效长时序推理的机器人路径规划场景。
扩散模型近年来成为轨迹规划的强大工具,但其固有的非序列特性限制了其在长时程推理任务中的表现。最近提出的蒙特卡洛树扩散(MCTD)通过结合扩散模型与基于树的搜索,在复杂规划问题上取得了顶尖性能。然而,我们的分析表明,MCTD因树搜索的顺序性及迭代去噪的开销,导致显著计算成本。为此,我们提出Fast-MCTD,一种更高效的变体,在保留MCTD优势的同时大幅提升速度与可扩展性。Fast-MCTD融合两项技术:并行MCTD,通过延迟树更新和冗余感知选择实现并行采样;稀疏MCTD,通过轨迹粗化减少采样长度。实验显示,Fast-MCTD相比标准MCTD最高提速100倍,同时保持或超越原有性能。尤为突出的是,其在某些任务上的推理速度甚至超过无需搜索的Diffuser,尽管后者解的质量较低。这些结果使Fast-MCTD成为基于扩散模型的推理时推理任务中实用且可扩展的解决方案。
原文摘要 · Abstract (English)
Diffusion models have recently emerged as a powerful approach for trajectory planning. However, their inherently non-sequential nature limits their effectiveness in long-horizon reasoning tasks at test time. The recently proposed Monte Carlo Tree Diffusion (MCTD) offers a promising solution by combining diffusion with tree-based search, achieving state-of-the-art performance on complex planning problems. Despite its strengths, our analysis shows that MCTD incurs substantial computational overhead due to the sequential nature of tree search and the cost of iterative denoising. To address this, we propose Fast-MCTD, a more efficient variant that preserves the strengths of MCTD while significantly improving its speed and scalability. Fast-MCTD integrates two techniques: Parallel MCTD, which enables parallel rollouts via delayed tree updates and redundancy-aware selection; and Sparse MCTD, which reduces rollout length through trajectory coarsening. Experiments show that Fast-MCTD achieves up to 100x speedup over standard MCTD while maintaining or improving planning performance. Remarkably, it even outperforms Diffuser in inference speed on some tasks, despite Diffuser requiring no search and yielding weaker solutions. These results position Fast-MCTD as a practical and scalable solution for diffusion-based inference-time reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。