扩散策略其实靠记忆动作查表,简单替代方案更快更省资源。
Demystifying Diffusion Policies: Action Memorization and Simple Lookup Table Alternatives
- 用隐空间相似度查找训练动作序列,实现无需泛化的快速响应。
- 在小数据下性能相当,推理速度提升294倍,内存减少99.15%。
- 可识别分布外输入,适合资源受限的实时机器人系统。
扩散策略在高维机器人操作任务中表现出色,且仅需少量示范即可训练。本文提出一个令人意外的假设:扩散策略本质上是记忆了一个动作查找表,这反而带来了优势。运行时,它在隐空间中寻找与测试图像最接近的训练图像,并回溯对应的训练动作序列,从而实现无需动作泛化的快速反应。该机制在数据稀疏场景下尤为有效。我们通过实验证明,即使输入猫狗等分布外图像,扩散策略仍输出训练数据中的动作序列。基于此,我们提出轻量级的动作查找表(ALT)策略,使用对比图像编码器作为哈希函数,显式索引最近的动作序列。实验表明,在小数据集上ALT性能媲美扩散模型,推理时间仅为0.0034倍,内存占用仅0.0085倍,显著提升闭环推理速度,适用于资源受限机器人。此外,当运行时图像与训练图像距离过远时,ALT能明确发出分布外警告,提供简单有效的运行时监控。更多信息见:https://stanfordmsl.github.io/alt/
原文摘要 · Abstract (English)
Diffusion policies have demonstrated remarkable dexterity and robustness in intricate, high-dimensional robot manipulation tasks, while training from a small number of demonstrations. However, the reason for this performance remains a mystery. In this paper, we offer a surprising hypothesis: diffusion policies essentially memorize an action lookup table -- and this is beneficial. We posit that, at runtime, diffusion policies find the closest training image to the test image in a latent space, and recall the associated training action sequence, offering reactivity without the need for action generalization. This is effective in the sparse data regime, where there is not enough data density for the model to learn action generalization. We support this claim with systematic empirical evidence. Even when conditioned on wildly out of distribution (OOD) images of cats and dogs, the Diffusion Policy still outputs an action sequence from the training data. With this insight, we propose a simple policy, the Action Lookup Table (ALT), as a lightweight alternative to the Diffusion Policy. Our ALT policy uses a contrastive image encoder as a hash function to index the closest corresponding training action sequence, explicitly performing the computation that the Diffusion Policy implicitly learns. We show empirically that for relatively small datasets, ALT matches the performance of a diffusion model, while requiring only 0.0034 of the inference time and 0.0085 of the memory footprint, allowing for much faster closed-loop inference with resource constrained robots. We also train our ALT policy to give an explicit OOD flag when the distance between the runtime image is too far in the latent space from the training images, giving a simple but effective runtime monitor. More information can be found at: https://stanfordmsl.github.io/alt/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。