AI in EE

AI IN DIVISIONS

AI in Communication Division

Fast-Convergent Federated Learning via Cyclic Aggregation, 2023 IEEE International Conference on Image Processing (ICIP) (강준혁 교수 연구실)

Title: Fast-Convergent Federated Learning via Cyclic Aggregation

Venue: 2023 IEEE International Conference on Image Processing (ICIP)

Abstract: Federated learning (FL) aims at optimizing a shared global model over multiple edge devices without transmitting (private) data to the central server. While it is theoretically well-known that FL yields an optimal model – centrally trained model assuming availability of all the edge device data at the central server – under mild condition, in practice, it often requires massive amount of iterations until convergence, especially under presence of statistical/computational heterogeneity. This paper utilizes cyclic learning rate at the server side to reduce the number of training iterations with increased performance without any additional computational costs for both the server and the edge devices. Numerical results validate that, simply plugging-in the proposed cyclic aggregation to the existing FL algorithms effectively reduces the number of training iterations with improved performance.

Main Figure:

강준혁교수님1