Toward Robust Hierarchical Federated Learning in Internet of Vehicles
Research output: Journal Publications and Reviews › RGC 21 - Publication in refereed journal › peer-review
Author(s)
Related Research Unit(s)
Detail(s)
Original language | English |
---|---|
Pages (from-to) | 5600-5614 |
Journal / Publication | IEEE Transactions on Intelligent Transportation Systems |
Volume | 24 |
Issue number | 5 |
Online published | 16 Feb 2023 |
Publication status | Published - May 2023 |
Link(s)
Abstract
The rapid growth of the Internet of Vehicles (IoV) paradigm sparks the generation of large volumes of distributed data at vehicles, which can be harnessed to build models for intelligent applications. Federated learning has recently received wide attentions, which allows model training over distributed datasets without requiring raw datasets to be shared out. However, federated learning is known to be vulnerable to poisoning attacks, where malicious clients may manipulate the local datasets or model updates to corrupt the global model. Such attacks have to be countered when federated learning is adopted in IoV systems, given that the training process is distributed among a large number of vehicles in an open environment. In addition, IoV systems present a hierarchical architecture in practice where other types of nodes sit between the cloud server and vehicles, allowing intermediate aggregation for reducing overall training latency. Yet the intermediate aggregation nodes may also pose threats. In this paper, we propose a robust hierarchical federated learning framework named RoHFL, which allows hierarchical federated learning to be suitably applied in the IoV with robustness against poisoning attacks. We develop a robust model aggregation scheme that contains a logarithm-based normalization mechanism to cope with scaled gradients from malicious vehicles. We integrate the notion of reputation into the aggregation process and develop a scheme for reputation updating. We provide a formal analysis of RoHFL's convergence guarantees. Experiment results over several popular datasets demonstrate the promising performance of RoHFL, which is superior to prior work in the robustness against poisoning attacks. © 2023 IEEE.
Research Area(s)
- Federated learning, Training, Servers, Internet of Vehicles, Convergence, Computational modeling, hierarchical federated learning, poisoning attacks, robustness, MODEL AGGREGATION, EDGE
Citation Format(s)
Toward Robust Hierarchical Federated Learning in Internet of Vehicles. / Zhou, Hongliang; Zheng, Yifeng; Huang, Hejiao et al.
In: IEEE Transactions on Intelligent Transportation Systems, Vol. 24, No. 5, 05.2023, p. 5600-5614.
In: IEEE Transactions on Intelligent Transportation Systems, Vol. 24, No. 5, 05.2023, p. 5600-5614.
Research output: Journal Publications and Reviews › RGC 21 - Publication in refereed journal › peer-review