[控场AI]
· 4 分钟阅读· 2,188 字

拜占庭鲁棒联邦学习与用户级差分隐私的最优收敛

拜占庭鲁棒联邦学习与用户级差分隐私的最优收敛

建立拜占庭鲁棒FL在用户级DP下的信息论下界,并提出阶最优的ByzULDP算法。

文章讨论了联邦学习在同时面对拜占庭攻击与用户级差分隐私时的收敛难题:简单地将鲁棒聚合与中心化DP噪声组合,会因非线性聚合放大噪声而无法收敛到稳定点邻域。针对这一空白,研究首次给出了拜占庭鲁棒经验风险最小化在用户级DP和异构客户端下的信息论下界,并提出中心化DP算法ByzULDP,在客户端和服务器两级引入动量,分别缓解随机梯度方差和补偿DP噪声影响。理论证明其在强凸和非凸目标下分别达到O(1/T)和O(1/√T)的阶最优收敛速率,与下界匹配,真实数据集实验也验证了不同隐私预算下的有效性。

联邦学习(FL)在实际部署中既要抵御恶意客户端的拜占庭攻击,又要保护用户数据隐私。拜占庭容错刻画了对抗性客户端的最坏行为,而差分隐私(DP)则提供隐私损失的统计保证。用户级DP保护客户端的整个贡献,能更真实地防止信息泄露。然而,拜占庭鲁棒FL在用户级DP下的极小极大最优收敛界一直悬而未决。近期一项研究为此建立了信息论下界,并提出了达到阶最优收敛的算法ByzULDP。

Optimal Byzantine-resilient Federated Learning with User-level Differential Privacy

核心挑战:隐私与鲁棒性的收敛难题

拜占庭鲁棒聚合与中心化DP噪声的简单组合,在无故障场景下有效,但在对抗性客户端存在时,非线性鲁棒聚合与DP噪声的叠加可能破坏收敛性,使算法无法收敛到稳定点的有界邻域。这一现象源于客户端更新构造方式影响隐私敏感度:用户级DP的敏感度取决于客户端更新如何计算,而鲁棒聚合的非线性操作会放大噪声影响。

现有研究虽在拜占庭鲁棒FL或用户级DP单方面取得进展,但两者的联合优化缺乏理论指导。该研究填补了这一空白,首次给出了拜占庭鲁棒经验风险最小化问题在用户级DP和异构客户端下的信息论下界,为算法设计设定了性能极限。

拜占庭容错与鲁棒聚合 拜占庭攻击指分布式系统中部分节点出现任意恶意行为(如发送错误梯度、篡改模型更新等),拜占庭鲁棒FL的目标是在至多f个恶意客户端存在时仍能正确聚合。常见的鲁棒聚合规则包括坐标中位数(Coordinate-wise Median)、修剪均值(Trimmed Mean)、Krum等,其本质是对客户端更新做非线性筛选或截断,以削弱异常值的杠杆作用。然而,这种非线性操作在压制大偏差更新的同时,也会对真实梯度分布产生一定扰动,与DP噪声叠加后可能导致聚合偏差累积,这是朴素组合难以收敛的根源。

差分隐私与用户级隐私敏感度 差分隐私通过向计算过程注入随机噪声,使相邻数据集的输出分布难以区分,从而限制单条记录的可推断信息。在FL中,中心化DP通常由服务器统一加噪,其隐私预算由全局敏感度决定;用户级DP将每个客户端的整个本地数据集视为一条“记录”,敏感度随客户端更新范数(常见做法是裁剪后再加噪)而变化。因此,客户端更新的构造方式——如是否裁剪、裁剪阈值大小——直接影响所需噪声尺度,这也是文中强调“敏感度与更新构造相关”的原因。

极小极大下界与阶最优性 极小极大下界刻画了在问题类(如强凸/非凸、给定拜占庭比例和隐私预算)上,任何算法在最坏情形下所能达到的最优收敛误差量级。达到该下界的算法称为阶最优(order-optimal)。这类下界通常借助信息论工具(如Fano不等式、Assouad引理)构造难以区分的分布对,从而给出算法必须承受的误差下限。

ByzULDP:双层动量驱动收敛

为突破收敛瓶颈,研究者提出ByzULDP——一种中心化DP算法,在客户端和服务器两级引入动量。客户端动量在鲁棒聚合过程中缓解随机梯度方差,提升聚合稳定性;服务器动量则利用用户级DP的独特性质——隐私敏感度与客户端更新构造相关——来赋能收敛。这种双层动量设计并非简单叠加,而是针对DP噪声与鲁棒聚合的相互作用进行补偿,从而克服朴素组合的不足。

理论分析表明,ByzULDP在强凸目标下达到阶最优收敛界,误差项以最优速率消失:强凸目标为O(1/T),非凸目标为O(1/√T)。这一结果与信息论下界匹配,证实了算法的极小极大最优性。

双层动量与方差缩减 动量方法通过在历史更新上做指数加权平均来降低随机梯度噪声带来的方差。在拜占庭鲁棒FL中,客户端动量在本地维护,可使被聚合的更新更平滑,避免单轮噪声主导鲁棒筛选结果;服务器动量则在聚合后对全局更新做平滑,以补偿中心化DP噪声引入的额外扰动。二者分别作用于“鲁棒聚合之前”和“加噪聚合之后”两个环节,形成对噪声与非线性聚合误差的联合补偿,这也是该设计区别于简单堆叠动量的关键。

收敛速率含义 文中O(1/T)和O(1/√T)分别描述强凸与非凸目标下误差随迭代轮数T的衰减量级:强凸情形下误差以线性速率倒数衰减,非凸情形下则对应梯度范数平方的经典随机优化速率。与下界匹配意味着在给定隐私预算和拜占庭比例下,算法在常数因子内已无法进一步改进,这为实际调参提供了理论参照。

实验验证与广泛适用性

研究在真实数据集上进行了数值实验,覆盖多种隐私预算。实验结果与理论分析一致,验证了ByzULDP在不同隐私保护强度下的有效性和鲁棒性。该工作为联邦学习在隐私敏感场景中的安全部署提供了理论基石和实用算法,推动了隐私保护与鲁棒性协同优化的研究。

关键要点

  • 首次建立拜占庭鲁棒FL在用户级DP下的信息论下界,揭示收敛极限。
  • 提出ByzULDP算法,通过客户端和服务器双层动量解决DP噪声与鲁棒聚合的冲突。
  • 理论证明ByzULDP达到阶最优收敛速率,强凸和非凸目标分别实现O(1/T)和O(1/√T)的误差衰减。
  • 实验验证算法在多种隐私预算下的有效性,为实际部署提供参考。
#拜占庭鲁棒联邦学习#用户级差分隐私#ByzULDP#收敛界#联邦学习隐私#鲁棒聚合
分享:

相关推荐