arXiv:2606.07464cs.ROcs.AI2026-06中稿 · ICRA被引 1

让自动驾驶模型压缩长时上下文,保留关键决策信息

Planning-aligned Token Compression for Long-Context Autonomous Driving

论文配图:Planning-aligned Token Compression for Long-Context Autonomous Driving
图 1 · 摘自论文原文
  • 基于规划意图的条件压缩,动态保留重要信息
  • 在相同计算预算下,成功率提升6%以上至68.3%
  • 适合需要长时记忆与实时决策的自动驾驶系统

单体视觉-动作模型是自动驾驶的新范式,但长时间上下文编码会超出实时计算预算。现有压缩方法多依赖规则,与决策规划脱节,易丢失关键信息。本文提出COMPACT-VA框架,基于条件VQ-VAE实现规划对齐的记忆压缩:压缩过程结合历史轨迹与从未来轨迹中学习的规划意图,后验编码器提取意图,先验编码器预测该意图。压缩记忆与预测隐状态拼接后输入策略网络,实现端到端优化。在高动态场景下评估,设计行为指标验证有效性。在相近令牌预算下,成功率达68.3%,优于基线超6%;消融实验验证了规划对齐的有效性。闭环测试表明,相比未压缩处理,本方法实现3.3倍加速和2.7倍内存减少,同时保持驾驶性能。

原文摘要 · Abstract (English)

Monolithic vision-action models represent an emerging paradigm in autonomous driving. However, this architecture produces token sequences that quickly exceed real-time computational budgets when encoding extended temporal context for complex interactions. While approaches like linear transformers and external memory try to make the context lightweight, token compression is most compatible with the architecture as it requires no backbone modifications. Yet existing compression adopts rule-based heuristics like temporal decay, decoupled from planning, risking loss of decision-critical information. We propose COMPACT-VA, a planning-aligned working memory framework built on conditional VQ-VAE, compressing extended context into bounded representations. Compression is conditioned on both historical trajectory and a learned planning intent that the posterior encoder distills from future trajectories during training, while the prior encoder learns to predict it from compressed observations. The compressed memory, concatenated with the predicted latent, feeds the policy for end-to-end optimization, planning with retained decision-critical information. We evaluate on high-signal dynamic scenarios where historical context is most critical for behavior correctness (e.g., stop, yield, or proceed), and accordingly design behavioral metrics. Under comparable token budgets, we achieve $>$6% improvement (68.3%) on success rates with consistent gains across metrics. Ablations validate planning-aligned coupling effectiveness. Closed-loop evaluation confirms that COMPACT-VA maintained general driving performance with 3.3* speedup and 2.7* memory reduction over uncompressed processing.

自动驾驶长时记忆压缩规划对齐

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