揭示联邦学习聚合导致性能下降的根本原因。
The Other Side of the Coin: Unveiling the Downsides of Model Aggregation in Federated Learning from a Layer-peeled Perspective
- 通过分层分析模型聚合对特征提取的影响。
- 发现特征质量随网络深度累积下降,影响最终分类效果。
- 解释现有方法为何有效,适合研究联邦学习优化者。
联邦学习中,聚合模型在本地数据上的表现常在初期下降,直到经过多轮本地训练才恢复。以往研究将其视为知识共享的固有代价,未予深入关注。本文构建分层分析框架,考察模型聚合过程中特征表示的演化。聚焦两点:(1)特征提取的质量,(2)特征与后续参数之间的对齐程度。分析表明,聚合会削弱特征质量,并破坏中间特征与后续层的耦合关系,这些关系在本地训练中原本被良好塑造。更重要的是,这种退化并非局限于特定层,而是随网络深度逐步累积,我们称之为累积特征退化(CFD)。CFD显著降低倒数第二层特征质量,并削弱其与分类器的关联,从而损害模型性能。我们进一步从分层视角审视多种常用解决方案,发现其有效性源于缓解上述特征退化,与我们的观察高度一致。
原文摘要 · Abstract (English)
It is often observed that the aggregated model in FL underperforms on local data until after several rounds of local training. This temporary performance drop can potentially slow down the convergence of the FL model. Prior work regards this performance drop as an inherent cost of knowledge sharing among clients and does not give it special attention. While some studies directly focus on designing techniques to alleviate the issue, its root causes remain poorly understood. To bridge this gap, we construct a framework that enables layer-peeled analysis of how feature representations evolve during model aggregation in FL. It focuses on two key aspects: (1) the intrinsic quality of extracted features, and (2) the alignment between features and their subsequent parameters -- both of which are critical to downstream performance. Using this framework, we first investigate how model aggregation affects internal feature extraction process. Our analysis reveals that aggregation degrades feature quality and weakens the coupling between intermediate features and subsequent layers, both of which are well shaped during local training. More importantly, this degradation is not confined to specific layers but progressively accumulates with network depth -- a phenomenon we term Cumulative Feature Degradation (CFD). CFD significantly impairs the quality of penultimate-layer features and weakens their coupling with the classifier, ultimately degrading model performance. We further revisit several widely adopted solutions through the lens of layer-peeled feature extraction to understand why they are effective in addressing aggregation-induced performance drop. Our results show that their effectiveness lies in mitigating the feature degradation described above, which is well aligned with our observations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。