arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.20567cs.LGmath.OCstat.ML

基于定向图的推-和基于去中心化优化的稳定性与泛化

Stability and Generalization of Push-Sum Based Decentralized Optimization over Directed Graphs

School of Cyber Science and Technology Sun Yat-sen University(南方科技大学)

Yifei Liang, Yan Sun, Xiaochun Cao, Li Shen

AI总结:

本文提出了一种统一的稳定性框架,用于分析基于推-和的去中心化优化算法在定向图上的稳定性与泛化性能,揭示了问题条件与拓扑结构之间的耦合关系。

摘要:

推-和基于去中心化学习使优化在定向通信网络上成为可能,其中信息交换可能是不对称的。尽管此类方法的收敛性质已得到充分理解,但其有限迭代稳定性及泛化行为仍不清楚,因为列随机混合和不对称误差传播引起的结构性偏差。在本文中,我们为随机梯度推(SGP)算法开发了一个统一的统一稳定性框架,该框架捕捉了定向拓扑的影响。一个关键技术成分是平衡意识的一致性界,用于推-和,它通过两个量控制共识偏差:静态分布不平衡参数δ和支配混合速度的谱间隙(1-λ)。这种分解使我们能够将统计效应与拓扑诱导的偏差分开。我们为凸目标函数和满足Polyak-Łojasiewicz条件的非凸目标函数建立了有限迭代稳定性及优化保证。对于凸问题,SGP在步长调度下达到的超额泛化误差顺序为~O(1/√(mn)+γ/(δ(1-λ)) + γ),并我们表征了相应的最优提前停止时间,该时间最小化此界。对于PŁ目标函数,我们获得了类似凸优化和泛化速率,其主导依赖于κ(1+1/(δ(1-λ))),揭示了问题条件与定向通信拓扑之间存在乘法耦合。我们的分析澄清了Push-Sum修正与标准去中心化SGD相比何时是必要的,并量化了不平衡和混合如何共同塑造最佳可达到的学习性能。

英文摘要:

Push-Sum-based decentralized learning enables optimization over directed communication networks, where information exchange may be asymmetric. While convergence properties of such methods are well understood, their finite-iteration stability and generalization behavior remain unclear due to structural bias induced by column-stochastic mixing and asymmetric error propagation. In this work, we develop a unified uniform-stability framework for the Stochastic Gradient Push (SGP) algorithm that captures the effect of directed topology. A key technical ingredient is an imbalance-aware consistency bound for Push-Sum, which controls consensus deviation through two quantities: the stationary distribution imbalance parameter $δ$ and the spectral gap $(1-λ)$ governing mixing speed. This decomposition enables us to disentangle statistical effects from topology-induced bias. We establish finite-iteration stability and optimization guarantees for both convex objectives and non-convex objectives satisfying the Polyak--Łojasiewicz condition. For convex problems, SGP attains excess generalization error of order $\tilde{\mathcal{O}}\!\left(\frac{1}{\sqrt{mn}}+\fracγ{δ(1-λ)}+γ\right)$ under step-size schedules, and we characterize the corresponding optimal early stopping time that minimizes this bound. For PŁ objectives, we obtain convex-like optimization and generalization rates with dominant dependence proportional to $κ\!\left(1+\frac{1}{δ(1-λ)}\right)$, revealing a multiplicative coupling between problem conditioning and directed communication topology. Our analysis clarifies when Push-Sum correction is necessary compared with standard decentralized SGD and quantifies how imbalance and mixing jointly shape the best attainable learning performance.

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

↑