Autofed: Heterogeneity-aware federated multimodal learning for robust autonomous driving

T Zheng, A Li, Z Chen, H Wang, J Luo - Proceedings of the 29th Annual …, 2023 - dl.acm.org
Proceedings of the 29th Annual International Conference on Mobile Computing …, 2023dl.acm.org
Object detection with on-board sensors (eg, lidar, radar, and camera) is crucial to
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, federated learning (FL) appears to be the necessary tool to
reach this potential: it enables autonomous vehicle s (AVs) to train machine learning models
without explicitly sharing raw sensory data. However, the multimodal sensors introduce …
Object detection with on-board sensors (e.g., lidar, radar, and camera) is crucial to 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, 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 based on client model similarities to improve training stability and convergence rate. Our experiments confirm that AutoFed substantially improves over status quo in both precision and recall, while demonstrating strong robustness to adverse weather.
ACM Digital Library
以上显示的是最相近的搜索结果。 查看全部搜索结果