atlas

Research map / Federated learning and privacy / Communication-efficient FL

Efficient Asynchronous Federated Learning with Prospective Momentum Aggregation and Fine-Grained Correction

AAAI 2024 · Federated learning and privacy › Communication-efficient FL · approach: Local drift correction · 31 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes FedAC, an asynchronous federated learning framework combining temporal-gradient client weighting, Nadam-style prospective momentum at the server, and fine-grained gradient correction at clients, evaluated on CIFAR-10, EMNIST-L and Shakespeare.

Problem

Asynchronous federated learning is inefficient because data heterogeneity and device heterogeneity cause poor client training quality, stale updates, and slow server convergence.

Approach

Weights each client by the cosine similarity between its pseudo-gradient and the global pseudo-gradient over the staleness interval, applies a Nadam-style momentum with adaptive learning rate on buffered asynchronous aggregation, and corrects every local SGD step with a SCAFFOLD-like global-minus-local control term computed from the client weights.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use FedAC instead of FedAvgCIFAR-10 · CIFAR-10, α=0.1accuracy: 66.8 -> 70.6Helped
Use FedAC instead of FedAvgCIFAR-10 · CIFAR-10, α=1accuracy: 76.1 -> 79.6Helped
Use FedAC instead of FedAdamCIFAR-10 · CIFAR-10, α=10accuracy: 78.8 -> 80.1Helped
Use FedAC instead of FedAdamCIFAR-10 · CIFAR-10, IIDaccuracy: 79.1 -> 80.3Helped
Add client weight evaluation based on temporal gradient (ablation)FedBuff-based framework · CIFAR-10accuracyHelped
Add adaptive server update with prospective weighted momentum (ablation)Basic-WTG · CIFAR-10convergence speedHelped
Add adaptive server update with prospective weighted momentum without fine-grained client correction (ablation)Basic-WTG · CIFAR-10accuracyHurt
Add fine-grained gradient correction during client updates (ablation)FedAC · CIFAR-10accuracyHelped

Limitations

Experimental setup

Datasets: CIFAR-10 EMNIST-L Shakespeare

Benchmarks: CIFAR-10 EMNIST-L Shakespeare

Metrics: accuracy number of client trips to target accuracy accuracy at fixed client trips

Baselines: FedAvg FedAdam FedProx FedAsync FedBuff

Builds on: FedBuff FedAdam SCAFFOLD FedAsync Nadam

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy