Search papers, labs, and topics across Lattice.
This paper introduces Jetson-PI, a novel method for deploying Vision-Language-Action (VLA) models on low-power onboard devices, addressing the challenges of high computational complexity that lead to inference latency and low control frequency. By implementing Foresight-Aligned Asynchronous Correction, the authors train a lightweight future correction module that predicts future environment representations based on committed actions, thereby aligning perception with execution. The method achieves significant performance improvements, with control frequency enhancements of 8.66x and 5.41x over existing implementations, while also increasing the average success rate by 14.8% on the LIBERO benchmark.
Jetson-PI achieves a staggering 8.66x boost in control frequency for VLA models on low-power devices, revolutionizing real-time robot control.
Vision-Language-Action (VLA) models have achieved impressive performance on diverse embodied tasks. However, deploying VLA models on low-power onboard devices, such as the Jetson Orin, remains challenging due to their high computational complexity, which leads to substantial inference latency and low control frequency. Asynchronous inference can partially mask this latency by parallelizing action execution and subsequent inference, but it introduces two critical issues: perception-execution misalignment and long reaction time. In this paper, we propose Jetson-PI, a method for efficient VLA deployment on onboard devices via Foresight-Aligned Asynchronous Correction. To address misalignment, we train a lightweight future correction module that predicts future environment representation conditioned on committed actions, enabling the action expert to directly predict actions from the future time step. To reduce reaction time, we introduce confidence-based scheduling optimization that adaptively balances VLM and action expert invocations, complemented by system-level accelerations including CUDA graph reuse, GPU-resident intermediate buffering, and flow unrolling. Extensive experiments demonstrate that Jetson-PI achieves 8.66x and 5.41x improvements in control frequency compared with naive PyTorch and vla.cpp on NVIDIA Jetson Orin, while outperforming VLASH by 14.8\% in average success rate on the LIBERO benchmark. The code of our asynchronous algorithm is available on https://github.com/PKU-SEC-Lab/Jetson-PI, and our efficient llama.cpp-based inference engine is available on https://github.com/PKU-SEC-Lab/Jetson-PI-Edge.
