rMuscle框架提升机器人视觉语言行动模型推理效率
rMuscle: Robotic Muscle Memory for Efficient Vision-Language-Action Model Inference
这个研究挺有意思,用类似肌肉记忆的缓存机制来优化机器人模型的推理速度,对实际应用应该有帮助。
这篇论文提出rMuscle框架,通过利用机器人重复执行任务间的相似性来加速推理。它使用上下文缓存重用视觉token输出,动作缓存重用神经元激活模式,在RTX 4090和Jetson Thor上测试,在LIBERO、RoboTwin和物理操作任务上实现了1.
29-1.42倍的速度提升。
rMuscle: Robotic Muscle Memory for Efficient Vision-Language-Action Model Inference
Factory work is a promising early scenario for embodied AI: assigning repetitive manual jobs to robots has clear economic payoff, and a structured station keeps the jobs tractable for current policies. Vision-Language-Action (VLA) models now dominate as the policy paradigm for these robots. The inference latency of VLA models directly affects robot responsiveness and motion smoothness. However, existing VLA inference frameworks do not fully exploit the characteristics of embodied workloads or account for the distinct bottlenecks across different stages of VLA inference. In this paper, we first characterize embodied workloads and identify substantial task similarity across repeated robot executions. We further find that such similarity extends beyond observations and action trajectories to internal model states. Drawing on these observations, we present rMuscle, a real-time VLA inference framework inspired by human muscle memory. It exploits cross-execution similarity through a dual-phase muscle-memory cache. The Context Cache reuses visual-token outputs to reduce computation, while the Action Cache reuses neuron activation patterns to reduce weight accesses. We keep both the cache memory footprint and access overhead low through online cache recomputation, sliding-window cache retrieval, and mask sharing across consecutive denoising steps. rMuscle achieves 1.29-1.42X speedup on RTX 4090 and Jetson Thor across LIBERO, RoboTwin, and physical manipulation tasks, while maintaining the original success rates on real-world robots.
本条情报归属 AITOP「论文」核心栏目,重点聚焦北美前沿实验室与硅谷科技巨头。在跨区域产业观察坐标中,该动态直接关联「AI 开发者工具链、Coding Agent 与工程化实践」研究主线,反映出当前技术演进与商业落地的最新风向。涉及核心实体与技术议题包括:rMuscle、VLA模型、推理优化、机器人,读者可通过下方标签进入对应专题页查阅历史脉络与关联对比。
信源核验说明:本文原始线索采编自「arXiv cs.AI」(@Kaijun Zhou, Zhiyang Li, Le Chen, Jinyu Gu),经 AITOP 多维评分管线结构化提炼。