基于车辆轨迹动态选车,提升自动驾驶目标检测精度。
Sense4FL: Vehicular Crowdsensing Enhanced Federated Learning for Object Detection in Autonomous Driving
- 根据车辆行驶轨迹动态选择参与联邦学习的车辆。
- 实验显示目标检测性能显著优于现有方法。
- 适合研究自动驾驶协同学习与车联网数据优化者。
为应对不断变化的道路状况,实时视觉模型训练对自动驾驶(AD)至关重要。联邦学习(FL)可利用车载计算资源实现车辆间协同训练,但现有车辆选择方案均假设车辆数据集固定且与位置无关,忽略了车辆沿行驶路线采集数据的事实,导致选择效果不佳。本文聚焦基础感知问题,提出 Sense4FL 框架,通过引入轨迹相关的车辆数据采集机制,提升特定区域内的目标检测质量。我们推导了考虑车辆不确定轨迹与上传概率影响的 FL 收敛边界,发现最小化训练损失等价于最小化局部与全局地球移动距离(EMD)加权和。基于此,构建了面向自动驾驶的轨迹依赖型车辆选择与数据收集模型。由于该问题是 NP-hard,我们设计了一种具有近似保证的高效算法。大量仿真结果表明,该方法在提升目标检测性能方面优于现有基准。
原文摘要 · Abstract (English)
To accommodate constantly changing road conditions, real-time vision model training is essential for autonomous driving (AD). Federated learning (FL) serves as a promising paradigm to enable autonomous vehicles to train models collaboratively with their onboard computing resources. However, existing vehicle selection schemes for FL all assume predetermined and location-independent vehicles' datasets, neglecting the fact that vehicles collect training data along their routes, thereby resulting in suboptimal vehicle selection. In this paper, we focus on the fundamental perception problem and propose Sense4FL, a vehicular crowdsensing-enhanced FL framework featuring \textit{trajectory-dependent} vehicular \textit{training data collection} to \rev{improve the object detection quality} in AD for a region. To this end, we first derive the convergence bound of FL by considering the impact of both vehicles' uncertain trajectories and uploading probabilities, from which we discover that minimizing the training loss is equivalent to minimizing a weighted sum of local and global earth mover's distance (EMD) between vehicles' collected data distribution and global data distribution. Based on this observation, we formulate the trajectory-dependent vehicle selection and data collection problem for FL in AD. Given that the problem is NP-hard, we develop an efficient algorithm to find the solution with an approximation guarantee. Extensive simulation results have demonstrated the effectiveness of our approach in improving object detection performance compared with existing benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。