Hierarchical Denoising For Multi-Step Visual Reasoning· 层次去噪用于多步视觉推理
Video models are evolving into vision foundation models, yet they still lack human-like multi-step reasoning. Streaming autoregressive diffusion models are efficient but limited in reasoning, while bidirectional diffusion enables global revision with high inference costs due to dense frame-level denoising. Both paradigms struggle to achieve logical consistency and low-latency streaming for complex reasoning tasks. We propose HDR (Hierarchical Denoising for Visual Reasoning), a unified framework that integrates hierarchical latents into causal video generation for multi-step reasoning. HDR organizes video latents into a tree-structured hierarchy, enabling coarse-to-fine reasoning before streaming output. Coarse denoising layers preserve uncertain hypotheses for global planning, while finer layers progressively refine them into concrete visual states. A sparse hierarchical attention pattern (SHAP) further reduces temporal attention costs. We introduce a level-stratified multi-step video reasoning benchmark with out-of-distribution cases, covering six tasks: maze navigation, Tower of Hanoi, one-line drawing, sliding puzzle, Sokoban, and water pouring. Compared with streaming autoregressive diffusion baselines, HDR improves success from 34.22 to 60.29 (76.2% relative gain) and increases average progress from 76.00 to 89.56, demonstrating more consistent reasoning trajectories. HDR maintains low-latency streaming at 0.70 seconds per latent, achieving 54.2 times faster inference than bidirectional diffusion. It also retains 82.9% of full-data performance with only 2% training data, compared with 52.0% for bidirectional diffusion. Real-world robot experiments further demonstrate HDR's potential for physical interaction and world modeling. Project demo: https://hierarchical-diffusion-reasoning.github.io/.
层次去噪模型实现高效多步视觉推理。
- 核心方法
- HDR框架通过树结构层次潜变量实现从粗到细的多步推理,同时利用稀疏层次注意力模式降低计算成本。
- 适合谁读
- 研究者 / 工程师
- 要解决的问题
- 现有的视频模型在多步视觉推理任务中缺乏逻辑一致性和低延迟,尤其是在复杂任务上。
- 关键实验
- 引入了六项多步视频推理任务的基准测试,并展示了实际机器人实验结果。
- 主要贡献
- 显著提高了多步视觉推理任务的成功率和推理速度,同时减少了所需的训练数据量。
- 意义与局限
- 提供了更高效的多步推理方法,适用于复杂视觉任务,显著降低了计算资源需求,但其在不同分布场景下的表现仍需进一步验证。