Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SpecGradFilter: A Spectral Gradient Filtering Framework for Taming Federated Heterogeneity

About

Federated Learning (FL) is fundamentally challenged by statistical heterogeneity, where non-identically distributed (non-IID) data induces client drift that severely hampers global convergence. While existing approaches attempt to mitigate this drift through spatial-domain gradient correction or regularization, they overlook the intrinsic spectral structure of optimization signals. In this work, we revisit client drift from a novel frequency-domain perspective and uncover a critical Spectral Bias of Drift: inter-client gradient divergence is predominantly concentrated in low-frequency components which encode client-specific distributional shifts, while high-frequency components representing fine-grained features remain relatively consistent. Motivated by this, we propose SpecGradFilter, a unified Spectral Gradient Filtering Framework that tames heterogeneity by suppressing discordant low-frequency signals. Crucially, we demonstrate that SpecGradFilter is a generalizable principle, effective not only via precise FFT-based truncation but also through spatial approximations like Gaussian detrending. Extensive experiments on benchmarks such as CIFAR-10/100 and Tiny-ImageNet demonstrate that SpecGradFilter significantly performs better performance in highly Non-IID settings with negligible communication overhead, establishing a new paradigm for robust federated optimization.

Liyang Yuan, Yibo Yang, Dandan Guo, Peter Richtarik, Zhouchen Lin• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationTiny ImageNet (test)
Accuracy38.29
859
Image ClassificationTiny-ImageNet Dirichlet alpha=0.1 (test)
Test Accuracy31.55
40
Image ClassificationCIFAR-100-C CS-3 1.0 (test)
Mean Accuracy37.04
17
Image ClassificationTiny ImageNet Dirichlet α=0.6 (test)
Accuracy36.73
16
Image ClassificationCIFAR-100 (test)
Test Accuracy49.36
10
Image ClassificationCIFAR-10 Dirichlet alpha=0.6 (test)
Test Accuracy85.11
10
Image ClassificationCIFAR-10 Dirichlet alpha=0.1 (test)
Test Accuracy78.05
10
Image ClassificationCIFAR-100 Dirichlet alpha=0.6 (test)
Test Accuracy45.99
10
Image ClassificationCIFAR-100 Dirichlet alpha=0.1 (test)
Test Accuracy40.16
10
Medical Image ClassificationBloodMNIST (test)
Accuracy (alpha=0.6)94.15
6
Showing 10 of 11 rows

Other info

Follow for update