Heterogeneous fairness algorithm based on federated learning in intelligent transportation system

Abstract

With the development of the Intelligent Transportation System, various distributed sensors (including GPS, radar, infrared sensors) process massive data and make decisions for emergencies. Federated learning is a new distributed machine learning paradigm, in which system heterogeneity is the difficulty of fairness design. This paper designs a system heterogeneous fair federated learning algorithm (SHFF). SHFF introduces the equipment influence factor I into the optimization target and dynamically adjusts the equipment proportion with other performance. By changing the global fairness parameter θ, the algorithm can control fairness according to the actual needs. Experimental results show that, compared with the popular q-FedAvg algorithm, the SHFF algorithm proposed in this paper improves the average accuracy of the Worst 10% by 26% and reduces the variance by 61%.