Search papers, labs, and topics across Lattice.
The paper introduces FedPSA, a novel Asynchronous Federated Learning (AFL) framework that addresses the performance degradation caused by staleness in asynchronous training. FedPSA uses parameter sensitivity to provide a fine-grained measure of model obsolescence, moving beyond simple round difference calculations. By establishing a dynamic momentum queue to assess the training phase, FedPSA dynamically adjusts tolerance for outdated information, leading to improved performance.
Ditch round-based staleness metrics in asynchronous federated learning: parameter sensitivity reveals a more nuanced picture, boosting performance by up to 6.37%.
Asynchronous Federated Learning (AFL) has emerged as a significant research area in recent years. By not waiting for slower clients and executing the training process concurrently, it achieves faster training speed compared to traditional federated learning. However, due to the staleness introduced by the asynchronous process, its performance may degrade in some scenarios. Existing methods often use the round difference between the current model and the global model as the sole measure of staleness, which is coarse-grained and lacks observation of the model itself, thereby limiting the performance ceiling of asynchronous methods. In this paper, we propose FedPSA (Parameter Sensitivity-based Asynchronous Federated Learning), a more fine-grained AFL framework that leverages parameter sensitivity to measure model obsolescence and establishes a dynamic momentum queue to assess the current training phase in real time, thereby adjusting the tolerance for outdated information dynamically. Extensive experiments on multiple datasets and comparisons with various methods demonstrate the superior performance of FedPSA, achieving up to 6.37\% improvement over baseline methods and 1.93\% over the current state-of-the-art method.
