通过调控梯度统计量提升联邦学习模型的全局迁移能力。
FedGTST: Boosting Global Transferability of Federated Models via Statistics Tuning
- 用跨客户端梯度范数交互优化全局迁移性
- 提升平均梯度范数并降低其方差,使目标损失更可控
- 适合追求高迁移性能的联邦学习研究者
迁移学习(TL)性能高度依赖于有效的预训练,但需大规模数据集和大量计算资源,对单个开发者而言难以实现。联邦学习(FL)通过客户端协作,间接扩展数据规模、分摊计算成本并保护隐私,缓解了这一问题。然而仍存在两大挑战:现有方法仅优化本地域内的迁移性,忽视全局学习域;多数方法依赖间接迁移性指标,无法准确反映最终目标损失或真实迁移程度。为此,本文提出两项改进:一是设计客户端-服务器交互协议,利用跨客户端雅可比(梯度)范数提升迁移性;二是在服务器端增大各客户端平均雅可比范数,并以此作为局部正则项以减小跨客户端雅可比范数方差。所提算法称为 FedGTST(Federated Global Transferability via Statistics Tuning),证明提升平均雅可比范数并降低其方差可更紧地控制目标损失,得到目标损失的上界表达式,该上界与源损失及源-目标域差异相关。在 MNIST 到 MNIST-M 与 CIFAR10 到 SVHN 等数据集上的实验表明,FedGTST 显著优于基线方法,包括 FedSR。当使用 LeNet 作为骨干网络时,在第二组数据上相比 FedSR 提升 9.8% 准确率,相比 FedIIR 提升 7.6%。
原文摘要 · Abstract (English)
The performance of Transfer Learning (TL) heavily relies on effective pretraining, which demands large datasets and substantial computational resources. As a result, executing TL is often challenging for individual model developers. Federated Learning (FL) addresses these issues by facilitating collaborations among clients, expanding the dataset indirectly, distributing computational costs, and preserving privacy. However, key challenges remain unresolved. First, existing FL methods tend to optimize transferability only within local domains, neglecting the global learning domain. Second, most approaches rely on indirect transferability metrics, which do not accurately reflect the final target loss or true degree of transferability. To address these gaps, we propose two enhancements to FL. First, we introduce a client-server exchange protocol that leverages cross-client Jacobian (gradient) norms to boost transferability. Second, we increase the average Jacobian norm across clients at the server, using this as a local regularizer to reduce cross-client Jacobian variance. Our transferable federated algorithm, termed FedGTST (Federated Global Transferability via Statistics Tuning), demonstrates that increasing the average Jacobian and reducing its variance allows for tighter control of the target loss. This leads to an upper bound on the target loss in terms of the source loss and source-target domain discrepancy. Extensive experiments on datasets such as MNIST to MNIST-M and CIFAR10 to SVHN show that FedGTST outperforms relevant baselines, including FedSR. On the second dataset pair, FedGTST improves accuracy by 9.8% over FedSR and 7.6% over FedIIR when LeNet is used as the backbone.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。