SwiftVLA: Unlocking Spatiotemporal Dynamics for Lightweight VLA Models at Minimal Overhead

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ni, Chaojun, Chen, Cheng, Wang, Xiaofeng, Zhu, Zheng, Zheng, Wenzhao, Wang, Boyuan, Chen, Tianrun, Zhao, Guosheng, Li, Haoyun, Dong, Zhehao, Zhang, Qiang, Ye, Yun, Wang, Yang, Huang, Guan, Mei, Wenjun
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909934902312960
author Ni, Chaojun
Chen, Cheng
Wang, Xiaofeng
Zhu, Zheng
Zheng, Wenzhao
Wang, Boyuan
Chen, Tianrun
Zhao, Guosheng
Li, Haoyun
Dong, Zhehao
Zhang, Qiang
Ye, Yun
Wang, Yang
Huang, Guan
Mei, Wenjun
author_facet Ni, Chaojun
Chen, Cheng
Wang, Xiaofeng
Zhu, Zheng
Zheng, Wenzhao
Wang, Boyuan
Chen, Tianrun
Zhao, Guosheng
Li, Haoyun
Dong, Zhehao
Zhang, Qiang
Ye, Yun
Wang, Yang
Huang, Guan
Mei, Wenjun
contents 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 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 masks 4D inputs to the VLM and trains the VLA to reconstruct them, enabling the VLM to learn effective 4D representations and allowing the 4D branch to be dropped at inference with minimal performance loss. Experiments in real and simulated environments show that SwiftVLA outperforms lightweight baselines and rivals VLAs up to 7 times larger, achieving comparable performance on edge devices while being 18 times faster and reducing memory footprint by 12 times.
format Preprint
id arxiv_https___arxiv_org_abs_2512_00903
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SwiftVLA: Unlocking Spatiotemporal Dynamics for Lightweight VLA Models at Minimal Overhead
Ni, Chaojun
Chen, Cheng
Wang, Xiaofeng
Zhu, Zheng
Zheng, Wenzhao
Wang, Boyuan
Chen, Tianrun
Zhao, Guosheng
Li, Haoyun
Dong, Zhehao
Zhang, Qiang
Ye, Yun
Wang, Yang
Huang, Guan
Mei, Wenjun
Computer Vision and Pattern Recognition
Robotics
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 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 masks 4D inputs to the VLM and trains the VLA to reconstruct them, enabling the VLM to learn effective 4D representations and allowing the 4D branch to be dropped at inference with minimal performance loss. Experiments in real and simulated environments show that SwiftVLA outperforms lightweight baselines and rivals VLAs up to 7 times larger, achieving comparable performance on edge devices while being 18 times faster and reducing memory footprint by 12 times.
title SwiftVLA: Unlocking Spatiotemporal Dynamics for Lightweight VLA Models at Minimal Overhead
topic Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2512.00903