提出深度解析联邦学习框架,解决模型异构难题。
DeepAFL: Deep Analytic Federated Learning
- 设计无梯度残差模块,实现深层解析模型训练
- 在三个数据集上性能领先基线5.68%~8.42%
- 适合需要抗数据异构的工业级联邦学习场景
联邦学习(FL)是一种打破数据孤岛的分布式学习范式。传统方法主要依赖梯度更新,面临异构性、可扩展性、收敛性和开销等问题。近期一些基于解析学习的工作通过闭式解消除梯度更新,提升了对数据异构的鲁棒性,但受限于单层线性模型和冻结预训练主干网络,缺乏表征学习能力,性能受限。本文提出深度解析联邦学习(DeepAFL),受ResNet启发,设计具有解析解的无梯度残差块,采用高效的逐层最小二乘协议,在联邦环境中分层训练深层解析模型。理论分析与实证评估表明,DeepAFL兼具异构性不变性与表征学习能力,相较于最先进基线,在三个基准数据集上性能提升达5.68%~8.42%。
原文摘要 · Abstract (English)
Federated Learning (FL) is a popular distributed learning paradigm to break down data silo. Traditional FL approaches largely rely on gradient-based updates, facing significant issues about heterogeneity, scalability, convergence, and overhead, etc. Recently, some analytic-learning-based work has attempted to handle these issues by eliminating gradient-based updates via analytical (i.e., closed-form) solutions. Despite achieving superior invariance to data heterogeneity, these approaches are fundamentally limited by their single-layer linear model with a frozen pre-trained backbone. As a result, they can only achieve suboptimal performance due to their lack of representation learning capabilities. In this paper, to enable representable analytic models while preserving the ideal invariance to data heterogeneity for FL, we propose our Deep Analytic Federated Learning approach, named DeepAFL. Drawing inspiration from the great success of ResNet in gradient-based learning, we design gradient-free residual blocks in our DeepAFL with analytical solutions. We introduce an efficient layer-wise protocol for training our deep analytic models layer by layer in FL through least squares. Both theoretical analyses and empirical evaluations validate our DeepAFL's superior performance with its dual advantages in heterogeneity invariance and representation learning, outperforming state-of-the-art baselines by up to 5.68%-8.42% across three benchmark datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。