arXiv:2605.29705cs.AI2026-05中稿 · CVPR

用1.58比特量化让大模型在边缘设备上高效预测轨迹

BitTP: The Lightweight Trajectory Prediction Model with BitLLM for Edge-Devices

论文配图:BitTP: The Lightweight Trajectory Prediction Model with BitLLM for Edge-Devices
图 1 · 摘自论文原文
  • 将大模型权重压缩至1.58比特,保持激活值全精度
  • 相比原模型降低14.3%平均位移误差,减少21.0%最终位移误差
  • 适合部署在资源受限的自动驾驶边缘设备

轨迹预测是自主系统的核心任务,需对多智能体交互与意图进行复杂推理。近期研究采用大语言模型(LLM)实现该任务,因其具备强大的上下文推理能力与可解释的语言化轨迹表示。然而,基于LLM的预测器内存与计算开销巨大,难以部署于车载等资源受限的边缘设备。为此,本文提出BitTP,将LLM轨迹预测器转换为轻量级的位线性架构。实验证明,仅对权重进行1.58比特量化(BitTP-Weight)达到最优效果。关键在于激活值必须保持全精度,否则会严重损害时空推理性能。实验表明,BitTP-Weight不仅保持甚至提升了预测质量,在平均ADE上降低14.29%,FDE降低20.97%,同时显著降低内存占用与推理延迟。结果说明,精心设计的量化可作为有效正则化手段,实现复杂LLM推理在边缘设备上的实用部署。代码已公开于:https://github.com/MintCat98/BitTP。

原文摘要 · Abstract (English)

Trajectory prediction is a fundamental task for autonomous systems, requiring complex reasoning about multi-agent interactions and intents. Large language models (LLMs) have recently been adopted for this task, as they provide strong contextual reasoning and interpretable, language-based trajectory representations. However, these LLM-based predictors are extremely memory- and compute-intensive, making them difficult to deploy on resource-constrained edge devices such as on-board computers in autonomous robots. To bridge this gap, we propose BitTP, which converts an LLM-based trajectory predictor into a lightweight bitlinear architecture. We demonstrate that weight-only quantization to 1.58-bit (BitTP-Weight) is optimal. Crucially, activations must remain in full precision, as quantizing them leads to severe degradation and instability in spatio-temporal reasoning. Empirically, BitTP-Weight not only preserves but improves prediction quality over the full-precision (BF16) LLM baseline, reducing ADE by 14.29% and FDE by 20.97% on average, while simultaneously reducing memory usage and inference latency relative to other quantization methods. These results demonstrate that carefully designed quantization acts as an effective regularizer, enabling the practical deployment of sophisticated LLM-based reasoning on edge devices. Code is available at: https://github.com/MintCat98/BitTP.

轨迹预测边缘计算量化大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。