提出SToRM框架,让多模态大模型在自动驾驶中大幅减少计算量仍保持高精度。
SToRM: Supervised Token Reduction for Multi-modal LLMs toward efficient end-to-end autonomous driving
- 用滑动窗口轻量级预测视觉标记重要性,筛选关键信息。
- 通过伪监督信号训练,使精简后性能接近全标记输入。
- 合并上下文标记到锚点,减少冗余,适合车载实时部署。
在自动驾驶中,端到端系统直接从传感器数据预测控制指令已取得显著进展。为应对意外场景,系统可借助自然语言指令等人工干预。使用多模态大语言模型(MLLM)能提升人车交互能力并改善表现,但其依赖大量视觉标记和大模型计算资源,受限于车载算力。现有方法虽尝试减少视觉标记,但常导致任务性能下降。为此,本文提出首个面向多模态大模型的有监督标记压缩框架SToRM。该框架包含三个核心组件:首先,采用带短期滑动窗口的轻量级重要性预测器估算标记重要性;其次,通过辅助路径获取来自全标记输入大模型的伪监督信号进行有监督训练;第三,设计锚点-上下文融合模块,将标记划分为锚点与上下文,并将上下文合并至相关锚点,以降低冗余并最小化信息损失。在LangAuto基准测试中,SToRM在相同缩减标记预算下优于现有最先进端到端驾驶模型,保持全标记性能的同时,计算成本降低最高达30倍,实现标准GPU上的实时端到端驾驶。
原文摘要 · Abstract (English)
In autonomous driving, end-to-end (E2E) driving systems that predict control commands directly from sensor data have achieved significant advancements. For safe driving in unexpected scenarios, these systems may additionally rely on human interventions such as natural language instructions. Using a multi-modal large language model (MLLM) facilitates human-vehicle interaction and can improve performance in such scenarios. However, this approach requires substantial computational resources due to its reliance on an LLM and numerous visual tokens from sensor inputs, which are limited in autonomous vehicles. Many MLLM studies have explored reducing visual tokens, but often suffer end-task performance degradation compared to using all tokens. To enable efficient E2E driving while maintaining performance comparable to using all tokens, this paper proposes the first Supervised Token Reduction framework for multi-modal LLMs (SToRM). The proposed framework consists of three key elements. First, a lightweight importance predictor with short-term sliding windows estimates token importance scores. Second, a supervised training approach uses an auxiliary path to obtain pseudo-supervision signals from an all-token LLM pass. Third, an anchor-context merging module partitions tokens into anchors and context tokens, and merges context tokens into relevant anchors to reduce redundancy while minimizing information loss. Experiments on the LangAuto benchmark show that SToRM outperforms state-of-the-art E2E driving MLLMs under the same reduced-token budget, maintaining all-token performance while reducing computational cost by up to 30x, and enabling real-time E2E driving on a standard GPU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。