跳到正文
arXiv · cs.RO· Guowen Zhang·· 22 小时前精选AI 评分70

OSWAM: 一种单步行动生成的有效世界行动模型

OSWAM: An Effective World Action Model with One-Step Action Generation

AI 导读

Guowen Zhang 提出OSWAM模型,通过构建想象令牌初始化扩散过程和引入特权信息蒸馏,实现单步高质量行动生成。该模型在自动驾驶和具身控制任务中表现优于需数十步去噪的基线模型,在NVIDIA RTX 5090 GPU上延迟低至40ms,推理速度提升约3倍。

推荐理由

OSWAM 将扩散模型去噪步数从数十步降至一步,在保持行动质量的同时实现40ms低延迟推理。

来源:arXiv · cs.RO · arxiv.org