arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.13740cs.ROcs.LG

SmolWAM:面向边缘设备机器人操作的潜在世界-动作模型

SmolWAM: A Latent World-Action Model for Robotic Manipulation on Edge Devices

ML Collective

Timi Owolabi, Hikmah Olawore, Frances Adelakun, Ayotomiwa Oyewumi

AI总结:

SmolWAM是从Cosmos Policy衍生的轻量潜在世界-动作模型,适配边缘设备,在LIBERO任务中实现95.5%的成功率,速度、运算量和内存占用均显著优于原模型。

摘要:

世界-动作模型(WAMs)可预测机器人的动作以及这些动作将产生的观测结果。然而,性能最优的模型依赖于视频扩散变换器,其计算成本对于资源受限的边缘设备而言过高。我们提出SmolWAM,这是一种潜在WAM,由参数规模为2B的Cosmos Policy衍生而来,保留其预训练主干并重构推理接口:单帧视觉潜变量、保留主干注意力汇的缓存文本特征、28个变换器块中的12个,以及从1008个token中通过一次整流流步骤读出动作和未来视觉潜变量。SmolWAM的训练耗时3.6 GPU小时,在400个LIBERO任务中成功率达95.5%,每16个动作块耗时57毫秒,浮点运算量为2.3 TFLOP,GPU内存占用2.4 GiB;相比其教师模型,速度提升11倍,浮点运算量减少18.5倍,内存减少3.2倍,且其世界头仍能预测场景运动。

英文摘要:

World-action models (WAMs) predict a robot's actions together with the observations those actions will produce. However, the strongest models rely on video diffusion transformers that are far too computationally expensive for resource-constrained edge devices. We present SmolWAM, a latent WAM that we derive from the 2B-parameter Cosmos Policy by keeping its pretrained trunk and rebuilding its inference interface: single-frame visual latents, cached text features that preserve the trunk's attention sinks, 12 of 28 transformer blocks, and 1,008 tokens from which actions and future visual latents are read out in one rectified-flow step. Trained in 3.6 GPU-hours, SmolWAM succeeds in 95.5% of 400 LIBERO episodes at 57 ms and 2.3 TFLOP per 16-action chunk with 2.4 GiB of GPU memory, 11x faster than its teacher with 18.5x fewer FLOPs and 3.2x less memory, and its world head still forecasts scene motion.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

项目信息

相关深度报道

↑