arXiv:2507.23318cs.CVcs.AI2025-07AAAI被引 26

通过重建前景信息高效剪枝视觉令牌,提升自动驾驶模型推理速度

FastDriveVLA: Efficient End-to-End Driving via Plug-and-Play Reconstruction-based Token Pruning

  • 基于像素重建的前景优先剪枝,保留驾驶关键区域
  • 在nuScenes数据集上不同剪枝率下均达最优性能
  • 可即插即用,适配多种视觉语言动作模型

视觉-语言-动作(VLA)模型在复杂场景理解与决策推理中展现巨大潜力,正被广泛应用于端到端自动驾驶系统。然而,其长序列视觉令牌导致计算开销过大。现有视觉令牌剪枝方法依赖视觉相似性或视觉-文本注意力,在自动驾驶场景中表现不佳。考虑到人类驾驶员聚焦于前景区域,我们提出FastDriveVLA,一种专为自动驾驶设计的基于重建的视觉令牌剪枝框架。该框架包含一个名为ReconPruner的即插即用剪枝器,通过类MAE像素重建机制优先保留前景信息。引入对抗性前景-背景重建策略训练ReconPruner,使其适配VLA模型的视觉编码器。训练完成后,ReconPruner无需重训即可应用于同视觉编码器的不同VLA模型。为训练该剪枝器,我们构建了大规模数据集nuScenes-FG,包含24.1万张带前景标注的图像-掩码对。实验表明,该方法在nuScenes开放环路规划基准测试中,于多种剪枝率下均达到当前最优表现。

原文摘要 · Abstract (English)

Vision-Language-Action (VLA) models have demonstrated significant potential in complex scene understanding and action reasoning, leading to their increasing adoption in end-to-end autonomous driving systems. However, the long visual tokens of VLA models greatly increase computational costs. Current visual token pruning methods in Vision-Language Models (VLM) rely on either visual token similarity or visual-text attention, but both have shown poor performance in autonomous driving scenarios. Given that human drivers concentrate on relevant foreground areas while driving, we assert that retaining visual tokens containing this foreground information is essential for effective decision-making. Inspired by this, we propose FastDriveVLA, a novel reconstruction-based vision token pruning framework designed specifically for autonomous driving. FastDriveVLA includes a plug-and-play visual token pruner called ReconPruner, which prioritizes foreground information through MAE-style pixel reconstruction. A novel adversarial foreground-background reconstruction strategy is designed to train ReconPruner for the visual encoder of VLA models. Once trained, ReconPruner can be seamlessly applied to different VLA models with the same visual encoder without retraining. To train ReconPruner, we also introduce a large-scale dataset called nuScenes-FG, consisting of 241K image-mask pairs with annotated foreground regions. Our approach achieves state-of-the-art results on the nuScenes open-loop planning benchmark across different pruning ratios.

自动驾驶视觉剪枝重建学习VLA模型

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