自然梯度下降在优化KL散度时,比普通梯度下降更稳定且收敛更快。
Convergence Properties of Natural Gradient Descent for Minimizing KL Divergence
- 用信息几何中的θ和η坐标系分析梯度下降行为
- 连续时间下自然梯度收敛率恒为2,优于普通梯度的变率
- 离散时间中自然梯度更抗噪声,适合实际训练
KL散度在概率机器学习中作为标准损失函数广泛应用,优化常在概率单纯形上进行,参数化方式显著影响收敛性。本文研究最小化KL散度问题,分析在信息几何框架下的两种对偶坐标系——指数族(θ坐标)与混合族(η坐标)中梯度算法的行为。比较欧氏梯度下降(GD)与坐标不变的自然梯度下降(NGD)。在连续时间下,θ坐标中GD的收敛率提供下界,η坐标中提供上界,而NGD收敛率为固定值2;在仿射重参数化下,GD在η和θ坐标中的收敛率可分别缩放为2c和2/c(c>0),而NGD保持不变。尽管连续时间下无统一优势,但在离散时间中,NGD展现出更快收敛与更强抗噪性,表现更优。分析基于最优解处KL散度的海森矩阵谱与条件数,其与费舍尔信息矩阵一致。
原文摘要 · Abstract (English)
The Kullback-Leibler (KL) divergence plays a central role in probabilistic machine learning, where it commonly serves as the canonical loss function. Optimization in such settings is often performed over the probability simplex, where the choice of parameterization significantly impacts convergence. In this work, we study the problem of minimizing the KL divergence and analyze the behavior of gradient-based optimization algorithms under two dual coordinate systems within the framework of information geometry$-$ the exponential family ($θ$ coordinates) and the mixture family ($η$ coordinates). We compare Euclidean gradient descent (GD) in these coordinates with the coordinate-invariant natural gradient descent (NGD), where the natural gradient is a Riemannian gradient that incorporates the intrinsic geometry of the underlying statistical model. In continuous time, we prove that the convergence rates of GD in the $θ$ and $η$ coordinates provide lower and upper bounds, respectively, on the convergence rate of NGD. Moreover, under affine reparameterizations of the dual coordinates, the convergence rates of GD in $η$ and $θ$ coordinates can be scaled to $2c$ and $\frac{2}{c}$, respectively, for any $c>0$, while NGD maintains a fixed convergence rate of $2$, remaining invariant to such transformations and sandwiched between them. Although this suggests that NGD may not exhibit uniformly superior convergence in continuous time, we demonstrate that its advantages become pronounced in discrete time, where it achieves faster convergence and greater robustness to noise, outperforming GD. Our analysis hinges on bounding the spectrum and condition number of the Hessian of the KL divergence at the optimum, which coincides with the Fisher information matrix.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。