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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | Tiny ImageNet (test) | Accuracy38.29 | 859 | |
| Image Classification | Tiny-ImageNet Dirichlet alpha=0.1 (test) | Test Accuracy31.55 | 40 | |
| Image Classification | CIFAR-100-C CS-3 1.0 (test) | Mean Accuracy37.04 | 17 | |
| Image Classification | Tiny ImageNet Dirichlet α=0.6 (test) | Accuracy36.73 | 16 | |
| Image Classification | CIFAR-100 (test) | Test Accuracy49.36 | 10 | |
| Image Classification | CIFAR-10 Dirichlet alpha=0.6 (test) | Test Accuracy85.11 | 10 | |
| Image Classification | CIFAR-10 Dirichlet alpha=0.1 (test) | Test Accuracy78.05 | 10 | |
| Image Classification | CIFAR-100 Dirichlet alpha=0.6 (test) | Test Accuracy45.99 | 10 | |
| Image Classification | CIFAR-100 Dirichlet alpha=0.1 (test) | Test Accuracy40.16 | 10 | |
| Medical Image Classification | BloodMNIST (test) | Accuracy (alpha=0.6)94.15 | 6 |