FedRAV: Hierarchically Federated Region-Learning for Traffic Object Classification of Autonomous Vehicles

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhai, Yijun, Zhou, Pengzhan, He, Yuepeng, Qu, Fang, Qin, Zhida, Jiao, Xianlong, Liu, Guiyan, Guo, Songtao
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912128577830912
author Zhai, Yijun
Zhou, Pengzhan
He, Yuepeng
Qu, Fang
Qin, Zhida
Jiao, Xianlong
Liu, Guiyan
Guo, Songtao
author_facet Zhai, Yijun
Zhou, Pengzhan
He, Yuepeng
Qu, Fang
Qin, Zhida
Jiao, Xianlong
Liu, Guiyan
Guo, Songtao
contents The emerging federated learning enables distributed autonomous vehicles to train equipped deep learning models collaboratively without exposing their raw data, providing great potential for utilizing explosively growing autonomous driving data. However, considering the complicated traffic environments and driving scenarios, deploying federated learning for autonomous vehicles is inevitably challenged by non-independent and identically distributed (Non-IID) data of vehicles, which may lead to failed convergence and low training accuracy. In this paper, we propose a novel hierarchically Federated Region-learning framework of Autonomous Vehicles (FedRAV), a two-stage framework, which adaptively divides a large area containing vehicles into sub-regions based on the defined region-wise distance, and achieves personalized vehicular models and regional models. This approach ensures that the personalized vehicular model adopts the beneficial models while discarding the unprofitable ones. We validate our FedRAV framework against existing federated learning algorithms on three real-world autonomous driving datasets in various heterogeneous settings. The experiment results demonstrate that our framework outperforms those known algorithms, and improves the accuracy by at least 3.69%. The source code of FedRAV is available at: https://github.com/yjzhai-cs/FedRAV.
format Preprint
id arxiv_https___arxiv_org_abs_2411_13979
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FedRAV: Hierarchically Federated Region-Learning for Traffic Object Classification of Autonomous Vehicles
Zhai, Yijun
Zhou, Pengzhan
He, Yuepeng
Qu, Fang
Qin, Zhida
Jiao, Xianlong
Liu, Guiyan
Guo, Songtao
Distributed, Parallel, and Cluster Computing
Artificial Intelligence
The emerging federated learning enables distributed autonomous vehicles to train equipped deep learning models collaboratively without exposing their raw data, providing great potential for utilizing explosively growing autonomous driving data. However, considering the complicated traffic environments and driving scenarios, deploying federated learning for autonomous vehicles is inevitably challenged by non-independent and identically distributed (Non-IID) data of vehicles, which may lead to failed convergence and low training accuracy. In this paper, we propose a novel hierarchically Federated Region-learning framework of Autonomous Vehicles (FedRAV), a two-stage framework, which adaptively divides a large area containing vehicles into sub-regions based on the defined region-wise distance, and achieves personalized vehicular models and regional models. This approach ensures that the personalized vehicular model adopts the beneficial models while discarding the unprofitable ones. We validate our FedRAV framework against existing federated learning algorithms on three real-world autonomous driving datasets in various heterogeneous settings. The experiment results demonstrate that our framework outperforms those known algorithms, and improves the accuracy by at least 3.69%. The source code of FedRAV is available at: https://github.com/yjzhai-cs/FedRAV.
title FedRAV: Hierarchically Federated Region-Learning for Traffic Object Classification of Autonomous Vehicles
topic Distributed, Parallel, and Cluster Computing
Artificial Intelligence
url https://arxiv.org/abs/2411.13979