Semi-asynchronous Hierarchical Federated Learning for Cooperative Intelligent Transportation Systems

Qimei Chen,Zehua You,Hao Jiang
DOI: https://doi.org/10.48550/arXiv.2110.09073
2021-10-18
Abstract:Cooperative Intelligent Transport System (C-ITS) is a promising network to provide safety, efficiency, sustainability, and comfortable services for automated vehicles and road infrastructures by taking advantages from participants. However, the components of C-ITS usually generate large amounts of data, which makes it difficult to explore data science. Currently, federated learning has been proposed as an appealing approach to allow users to cooperatively reap the benefits from trained participants. Therefore, in this paper, we propose a novel Semi-asynchronous Hierarchical Federated Learning (SHFL) framework for C-ITS that enables elastic edge to cloud model aggregation from data sensing. We further formulate a joint edge node association and resource allocation problem under the proposed SHFL framework to prevent personalities of heterogeneous road vehicles and achieve communication-efficiency. To deal with our proposed Mixed integer nonlinear programming (MINLP) problem, we introduce a distributed Alternating Direction Method of Multipliers (ADMM)-Block Coordinate Update (BCU) algorithm. With this algorithm, a tradeoff between training accuracy and transmission latency has been derived. Numerical results demonstrate the advantages of the proposed algorithm in terms of training overhead and model performance.
Machine Learning
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is how to effectively manage a large amount of data in the Cooperative Intelligent Transportation System (C - ITS) while ensuring communication efficiency and the accuracy of model training. Specifically, the paper proposes a new Semi - asynchronous Hierarchical Federated Learning (SHFL) framework, aiming to enable C - ITS from data perception through elastic edge - to - cloud model aggregation. In addition, the paper also proposes a joint edge - node association and resource allocation problem to prevent the individuality of different types of road vehicles and achieve communication efficiency. To deal with the proposed Mixed - Integer Nonlinear Programming (MINLP) problem, the authors introduce the Distributed Alternating Direction Method of Multipliers (ADMM) - Block Coordinate Update (BCU) algorithm. This algorithm can find a balance between training accuracy and transmission delay. Through numerical results, the advantages of the proposed algorithm in terms of training cost and model performance are proved.