arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.01034cs.ROcs.CVcs.LG
收藏

面向资源高效机器人控制的量化感知模仿学习

Quantization-Aware Imitation-Learning for Resource-Efficient Robotic Control

Hanyang University(汉阳大学);Hyundai Motor Company(现代汽车公司)

Seongmin Park, Hyungmin Kim, Wonseok Jeon, Juyoung Yang, Byeongwook Jeon, Yoonseon Oh, Jungwook Choi

AI总结:

本文提出量化感知模仿学习框架,通过训练时微调提升策略模型对低比特误差的鲁棒性,在机器人和自动驾驶任务中显著提速降耗并保持精度。

摘要:

基于深度神经网络(DNN)的策略模型(如视觉-语言-动作(VLA)模型)通过解释多模态数据,正在变革各类应用中复杂决策的自动化。然而,扩展这些模型会大幅增加计算成本,从而给机器人操作和自动驾驶等需要快速、准确响应的领域带来挑战。为满足在资源受限硬件上部署的需求,我们提出了一种面向基于模仿学习(IL)的策略模型的新型量化框架,该框架在训练期间微调参数,以增强模型对低比特精度误差的鲁棒性,从而在受限条件下保持效率和可靠性。我们在真实边缘GPU上针对具有代表性的机器人操作进行4比特权重量化评估,结果表明,我们的框架在保持精度的同时,实现了最高2.5倍的加速和2.5倍的能耗节省。对于4比特权重和激活量化的自动驾驶模型,该框架在低端GPU上实现了最高3.7倍的加速和3.1倍的能耗节省。这些结果凸显了在资源受限设备上部署基于IL的策略模型的实际潜力。

英文摘要:

Deep neural network (DNN)-based policy models like vision-language-action (VLA) models are transformative in automating complex decision-making across applications by interpreting multi-modal data. However, scaling these models greatly increases computational costs, which presents challenges in fields like robot manipulation and autonomous driving that require quick, accurate responses. To address the need for deployment on resource-limited hardware, we propose a new quantization framework for IL-based policy models that fine-tunes parameters to enhance robustness against low-bit precision errors during training, thereby maintaining efficiency and reliability under constrained conditions. Our evaluations with representative robot manipulation for 4-bit weight-quantization on a real edge GPU demonstrate that our framework achieves up to 2.5x speedup and 2.5x energy savings while preserving accuracy. For 4-bit weight and activation quantized self-driving models, the framework achieves up to 3.7x speedup and 3.1x energy saving on a low-end GPU. These results highlight the practical potential of deploying IL-based policy models on resource-constrained devices.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑