HeiSD: Hybrid Speculative Decoding for Embodied Vision-Language-Action Models with Kinematic Awareness

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zheng, Zihao, Mao, Zhihao, Tian, Sicheng, Li, Maoliang, Chen, Jiayu, Sun, Xinhao, Zhang, Zhaobo, Liu, Xuanzhe, Cao, Donggang, Mei, Hong, Chen, Xiang
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913063774453760
author Zheng, Zihao
Mao, Zhihao
Tian, Sicheng
Li, Maoliang
Chen, Jiayu
Sun, Xinhao
Zhang, Zhaobo
Liu, Xuanzhe
Cao, Donggang
Mei, Hong
Chen, Xiang
author_facet Zheng, Zihao
Mao, Zhihao
Tian, Sicheng
Li, Maoliang
Chen, Jiayu
Sun, Xinhao
Zhang, Zhaobo
Liu, Xuanzhe
Cao, Donggang
Mei, Hong
Chen, Xiang
contents Vision-Language-Action (VLA) Models have become the mainstream solution for robot control, but suffer from slow inference speeds. Speculative Decoding (SD) is a promising acceleration method which can be divided into two categories: drafter-based SD and retrieval-based SD. Each of the two methods demonstrates complementary advantages and limitations when applied to VLA models, leading to the hypothesis that a hybrid approach integrating these two methods will yield better performance. In this paper, we first conduct a series of detailed analyses to reveal the advantages and feasibility of hybrid utilization. However, even with the aforementioned key insights, implementing hybrid SD in VLA models presents several challenges: (1) draft rejection and persistent errors in retrieval-based SD; (2) difficulty in determining the hybrid boundary. To address these, we propose the HeiSD framework. We propose a retrieval-based SD optimization method in HeiSD, which contains a verify-skip mechanism and a sequence-wise relaxed acceptance strategy. Moreover, we proposed a kinematic-based fused metric in HeiSD to automatically determine the hybrid boundary. Experimental results demonstrate that HeiSD attains a speedup of up to 2.45x in simulation benchmarks and 2.06x~2.41x in real-world scenarios, while sustaining a high task success rate.
format Preprint
id arxiv_https___arxiv_org_abs_2603_17573
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle HeiSD: Hybrid Speculative Decoding for Embodied Vision-Language-Action Models with Kinematic Awareness
Zheng, Zihao
Mao, Zhihao
Tian, Sicheng
Li, Maoliang
Chen, Jiayu
Sun, Xinhao
Zhang, Zhaobo
Liu, Xuanzhe
Cao, Donggang
Mei, Hong
Chen, Xiang
Robotics
Databases
Machine Learning
Vision-Language-Action (VLA) Models have become the mainstream solution for robot control, but suffer from slow inference speeds. Speculative Decoding (SD) is a promising acceleration method which can be divided into two categories: drafter-based SD and retrieval-based SD. Each of the two methods demonstrates complementary advantages and limitations when applied to VLA models, leading to the hypothesis that a hybrid approach integrating these two methods will yield better performance. In this paper, we first conduct a series of detailed analyses to reveal the advantages and feasibility of hybrid utilization. However, even with the aforementioned key insights, implementing hybrid SD in VLA models presents several challenges: (1) draft rejection and persistent errors in retrieval-based SD; (2) difficulty in determining the hybrid boundary. To address these, we propose the HeiSD framework. We propose a retrieval-based SD optimization method in HeiSD, which contains a verify-skip mechanism and a sequence-wise relaxed acceptance strategy. Moreover, we proposed a kinematic-based fused metric in HeiSD to automatically determine the hybrid boundary. Experimental results demonstrate that HeiSD attains a speedup of up to 2.45x in simulation benchmarks and 2.06x~2.41x in real-world scenarios, while sustaining a high task success rate.
title HeiSD: Hybrid Speculative Decoding for Embodied Vision-Language-Action Models with Kinematic Awareness
topic Robotics
Databases
Machine Learning
url https://arxiv.org/abs/2603.17573