Sense4FL:基于车际感知增强的联邦学习用于自动驾驶

Sense4FL: Vehicular Crowdsensing Enhanced Federated Learning for Autonomous Driving

摘要 Abstract

为了适应不断变化的道路状况,实时模型训练对于自动驾驶(AD)至关重要。联邦学习(FL)作为一种有前景的范式,能够使自动驾驶车辆利用车载计算资源协同训练模型。然而,现有的联邦学习车辆选择方案均假设车辆的数据集是预先确定且位置无关的,忽略了车辆沿其行驶路线收集训练数据的事实,从而导致次优的车辆选择结果。为提高特定区域的感知质量,我们提出了Sense4FL,这是一种基于车际感知增强的联邦学习框架,具有轨迹相关的车辆训练数据收集特性。为此,我们首先通过考虑车辆不确定轨迹及其上传概率的影响,推导了联邦学习的收敛界,并发现最小化训练损失等价于最小化车辆收集数据分布与全局数据分布之间的局部和全局地球移动者距离(EMD)的加权和。基于这一观察,我们将联邦学习在自动驾驶中的轨迹相关车辆选择与数据收集问题形式化。鉴于该问题是NP难问题,我们开发了一种高效的算法,在保证近似解的情况下找到解决方案。大量的仿真结果表明,与现有基准方法相比,我们的方法在提升目标检测性能方面具有显著优势。

To accommodate constantly changing road conditions, real-time 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. To improve the perception quality in AD for a region, we propose Sense4FL, a vehicular crowdsensing-enhanced FL framework featuring trajectory-dependent vehicular training data collection. 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.