VAT: Vision Action Transformer by Unlocking Full Representation of ViT

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Wenhao, Ma, Chengwei, Mao, Weixin
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911408207167488
author Li, Wenhao
Ma, Chengwei
Mao, Weixin
author_facet Li, Wenhao
Ma, Chengwei
Mao, Weixin
contents In robot learning, Vision Transformers (ViTs) are standard for visual perception, yet most methods discard valuable information by using only the final layer's features. We argue this provides an insufficient representation and propose the Vision Action Transformer (VAT), a novel architecture that is extended from ViT and unlocks the full feature hierarchy of ViT. VAT processes specialized action tokens with visual features across all transformer layers, enabling a deep and progressive fusion of perception and action generation. On a suite of simulated manipulation tasks, VAT achieves a 98.15\% average success rate across four LIBERO benchmarks, establishing a new state-of-the-art by outperforming prior methods like OpenVLA-OFT. Our work presents not only a powerful model for imitation learning but also demonstrates the critical importance of leveraging the complete ''representation trajectory'' of vision models to advance robotic policy. The GitHub URL for the project code is https://github.com/sellerbubble/VAT.
format Preprint
id arxiv_https___arxiv_org_abs_2512_06013
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle VAT: Vision Action Transformer by Unlocking Full Representation of ViT
Li, Wenhao
Ma, Chengwei
Mao, Weixin
Computer Vision and Pattern Recognition
Robotics
In robot learning, Vision Transformers (ViTs) are standard for visual perception, yet most methods discard valuable information by using only the final layer's features. We argue this provides an insufficient representation and propose the Vision Action Transformer (VAT), a novel architecture that is extended from ViT and unlocks the full feature hierarchy of ViT. VAT processes specialized action tokens with visual features across all transformer layers, enabling a deep and progressive fusion of perception and action generation. On a suite of simulated manipulation tasks, VAT achieves a 98.15\% average success rate across four LIBERO benchmarks, establishing a new state-of-the-art by outperforming prior methods like OpenVLA-OFT. Our work presents not only a powerful model for imitation learning but also demonstrates the critical importance of leveraging the complete ''representation trajectory'' of vision models to advance robotic policy. The GitHub URL for the project code is https://github.com/sellerbubble/VAT.
title VAT: Vision Action Transformer by Unlocking Full Representation of ViT
topic Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2512.06013