用分桶法压缩自动驾驶规划模型,兼顾精度与安全
BucketKD: A Safety-Aware Bucket-Based Knowledge Distillation Framework for End-to-End Motion Planning

- 将环境变量分桶表示,捕捉更丰富场景语义
- 通过碰撞时间评估风险,提升学生模型安全性
- 在CARLA上实现更高精度与更强压缩比
端到端运动规划在自动驾驶中展现出巨大潜力,直接从原始传感器数据映射控制指令。然而其庞大模型尺寸限制了在资源受限平台的部署。本文提出BucketKD,一种基于分桶的知识蒸馏框架,生成紧凑且具备安全意识的端到端规划器。相较于依赖简化状态表示的现有方法,BucketKD将关键环境变量自适应分桶,保留效率的同时捕获更丰富的场景语义。此外,设计了安全感知的航点注意力机制,通过广泛用于交通研究的时间到碰撞(TTC)公式,综合障碍物距离与相对运动评估每个航点的风险等级,使学生模型在蒸馏过程中更好保留安全关键行为。在CARLA平台使用Bench2Drive数据集的大量实验表明,BucketKD在规划准确性和安全性方面显著优于现有最先进方法,同时保持强压缩比。
原文摘要 · Abstract (English)
End-to-end motion planning has emerged as a promising paradigm in autonomous driving, directly mapping raw sensor data to control commands via deep neural networks. Despite its advantages, its large model size hinders deployment in resource-constrained platforms. In this paper, we present BucketKD, a bucket-based knowledge distillation framework that yields compact and safety-aware end-to-end planners. Compared to the state-of-the-art approach, which relies on simplified planning state representations, BucketKD discretizes critical environmental variables into adaptive buckets that capture richer scene semantics while preserving efficiency. In addition, we design a safety-aware waypoint attention mechanism that evaluates each waypoint's risk level by accounting for both obstacle proximity and relative motion through a time-to-collision (TTC) formulation widely used in transportation research. This enables the student model to better retain safety-critical behaviors during distillation. Extensive experiments in CARLA using the Bench2Drive dataset show that BucketKD significantly outperforms the state-of-the-art in both planning accuracy and safety while maintaining strong compression ratios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。