Skip to content

第6章:梯度、投影与近端方法

光滑性与下降引理

设 f 可微且梯度 L-Lipschitz,即 f(x)f(y)Lxy\|\nabla f(x)-\nabla f(y)\|\le L\|x-y\|。沿线段积分并界定梯度变化可得

f(y)f(x)+f(x)T(yx)+L2yx2.f(y)\le f(x)+\nabla f(x)^T(y-x)+\frac L2\|y-x\|^2.

y=xηf(x)y=x-\eta\nabla f(x),若 0<η1/L0<\eta\le1/L,则

f(y)f(x)η2f(x)2.f(y)\le f(x)-\frac\eta2\|\nabla f(x)\|^2.

梯度方向有下降倾向不代表任意步长都下降;步长过大可能跨过谷底并发散。

收敛率与条件数

对有最优解的光滑凸函数,步长 1/L1/L 的梯度法有函数值误差 O(1/k)O(1/k) 上界。若再有 μ\mu 强凸性,可获得与 1μ/L1-\mu/L 有关的线性收敛界。

例如二次函数 Hessian 为 diag(1,100)\operatorname{diag}(1,100),L=100、μ=1\mu=1。安全固定步长为 0.01,较平坦方向每步只乘 0.99,解释了条件数大时的缓慢进展。

凸情形的误差界

xx^* 是最优解,取步长 1/L1/L,记 gk=f(xk)g_k=\nabla f(x_k)。展开平方距离可得

xk+1x2=xkx22LgkT(xkx)+1L2gk2.\|x_{k+1}-x^*\|^2 =\|x_k-x^*\|^2-\frac2L g_k^T(x_k-x^*)+\frac1{L^2}\|g_k\|^2.

凸性给出 f(xk)f(x)gkT(xkx)f(x_k)-f(x^*)\le g_k^T(x_k-x^*),下降引理给出 f(xk+1)f(xk)gk2/(2L)f(x_{k+1})\le f(x_k)-\|g_k\|^2/(2L)。把两式代入,梯度范数项恰好消去:

f(xk+1)f(x)L2(xkx2xk+1x2).f(x_{k+1})-f(x^*) \le\frac L2\left(\|x_k-x^*\|^2-\|x_{k+1}-x^*\|^2\right).

从第 0 步累加到第 k1k-1 步,右侧距离项相消。由于函数值单调下降,左侧每项至少是 f(xk)f(x)f(x_k)-f(x^*),于是

f(xk)f(x)Lx0x22k.f(x_k)-f(x^*)\le\frac{L\|x_0-x^*\|^2}{2k}.

这不仅说明最终趋近最优值,也说明常数取决于光滑参数和初始距离。它不要求强凸,却也没有给出强凸条件下那样的几何速度。

投影梯度

对闭凸约束 C,使用 x+=ΠC(xηf(x))x^+=\Pi_C(x-\eta\nabla f(x))。先沿负梯度移动,再找最近可行点。投影可能抵消部分移动,约束最优点因此不必满足 f=0\nabla f=0

固定点条件等价于 0f(x)+NC(x)0\in\nabla f(x)+N_C(x),把几何法锥与最优性联系起来。

近端梯度

若目标为光滑 f 加非光滑凸 g,定义

proxηg(v)=argminx{g(x)+12ηxv2},\operatorname{prox}_{\eta g}(v)=\arg\min_x\left\{g(x)+\frac1{2\eta}\|x-v\|^2\right\},

再令 x+=proxηg(xηf(x))x^+=\operatorname{prox}_{\eta g}(x-\eta\nabla f(x))。取 g 为集合指示函数就退化为投影梯度。

g(x)=λx1g(x)=\lambda\|x\|_1,逐坐标求解得到软阈值 xi=sign(vi)max(viηλ,0)x_i=\operatorname{sign}(v_i)\max(|v_i|-\eta\lambda,0)。小坐标被直接置零,说明稀疏性来自最优性条件,而不只是数值舍入。

练习

  1. f(x)=50x2f(x)=50x^2,比较步长 0.01、0.02、0.03 的迭代行为。
  2. 从一维次梯度条件推导软阈值。
  3. 为什么投影后梯度仍不为零可以是正确终止点?

上次更新: