arXiv:2512.23071stat.MLcs.DC2025-12被引 1

通过概率门控实现稀疏联邦学习,提升通信效率与模型泛化能力。

Federated Learning With L0 Constraint Via Probabilistic Gates For Sparsity

  • 用概率门控和连续松弛实现参数稀疏性约束,优化模型密度。
  • 在数据与客户端异构下,目标稀疏度低至0.005仍保持良好性能。
  • 适合资源受限场景下的联邦学习,尤其关注通信效率与模型压缩。

联邦学习(FL)是一种分布式机器学习范式,允许多个客户端在保护数据隐私的前提下协作训练模型。然而,数据与模型中固有的稀疏性未被充分利用,常导致模型过于密集,且在数据及客户端参与异构条件下泛化能力差。本文提出一种基于概率门控及其连续松弛的L0稀疏约束方法,将原用于集中式学习的稀疏机制引入联邦学习。我们证明,带有L0约束的随机最小化目标可自然源于随机门控的熵最大化问题,并设计了基于联邦随机梯度下降的算法。实验表明,在线性与非线性模型(如线性回归、逻辑回归、Softmax多分类、多标签分类、卷积神经网络)中,即使在数据与客户端异构条件下,也能实现目标稀疏度ρ = 0.05以下的稀疏结构,且统计性能损失极小。在合成数据及公开数据集RCV1、MNIST、EMNIST上,ρ低至0.005时,本方法在通信效率和统计性能上均优于基于幅度剪枝的阈值法。

原文摘要 · Abstract (English)

Federated Learning (FL) is a distributed machine learning setting that requires multiple clients to collaborate on training a model while maintaining data privacy. The unaddressed inherent sparsity in data and models often results in overly dense models and poor generalizability under data and client participation heterogeneity. We propose FL with an L0 constraint on the density of non-zero parameters, achieved through a reparameterization using probabilistic gates and their continuous relaxation: originally proposed for sparsity in centralized machine learning. We show that the objective for L0 constrained stochastic minimization naturally arises from an entropy maximization problem of the stochastic gates and propose an algorithm based on federated stochastic gradient descent for distributed learning. We demonstrate that the target density (rho) of parameters can be achieved in FL, under data and client participation heterogeneity, with minimal loss in statistical performance for linear and non-linear models: Linear regression (LR), Logistic regression (LG), Softmax multi-class classification (MC), Multi-label classification with logistic units (MLC), Convolution Neural Network (CNN) for multi-class classification (MC). We compare the results with a magnitude pruning-based thresholding algorithm for sparsity in FL. Experiments on synthetic data with target density down to rho = 0.05 and publicly available RCV1, MNIST, and EMNIST datasets with target density down to rho = 0.005 demonstrate that our approach is communication-efficient and consistently better in statistical performance.

联邦学习稀疏性概率门控通信效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。