Towards Efficient Embodied Reasoning: Mixture-of-Depth Compute Allocation for Vision-Language-Action Model
Weiying Xie, Qingchen Zeng, Zihan Meng, Jiayun Tian, Sibo He, Danian Yang, Jie Du, Yunke Wang, Daixun Li, Hengyi Wang, Jitao Ma, Leyuan Fang, Yunsong Li
Abstract
Vision-Language-Action (VLA) model plays a crucial role in embodied decision making. While practical deployment requires fast inference under limited onboard computation, a full forward pass through the vision-language model makes such deployment challenging. To address this issue, existing methods typically employ lightweight techniques to compress the backbone. However, these information-lossy methods degrade spatial representations for action generation. In contrast, rate-distortion principles aim to reduce computation while retaining control-sufficient information. Inspired by this insight, we introduce Effective-Edge Flow, an action-aligned attribution measure that quantifies the marginal contribution of token interactions across network depth. This analysis reveals a consistent depth asymmetry, with visual evidence dominating early layers and linguistic reasoning sustaining task-relevant influence into deeper layers. Building on this structure, we propose MoDeVLA, the first rate-distortion driven efficient VLA model that performs token-wise depth allocation via Mixture-of-Depth Conditioning and integrates shallow visual-spatial with deep textual-logical features for action conditioning. Extensive real-robot evaluations across 20 tasks and multiple embodiments demonstrate that MoDeVLA preserves task performance while reducing latency by about 38% and FLOPs by 86% on edge device NVIDIA Jetson Orin, highlighting its strong ability for embodied systems deployment.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get a6591ce6-410e-4d16-8cd3-2375c0769f5cRelated papers
- SP-VLA: A Joint Model Scheduling and Token Pruning Approach for VLA Model AccelerationYe Li, Yuan Meng, Zewen Sun, Kangye Ji et al.ICLR 2026 · 60 citations
- QVLA: Not All Channels Are Equal in Vision-Language-Action Model's QuantizationYuhao Xu, Yantai Yang, Zhenyang Fan, Yufan Liu et al.ICLR 2026 · 21 citations
- EfficientVLA: Training-Free Acceleration and Compression for Vision-Language-Action ModelsYantai Yang, Yuhao Wang, Zichen Wen, Luo Zhongwei et al.NeurIPS 2025 · 94 citations
- CogVLA: Cognition-Aligned Vision-Language-Action Models via Instruction-Driven Routing & SparsificationWei Li, Renshan Zhang, Rui Shao, Jie He et al.NeurIPS 2025 · 87 citations
- Characterizing Vision-Language-Action Models across XPUs: Constraints and Acceleration for On-Robot DeploymentKaijun Zhou, Qiwei Chen, Da Peng, Zhiyang Li et al.ICML 2026 · 4 citations
