SwiftVLA:以极小开销为轻量 VLA 模型解锁时空动态
SwiftVLA: Unlocking Spatiotemporal Dynamics for Lightweight VLA Models at Minimal Overhead
用预训练的 4D 视觉几何 Transformer 加时序缓存,从 2D 图像流增量提取 4D 特征,再通过可学习的 Fusion Tokens 与未来预测目标把 4D 特征注入轻量 VLM;掩码重建策略让 4D 分支在推理时可丢弃,为视觉侧提供了一条「训练时用几何先验、部署时零开销」的可复用范式。
原文依据 a pretrained 4D visual geometry transformer with a temporal cache that incrementally extracts 4D features from 2D images
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论基于预训练视觉语言模型(VLM)构建的视觉-语言-动作(VLA)模型展现出强大潜力,但庞大的参数量限制了其实用性。为缓解这一问题,已有工作尝试使用轻量 VLM,但这会损害时空推理能力。尽管部分方法表明引入额外的 3D 输入有所帮助,但它们通常依赖大型 VLM 来融合 3D 与 2D 输入,且仍缺乏时序理解。为此,我们提出 SwiftVLA,一种在保持设计高效性的同时为紧凑模型增强 4D 理解的架构。具体而言,我们的方法采用一个预训练的 4D 视觉几何 Transformer,并配备时序缓存,从 2D 图像中增量式提取 4D 特征。随后,为增强 VLM 同时利用 2D 图像与 4D 特征的能力,我们引入 Fusion Tokens——一组以未来预测为目标训练的可学习 token,用于生成面向动作生成的统一表征。最后,我们提出掩码-重建策略,随机掩码输入 VLM 的 4D 输入,并训练 VLA 重建被掩码的特征。这一自重建目标有助于学习有效的 4D 表征,使得 4D 分支可在推理时被丢弃而仅带来极小的性能损失。在真实与仿真环境中的大量实验表明,SwiftVLA 优于轻量基线,并可与规模大至 7 倍的 VLA 模型相媲美。在边缘设备上,SwiftVLA 在取得相当性能的同时,比 π_0 快 18 倍,并将显存占用降低 12 倍。
英文原文摘要 1579 字符
Vision-Language-Action (VLA) models built on pretrained Vision-Language Models (VLMs) show strong potential but are limited in practicality due to their large parameter counts. To mitigate this issue, using a lightweight VLM has been explored, but it compromises spatiotemporal reasoning. Although some methods suggest that incorporating additional 3D inputs can help, they usually rely on large VLMs to fuse 3D and 2D inputs and still lack temporal understanding. Therefore, we propose SwiftVLA, an architecture that enhances a compact model with 4D understanding while preserving design efficiency. Specifically, our approach features a pretrained 4D visual geometry transformer with a temporal cache that incrementally extracts 4D features from 2D images. Then, to enhance the VLM's ability to exploit both 2D images and 4D features, we introduce Fusion Tokens, a set of learnable tokens trained with a future prediction objective to generate unified representations for action generation. Finally, we introduce a mask-and-reconstruct strategy that randomly masks 4D inputs to the VLM and trains the VLA to reconstruct the masked features. This self-reconstruction objective helps learn effective 4D representations, allowing the 4D branch to be dropped at inference with minimal performance loss. Extensive experiments in real and simulated environments show that SwiftVLA outperforms lightweight baselines and rivals VLAs up to 7xlarger. On edge devices, SwiftVLA achieves comparable performance while being 18xfaster than the \pi_0 and reducing the memory footprint by 12x.