An Efficient Multi-Dimensional Adaptive Federated Learning Algorithm

With the rapid development of the Internet of Vehicles (IoV),Nevertheless, Federated Learning (FL) plays an increasingly crucial role in the secure data sharing within this field. Nevertheless, data in IoV settings frequently display Non-IID traits, which exerts a considerable influence on the efficiency and performance of FL. To enhance the model convergence speed and global performance under conditions of data imbalance and high heterogeneity, this paper presents a FL algorithm, named FedAE, that dynamically adjusts the local training epochs based on the changes in the AUC of model. FedAE assigns distinct local training epochs based on the quantity of training data in the initial stage and adaptively modifies them in subsequent stages in accordance with the dynamic changes in the AUC of both the global model and each client's model. Based on the performance disparity of the local training equipment, the algorithm establishes diverse training iterations to enhance the utilization of the local equipment and the training efficiency of the algorithm. Experimental results show that, compared to FedProx, the proposed approach significantly enhances model convergence efficiency and generalization ability. Specifically, FedAE reduces the loss by 26 %, demonstrating greater robustness, especially in scenarios with imbalanced data distributions.