arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.18656stat.MLcs.AIcs.LGstat.ME

统计界限与差分隐私联邦学习的高效算法

Statistical Limits and Efficient Algorithms for Differentially Private Federated Learning

Department of Statistics(统计系)

Arnab Auddy, Xiangni Peng, Subhadeep Paul

AI总结:

本文研究了差分隐私联邦学习中估计精度、隐私约束和通信成本之间的权衡,提出了FedHybrid和FedNewton两种高效算法,通过减少通信成本提升准确性,并建立了均方误差的上界和下界以评估算法性能。

摘要:

联邦学习是训练机器学习和人工智能模型的一种主流框架,用于在众多用户设备或数据库之间协同训练。我们研究了差分隐私(DP)联邦M估计中估计精度、隐私约束和通信成本之间的权衡。文献中的两种标准方法是FedAvg,可能面临较高的联邦偏差,以及FedSGD,可能导致较高的通信成本。为了在减少通信成本的同时提高准确性,我们提出了FedHybrid,它使用FedSGD,但起始时通过FedAvg估计器改进初始化。我们还提出了FedNewton,通过平均本地牛顿迭代来减少FedAvg的偏差,从而在客户端数量增长缓慢时,以更少的通信轮次达到与FedSGD相当的估计精度。我们建立了这些估计器的DP版本的均方误差率的有限样本上界,作为客户端数量、本地样本大小、隐私预算和迭代次数的函数。我们进一步推导了任何迭代私有联邦过程的均方误差的最小最大下界,以作为评估这些方法最优性差距的基准。我们还通过在MNIST和CIFAR-10计算机视觉数据集上训练逻辑回归和神经网络来数值评估我们的方法。

英文摘要:

Federated Learning is a leading framework for training ML and AI models collaboratively across numerous user devices or databases. We study the trade-offs among estimation accuracy, privacy constraints, and communication cost for differentially private (DP) federated M estimation. The two standard methods in the literature are FedAvg, which may suffer from high federation bias, and FedSGD, which can incur high communication cost. Aimed at improving accuracy at a reduced communication cost, we propose FedHybrid, which uses FedSGD starting with an improved initialization by the FedAvg estimator. We propose FedNewton, which averages local Newton iterations to reduce bias in FedAvg, achieving an estimation accuracy comparable to FedSGD with much fewer communication rounds when the number of clients grows sufficiently slowly. We establish finite sample upper bounds on the mean-squared error rates of the DP versions of these estimators as functions of the number of clients, local sample sizes, privacy budget, and number of iterations. We further derive a minimax lower bound on the MSE of any iterative private federated procedure that provides a benchmark to assess the optimality gap of these methods. We numerically evaluate our methods for training a logistic regression and a neural network on the computer vision datasets MNIST and CIFAR-10.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑