Advancing Federated Learning: A Comprehensive Solution for Model Aggregation, Heterogeneity, Privacy, and Security

被引:0
作者
Bifta Sama Bari [1 ]
Kumar Yelamarthi [1 ]
机构
[1] Department of Electrical and Computer Engineering, College of Engineering, Tennessee Tech University, Cookeville, 38505, TN
关键词
Centralized machine learning; Convolutional neural networks; Edge computing; Federated learning; Vehicular communication;
D O I
10.1007/s42979-025-03934-1
中图分类号
学科分类号
摘要
Secure vehicular communication is one of the challenges that is crucial for Intelligent Transportation Systems (ITS). Although Federated Learning (FL) enhances data privacy compared to Centralized Learning (CL), it faces challenges with data heterogeneity and resource allocation. To address these challenges, the Hybrid Federated-Centralized Learning (HFCL) framework improves efficiency and reduces communication overhead but still lacks comprehensive solutions to existing key issues including FL model aggregation, client heterogeneity, privacy-preserving communication, and enhanced security protocols. This paper proposes a unified HFCL framework incorporating the existing key issues that aims to develop and analyze model aggregation ensuring equitable client contributions, secure communications, and protect against various security threats. The implementation with FedAvg aggregation algorithm on diverse clients and datasets ensures robust and efficient communication, while Differential Privacy (DP) in HFCL enhances security and resilience against adversaries including poisoning attacks. Performance analysis reveals that HFCL maintains high accuracy and stability across varying learning rates and client configurations. Notably, the proposed unified privacy preserving HFCL achieves nearly 99.15% accuracy on the MNIST dataset and 68.4% on the CIFAR10 dataset, demonstrating its efficacy in different scenarios. By addressing the existing critical challenges, the proposed framework enhances the scalability, performance, and security of distributed ML systems, with a primary focus on vehicular communication. It highlights the significant advantages of HFCL for data processing and decision-making in applications like collision avoidance and autonomous driving. Additionally, the framework’s applicability could be extended to healthcare and finance, enabling secure and efficient analysis of sensitive data. © The Author(s), under exclusive licence to Springer Nature Singapore Pte Ltd. 2025.
引用
收藏
相关论文
共 62 条
[31]  
Shit R.C., Sharma S., Yelamarthi K., Puthal D., AI-enabled fingerprinting and crowdsource-based vehicle localization for resilient and safe transportation systems, IEEE Trans Intell Transp Syst, 22, 7, pp. 4660-4669, (2021)
[32]  
Feng C., Yang H.H., Wang S., Zhao Z., Quek T.Q., Hybrid learning: when centralized learning meets federated learning in the mobile edge computing systems, IEEE Trans Commun, 71, pp. 7008-7022, (2023)
[33]  
Hijji M., Iqbal R., Kumar Pandey A., Doctor F., Karyotis C., Rajeh W., Et al., 6G connected vehicle framework to support intelligent road maintenance using deep learning data fusion, IEEE Trans Intell Transp Syst, 24, 7, pp. 7726-7735, (2023)
[34]  
Xie C., Koyejo S., Gupta I., Asynchronous Federated Optimization, (2019)
[35]  
Nguyen J., Malik K., Zhan H., Yousefpour A., Rabbat M., Malek M., Huba D., Federated learning with buffered asynchronous aggregation, International conference on artificial intelligence and statistics. PMLR
[36]  
2022, pp. 3581-3607
[37]  
Su N., Li B., How asynchronous can federated learning be?, IEEE/ACM 30th international symposium on quality of service (IWQoS). IEEE
[38]  
, 2022, pp. 1-11, (2022)
[39]  
Bhatti D.M., Nam H., FedCLS: class-aware federated learning in a heterogeneous environment, IEEE Trans Netw Serv Manag, 20, 2, pp. 1517-1528, (2023)
[40]  
Sattler F., Muller K.R., Samek W., Clustered federated learning: model-agnostic distributed multitask optimization under privacy constraints, IEEE Trans Neural Netw Learn Syst, 32, 8, pp. 3710-3722, (2020)