Efficient Block-Layer Parallel Inference for Vision-Language-Action on Hybrid Architectures
Proposes a block-level heterogeneous CPU-GPU inference framework with cross-frame asynchronous pipeline, reducing latency by over 20% on VLA models.
Key Findings
Methodology
This work introduces a block-layer partitioning strategy for VLA models, dividing Transformer backbone into GPU prefix and CPU suffix. It employs a cross-frame asynchronous pipeline to overlap inference stages, dynamically adjusting layer splits via resource-aware scheduling. Experiments on Orion and MindDrive demonstrate latency reductions of 21.7% and 30.9%, with GPU memory dropping from 45GB to 29GB. The system is validated on real vehicle platforms, enabling deployment under limited GPU memory conditions.
Key Results
- On Bench2Drive, Orion's average latency decreased from 521ms to 408ms, a 21.7% reduction; MindDrive from 443ms to 306.2ms, a 30.9% reduction.
- GPU peak memory was reduced from 45GB to 29GB, easing hardware constraints.
- In real vehicle tests with Autoware.Universe, native Orion failed due to memory limits, but the hybrid system operated successfully, confirming practicality.
Significance
This study addresses critical bottlenecks in deploying large-scale VLA models for autonomous driving. By leveraging heterogeneous resources and asynchronous pipelines, it achieves low-latency, memory-efficient inference suitable for real-world vehicles. The approach bridges the gap between model scale and hardware capability, paving the way for scalable, high-performance autonomous systems with broader industrial impact.
Technical Contribution
The paper introduces a novel block-layer partitioning combined with cross-frame asynchronous inference, enabling effective CPU-GPU collaboration. It preserves model integrity by partitioning at Transformer blocks, avoids disrupting feature continuity, and employs dynamic scheduling for optimal resource utilization. These innovations significantly improve inference efficiency and deployment flexibility in resource-constrained environments.
Novelty
This is the first work to integrate block-layer model partitioning with cross-frame asynchronous pipelines for VLA inference in autonomous driving. Unlike prior model compression or layer skipping methods, it exploits temporal continuity and heterogeneous hardware synergy, offering a fundamentally new approach to real-time multi-frame multimodal inference.
Limitations
- Dependence on Transformer architecture limits applicability to non-Transformer models.
- Dynamic scheduling may face challenges under extreme resource fluctuations, affecting robustness.
- Communication overhead, though minimized, can still impact ultra-low latency scenarios.
Future Work
Future research will focus on adaptive multi-modal fusion strategies, hardware-aware scheduling algorithms, and extending the framework to multi-task scenarios. Enhancing robustness under diverse hardware conditions and exploring model compression techniques will further improve deployment scalability.
AI Executive Summary
Autonomous driving demands rapid, reliable perception and decision-making, yet current large-scale multimodal models like VLA face significant deployment challenges. Their high computational complexity leads to increased latency and GPU memory consumption, especially problematic in resource-constrained vehicle platforms. Traditional inference approaches, often serial and device-specific, cannot meet the real-time requirements of dynamic traffic environments.
To address this, the authors propose a novel hybrid CPU-GPU inference framework that leverages the layered structure of Transformer-based VLA models. By partitioning the model at the block level, the system assigns the visual encoder and language model prefix to the GPU, while offloading the suffix to the CPU. This partitioning preserves model integrity and enables flexible resource management. The key innovation is the cross-frame asynchronous pipeline, which overlaps GPU processing of the current frame with CPU processing of the previous frame's suffix, effectively pipelining multiple frames and reducing overall latency.
Experimental results on benchmarks such as Bench2Drive demonstrate substantial improvements: the Orion model's latency drops from 521ms to 408ms, and MindDrive's from 443ms to 306.2ms, with GPU memory usage decreasing from 45GB to 29GB. These gains translate into more responsive, memory-efficient autonomous systems capable of real-world deployment. The framework's adaptability was validated on an actual vehicle platform, where it successfully integrated with the full vehicle stack, overcoming the limitations faced by native models.
This work marks a significant step toward scalable, real-time VLA inference in autonomous driving, balancing model complexity with hardware constraints. Future directions include refining scheduling algorithms, extending to multi-task scenarios, and integrating more sophisticated multi-modal fusion techniques, promising broader impacts in intelligent transportation systems.
Deep Analysis
Background
Recent advances in autonomous driving have shifted from modular pipelines toward end-to-end models that unify perception, reasoning, and control. Large multimodal models like LMDrive, CoVLA, and AutoVLA exemplify this trend, leveraging transformer architectures for scene understanding and decision-making. Industrial implementations such as Xiaomi's ORION and MindDrive demonstrate practical deployment efforts. However, these models' computational demands—especially in perception and reasoning—pose significant challenges for real-time inference on embedded vehicle hardware. Existing solutions focus on model compression, layer skipping, or quantization, but often neglect system-level resource management and heterogeneous hardware utilization, which are critical for scalable deployment.
Core Problem
The core challenge lies in executing large VLA models efficiently within the limited GPU memory and computational budget of autonomous vehicles. High latency and memory bottlenecks hinder real-time responsiveness, essential for safe driving. Traditional serial inference, where the entire model runs on GPU, leads to underutilized CPU resources and excessive memory consumption. Balancing the workload between CPU and GPU while maintaining model accuracy and low latency remains unresolved, especially under continuous multi-frame streaming scenarios with tight real-time constraints.
Innovation
The paper introduces a block-layer partitioning strategy that exploits the hierarchical structure of transformer models, enabling natural division at Transformer blocks. This preserves model integrity and facilitates heterogeneous execution. Additionally, a cross-frame asynchronous pipeline overlaps GPU prefix computation with CPU suffix processing across consecutive frames, significantly reducing end-to-end latency. A dynamic resource scheduler adjusts the layer split point based on runtime conditions, balancing load and communication overhead. Hardware-aware acceleration techniques further optimize CPU inference. These innovations collectively enable efficient, scalable deployment of VLA models in resource-constrained autonomous vehicles.
Methodology
- �� Model partitioning at Transformer block layer, dividing the backbone into GPU prefix and CPU suffix, ensuring structural integrity.
- �� Implementation of a cross-frame asynchronous pipeline, where GPU processes current frame's prefix while CPU handles previous frame's suffix, overlapping computation.
- �� Dynamic adjustment of partition point via resource-aware scheduler, based on real-time system load and memory availability.
- �� Hardware-aware CPU acceleration, utilizing optimized libraries to enhance inference speed.
- �� Validation on Orion and MindDrive models, measuring latency, memory, and inference frequency.
- �� Deployment on real vehicle platform with Autoware.Universe, demonstrating practical feasibility.
- �� Ablation studies to analyze the impact of partition point, pipeline overlap, and scheduling strategies.
Experiments
The evaluation employed Bench2Drive benchmarks, comparing the proposed hybrid framework against baseline full-GPU inference. Metrics included average latency, GPU peak memory, and inference frequency. Various layer split points were tested to identify optimal trade-offs. Real vehicle tests involved integrating the system with Autoware.Universe, assessing stability, responsiveness, and resource utilization under actual driving conditions. Hyperparameters such as batch size, frame rate, and communication buffer sizes were tuned for optimal performance. Ablation experiments isolated the effects of pipeline overlap and dynamic scheduling, confirming their contributions to efficiency gains.
Results
Results show that the hybrid approach reduces Orion's average latency from 521ms to 408ms, a 21.7% improvement, and MindDrive from 443ms to 306.2ms, a 30.9% reduction. GPU peak memory consumption drops from 45GB to 29GB, alleviating hardware constraints. The cross-frame asynchronous pipeline maintains high inference frequency (>3Hz), suitable for real-time control. In real vehicle deployment, the hybrid system successfully integrated with the full vehicle stack, whereas native Orion failed due to memory limitations. These findings demonstrate the method's effectiveness in balancing latency, memory, and system integration challenges.
Applications
This framework is directly applicable to autonomous vehicles requiring real-time perception and decision-making with constrained hardware resources. It enables scalable deployment of large multimodal models, improving responsiveness and safety. The approach can be extended to multi-task learning, multi-modal fusion, and adaptive resource management, supporting future intelligent transportation systems. Its flexibility allows integration with existing perception stacks, facilitating incremental upgrades and system robustness.
Limitations & Outlook
The approach relies on Transformer block structures, limiting applicability to models with different architectures. Dynamic scheduling may face challenges under highly variable system loads, affecting stability. Communication overhead, although minimized, could still impact ultra-low latency scenarios. Further research is needed to generalize the method across diverse hardware platforms and model types, and to optimize scheduling under extreme conditions.
Plain Language Accessible to non-experts
想象你在厨房做饭,有两个厨师:一个负责切菜(GPU),一个负责炒菜(CPU)。平时,切菜和炒菜是分开做的,但如果你让切菜的厨师专注切菜,把炒菜的厨师专注炒菜,还让他们在不同的时间同时工作,厨房的效率就会大大提高。现在,假设你还可以提前准备下一道菜的材料,让厨师们同时工作,不用等一个步骤全部完成再开始下一个。这个方法就像把复杂的模型拆成几个部分,让不同的硬件同时工作,既快又省力。
ELI14 Explained like you're 14
想象你在学校里做一个超级大项目,里面有很多步骤,比如画画、写报告、做演示。以前,你一个人做所有事情,花了很长时间。而现在,你请了几个朋友帮忙:一个负责画图,一个写报告,一个做演示。你们还安排好时间,让画图的人先完成,写报告的人在他做完后开始,演示的人最后准备。这样一来,大家同时工作,整个项目就能更快完成。这就像论文里的技术,把模型拆成几部分,让不同的硬件同时工作,节省时间,效率更高!
Glossary
Transformer (变换器)
一种深度学习模型结构,擅长处理序列数据,广泛应用于自然语言处理和视觉任务。
论文中用于构建VLA模型的核心结构,块层划分基于Transformer块。
异步流水线 (Asynchronous Pipeline)
一种并行处理技术,允许不同阶段同时进行,减少等待时间,提高效率。
本文设计的跨帧异步流水线实现GPU和CPU的并行推理。
块层划分 (Block-layer Partitioning)
将模型按Transformer块粒度划分为不同执行单元,便于异构设备调度。
核心创新之一,用于平衡GPU和CPU的计算负载。
多模态模型 (Multimodal Model)
融合多种数据类型(如视觉、语言)进行联合推理的深度学习模型。
VLA模型即为典型的多模态模型。
边界隐藏状态 (Boundary Hidden State)
模型中跨设备传输的中间激活,用于连接不同设备的模型部分。
在模型划分中传输的中间激活,保证模型完整性。
Open Questions Unanswered questions from this research
- 1 未来需探索多模态信息融合的优化策略,结合硬件感知调度算法,提升系统鲁棒性和适应性。
- 2 在极端硬件变化下,通信开销和调度策略仍需优化,以确保系统在所有场景下都能高效运行。
Abstract
Vision-Language-Action (VLA) models are becoming a promising paradigm for autonomous driving, but their deployment on existing vehicle platforms remains difficult because they introduce both high inference latency and strong GPU-side resource pressure. In a full autonomous driving stack, this problem is even more pronounced: legacy vehicle platforms were provisioned for modular pipelines, yet after several planning-related functions are absorbed into a unified VLA model, part of the original CPU budget becomes underutilized, while the visual encoder and the main reasoning path still concentrate most computation and memory demand on the GPU. As a result, directly deploying VLA together with the rest of the onboard system can be hard under realistic GPU memory constraints. To address this issue, we present a hybrid CPU--GPU inference framework with flexible resource scheduling for autonomous driving. Our design partitions the VLA backbone at the block-layer granularity, executes the visual encoder and LLM prefix on the GPU, and offloads the LLM suffix to the CPU through a cross-frame asynchronous pipeline, thereby exposing a schedulable boundary for redistributing compute and memory pressure across heterogeneous processors. We evaluate the proposed framework on two representative driving VLA models, Orion and MindDrive. On Bench2Drive, our method reduces average latency from 521ms to 408.0ms for Orion and from 443ms to 306.2ms for MindDrive, corresponding to 21.7% and 30.9% reduction, respectively. For Orion, the estimated peak GPU memory is further reduced from 45GB to 29GB. In real-vehicle deployment under coexistence with Autoware.Universe, native Orion cannot run because the onboard GPU memory budget is insufficient, whereas the hybrid version runs successfully together with the full vehicle stack.