通过频域滤波解决联邦学习中局部优化导致的客户端漂移问题。
FedFFT: Taming Client Drift in Federated SAM via Spectral Perturbation Filtering
- 在频域过滤SAM扰动的低频成分,抑制客户端差异。
- 无需额外通信开销,显著提升非独立同分布场景下的性能。
- 适配多种模型与数据集,对联邦学习研究者有实用价值。
联邦学习(FL)可在不共享数据的前提下实现分布式训练,但客户端间的统计异质性会导致客户端漂移、泛化能力下降及尖锐极小值,相较集中式训练表现更差。尖锐感知最小化(SAM)虽能提升泛化性能,但在联邦设置下仍易发散,因扰动由各客户端本地计算,反映的是特定客户端的损失几何结构。我们从频域新视角揭示:联邦环境下SAM扰动的跨客户端不一致性主要集中在低频谱。基于此洞察,提出联邦学习中基于频域滤波的SAM扰动方法(FedFFT)。该方法轻量且可即插即用,无需额外通信,通过去除SAM扰动的低频成分,抑制更新中的不一致信号,同时保留一致的学习信息。多基准测试与多样化骨干网络的实验表明,FedFFT在严重非独立同分布条件下持续优于现有基于SAM的联邦学习方法,验证了其有效性、可扩展性与普适性。
原文摘要 · Abstract (English)
Federated Learning (FL) enables decentralized training without data sharing, but suffers from statistical heterogeneity across clients, leading to client drift, poor generalization, and sharp minima compared to centralized training. Sharpness-Aware Minimization (SAM) has emerged as a promising approach to improve generalization, yet its application in federated learning still suffers from divergence problems, since perturbations are computed locally and reflect client-specific loss geometries. To better understand this issue, we provide experimental evidence from a new perspective, the frequency domain, for SAM perturbations in federated settings, revealing that inter-client perturbation inconsistencies are predominantly concentrated in the low-frequency spectrum. Motivated by this insight, we propose Federated learning with Frequency-domain Filtering of SAM perturbations (FedFFT). It is a lightweight and plug-and-play method that filters out low-frequency components of SAM perturbations without requiring additional communication, thereby suppressing inconsistent components in client updates while preserving consistent learning signals. Extensive experiments across multiple benchmarks and diverse backbones demonstrate that FedFFT consistently outperforms SAM-based FL methods, particularly under severe non-IID distributions. These results highlight the effectiveness, scalability, and general applicability of our frequency-domain perspective for sharpness-aware federated optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。