Federated Learning with Heterogeneous Data Handling for Robust Vehicular Object Detection

Ahmad Khalil,Tizian Dege,Pegah Golchin,Rostyslav Olshevskyi,Antonio Fernandez Anta,Tobias Meuser
2024-05-02
Abstract:In the pursuit of refining precise perception models for fully autonomous driving, continual online model training becomes essential. Federated Learning (FL) within vehicular networks offers an efficient mechanism for model training while preserving raw sensory data integrity. Yet, FL struggles with non-identically distributed data (e.g., quantity skew), leading to suboptimal convergence rates during model training. In previous work, we introduced FedLA, an innovative Label-Aware aggregation method addressing data heterogeneity in FL for generic scenarios.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to improve the robustness and convergence speed of the object detection model training in federated learning (FL) in vehicle networks under the heterogeneous data environment. Specifically: 1. **Heterogeneous data problem**: In federated learning, when the data distribution of clients is inconsistent (i.e., non - independent and identically distributed, non - IID), it will lead to a slower convergence speed of model training and a decline in performance. Especially in vehicle networks, the data collected by different vehicles may have significant differences, such as sensor data in different locations, times or environmental conditions. 2. **Limitations of existing methods**: Traditional federated learning methods such as FedAvg perform poorly when dealing with non - IID data, and although the existing improved methods have been improved in some aspects, they still have deficiencies in vehicle perception scenarios. To solve these problems, the author proposes a new method named **FedProx + LA**, which combines the advantages of previously proposed FedLA and FedProx. FedProx + LA copes with data heterogeneity by introducing label - aware aggregation and proximal term, thereby achieving better performance and faster convergence speed in vehicle object detection tasks. ### Main contributions: - **Propose FedProx + LA**: A new method that combines the advantages of FedProx and FedLA, specifically targeting the heterogeneous data problem in vehicle networks. - **Experimental verification**: Through comparative experiments with traditional methods (such as FedAvg, FedProx, etc.), it is proved that FedProx + LA has superior performance under non - IID data distribution. - **Performance improvement**: The experimental results on the NuScenes dataset show that FedProx + LA improves the detection performance (mAP) by 6% compared with the baseline method and increases the convergence speed by 30%. ### Summary: The core objective of this paper is to solve the challenges brought by data heterogeneity in vehicle networks through improving the federated learning algorithm, thereby achieving more efficient and accurate object detection model training.