arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.08646cs.LGcs.AIcs.CV
收藏

AutoFed:用于鲁棒自动驾驶的异构感知联邦多模态学习

AutoFed: Heterogeneity-Aware Federated Multimodal Learning for Robust Autonomous Driving

Nanyang Technological University (NTU)(南洋理工大学);University of Maryland(马里兰大学);China-Singapore International Joint Research Institute(中新国际联合研究院)

Tianyue Zheng, Ang Li, Zhe Chen, Hongbo Wang, Jun Luo

摘要:

利用机载传感器进行目标检测(例如,激光雷达、雷达和摄像头)在自动驾驶(AD)中发挥着至关重要的作用,这些传感器在模式上相互补充。虽然crowdsensing可能会利用这些传感器(数量巨大)来获得更全面的知识,但\textit{federated learning}(FL)似乎是实现这一潜力的必要工具:它使自动驾驶汽车(AV)能够在不显式共享原始传感数据的情况下训练机器学习模型。然而,多模态传感器在分布式AV之间引入了各种数据异质性(例如,标签数量偏差和不同的方式),提出了关键的挑战,有效的FL。为此,我们提出AutoFed作为异质性意识FL框架,充分利用多模态感官数据的AV,从而使强大的AD。具体来说,我们首先提出了一种新的模型,利用伪标签,以避免错误地对待未标记的对象作为背景。我们还提出了一种基于自动编码器的数据填补方法,以填补缺失的数据模态(某些AV)与可用的。为了进一步协调异构性,我们最后提出了一个客户端选择机制,利用客户端模型之间的相似性,以提高训练稳定性和收敛速度。我们在基准数据集上的实验证实,AutoFed在准确率和召回率方面都大大提高了现状方法,同时对恶劣天气条件表现出强大的鲁棒性。

英文摘要:

Object detection with on-board sensors (e.g., lidar, radar, and camera) play a crucial role in autonomous driving (AD), and these sensors complement each other in modalities. While crowdsensing may potentially exploit these sensors (of huge quantity) to derive more comprehensive knowledge, \textit{federated learning} (FL) appears to be the necessary tool to reach this potential: it enables autonomous vehicles (AVs) to train machine learning models without explicitly sharing raw sensory data. However, the multimodal sensors introduce various data heterogeneity across distributed AVs (e.g., label quantity skews and varied modalities), posing critical challenges to effective FL. To this end, we present AutoFed as a heterogeneity-aware FL framework to fully exploit multimodal sensory data on AVs and thus enable robust AD. Specifically, we first propose a novel model leveraging pseudo-labeling to avoid mistakenly treating unlabeled objects as the background. We also propose an autoencoder-based data imputation method to fill missing data modality (of certain AVs) with the available ones. To further reconcile the heterogeneity, we finally present a client selection mechanism exploiting the similarities among client models to improve both training stability and convergence rate. Our experiments on benchmark dataset confirm that AutoFed substantially improves over status quo approaches in both precision and recall, while demonstrating strong robustness to adverse weather conditions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑