Layer-Wise Personalized Federated Learning for DAG Blockchain-Enabled Internet of Vehicles
In the Internet of Vehicles (IoV), personalized federated learning (PFL) can generate personalized models tailored to different local data distributions, thereby improving driving decisions. Existing approaches often perform weighted aggregation on similar vehicles, where the weights are determined by the entire model parameters or loss values. Such model-wise aggre-gation results in inaccurate combination weights, as it overlooks differences in layer-level functionality and imbalances in inter-user accuracy. Additionally, the openness of IoV poses significant threats to system security. In this paper, we propose a blockchain-enabled layer-wise PFL (BLPFL) algorithm to improve system security, model accuracy, and user fairness. Specifically, to preserve personalized knowledge, each local model is split into a globally shared feature extractor and a locally private decision head. To secure the sharing of extractors, Directed Acyclic Graph (DAG)-based blockchain technology is employed. Furthermore, a fine-grained personalized model aggregation mechanism is designed to enhance the model accuracy and user fairness, which explores vehicle similarity at layer granularity and considers special vehicles with severely skewed label distributions. Finally, simulation results demonstrate the effectiveness of the proposed approach in highly heterogeneous scenarios.
Authors
- Chengchao Liang (ORCID: https://orcid.org/0000-0002-4125-2840)
- Xiaoge Huang (ORCID: https://orcid.org/0000-0002-4559-9606)
- Qianbin Chen (ORCID: https://orcid.org/0000-0001-6868-6860)
- Wenjing Li (ORCID: https://orcid.org/0000-0002-8579-4641)
- Caiwei Yu
Institutions
- Chongqing University of Posts and Telecommunications (CN)
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1109/wcsp62071.2024.10827741
- Primary Topic
- Privacy-Preserving Technologies in Data
- Type
- article
- Field-Weighted Citation Impact
- 0.00