形式陈述
在有限维空间 R n 中考虑
min x F ( x ) = g ( x ) + h ( x ) . 本页的收敛结论采用以下条件:g 是凸的 C 1 函数,梯度在全空间上满足 L -Lipschitz 条件(L > 0 );h 是 proper、闭凸函数;最小点集合非空;x 0 ∈ dom h 。固定 0 < α ≤ 1 / L ,每次精确计算
x k + 1 = prox α h ( x k − α ∇ g ( x k ) ) . 这是一种前向—后向分裂:先做梯度下降 公理库 梯度下降法 Gradient descent method · Euclidean steepest descent method 反复沿当前负梯度方向取步以降低可微目标的基础一阶算法。 的显式前向步,再做近端算子 公理库 近端算子 Proximal operator · Proximity operator 在降低凸函数值与保持靠近输入点之间取得精确平衡的单值算子。 的隐式后向步。这里两部分分别是凸目标的梯度与凸次微分;更一般的单调算子分裂不必来自这样的目标。h 可以取 + ∞ ,因而允许用闭凸集合的指标函数表示约束;初值位于有效域,使第一轮开始的目标值有限。一般更新式也可使用变步长,但仅有 0 < α k ≤ 1 / L 并不足以继承本页的收敛结论。
定义近端梯度映射
G α ( x ) = x − prox α h ( x − α ∇ g ( x ) ) α . prox 最优性给出 G α ( x ) = 0 当且仅当 0 ∈ ∇ g ( x ) + ∂ h ( x ) ,即复合凸问题的一阶最优性条件 公理库 一阶最优性条件 First-order optimality condition · Variational inequality optimality condition 以梯度和所有可行方向的非负内积充要刻画可微凸问题的全局极小点。 。因此可以用它衡量固定点残差;h 不可微时不能改报不存在的 ‖ ∇ F ( x ) ‖ 。若要认证还剩多少目标值误差,还需能计算的误差界,例如后文链接的 Lasso 对偶间隙。
直觉
拆分的依据是计算结构:g 的梯度便宜,h 的 prox 便宜。把更新配方,可见新点精确最小化
g ( x k ) + ⟨ ∇ g ( x k ) , u − x k ⟩ + ‖ u − x k ‖ 2 2 α + h ( u ) . 当 α ≤ 1 / L 时,光滑凸性 公理库 光滑凸函数 Smooth convex function · L-smooth convex function 同时具有凸性与全局 Lipschitz 梯度的函数类,其曲率被零与有限上界夹住。 与下降引理 公理库 下降引理 Descent lemma · Quadratic upper-bound lemma 以 Lipschitz 梯度常数给出函数相对一阶模型的全局二次上界。 保证这是一份接触当前目标值的二次上界。它线性化光滑项,却完整保留惩罚的尖角,所以能产生精确零坐标。
图片加载失败 梯度候选与结构化近端收缩 图中选取 α h ( x ) = ‖ x ‖ 1 ,故近端步是阈值为 1 的逐坐标软阈值:( 2.1 , − 0.3 , 0.2 , 1.4 ) ↦ ( 1.1 , 0 , 0 , 0.4 ) 。
当 h = 0 ,prox 是恒等映射;当 h 是集合指标函数,prox 是投影。步长约束来自 g 的曲率,不需要为非光滑的整个 F 编造 Hessian。若每轮改为对整个 F 求 prox,就变成近端点方法 公理库 近端点方法 Proximal point method · Proximal point algorithm 反复精确求解带二次稳定项的子问题以逼近闭凸函数最小点的方法。 ,通常会得到另一个、更难的子问题。
例子与边界
一维收缩与耦合回归
保留最简单的计算作为起点:F ( x ) = ( x − 3 ) 2 / 2 + | x | ,L = 1 。从 x 0 = 0 取 α = 1 / 2 ,软阈值 S τ ( z ) = sign ( z ) ( | z | − τ ) + 给出
x 1 = S 1 / 2 ( 3 / 2 ) = 1 , x 2 = S 1 / 2 ( 2 ) = 3 / 2 , x 3 = S 1 / 2 ( 9 / 4 ) = 7 / 4. 目标值依次为 4.5 , 3 , 2.625 , 2.53125 ,趋向 x ∗ = 2 的 F ∗ = 2.5 。若取 α = 1 ,梯度候选恒为 3 ,于是一步到解。这里二次项恰好消除了初值影响;一般设计矩阵的列相互耦合时,不会如此。
具体地,令
A = ( 1 1 1 0 0 1 ) , b = ( 3 / 2 3 / 2 0 ) , P ( x ) = 1 2 ‖ A x − b ‖ 2 + ‖ x ‖ 1 . 此时 A T A = ( 2 1 1 2 ) 的最大特征值为 3 ,可以取 α = 1 / 3 。从零开始,前三个迭代点为 ( 2 / 3 , 1 / 6 ) 、( 5 / 6 , 0 ) 、( 17 / 18 , 0 ) 。第二步恰把处于阈值边界的第二坐标送到零。完整候选计算、零坐标最优性与逐轮误差证书见Lasso 的最优性与对偶间隙 公理库 Lasso 的最优性与对偶间隙 Lasso optimality conditions · Lasso duality gap · Lasso primal-dual certificate 从残差相关性核验 Lasso 的零与非零坐标,并用可行对偶值认证剩余优化误差。 ;那里同时说明为何不能先求最小二乘解再一次软阈值。
同一组数据也可用交替方向乘子法(ADMM) 公理库 交替方向乘子法(ADMM) Alternating direction method of multipliers · ADMM 将凸目标拆成两个近端子问题,用乘子累积一致性误差,并以原始与对偶残差检查最优性的算法。 求解:它对平方损失精确解一个耦合线性系统,对复制变量做软阈值,再累积两份变量的不一致量。近端梯度在这里使用显式梯度候选,ADMM 则保留完整二次损失;该页给出四轮分数计算,并说明为什么停止时需要同时检查原始残差与对偶残差。
步长与结论的边界
“小于安全上界”不意味着累积推进足够。令 h = 0 、g ( x ) = x 2 / 2 、α k = 2 − k − 2 ,则
x k = x 0 ∏ j = 0 k − 1 ( 1 − α j ) . 这些步长均不超过 1 / L = 1 ,但和有限。因为 0 ≤ α j ≤ 1 / 4 ,有 log ( 1 − α j ) ≥ − 2 α j ,所以无穷乘积严格为正;非零初值不会趋于最优点零。这就是本页明确固定正步长的原因。
过大步长也可能破坏下降。对上面的耦合例取 α = 1 ,第一步是 ( 2 , 1 / 2 ) ,目标值从 9 / 4 升到 13 / 4 。这展示一个实际失败,不表示每个 α > 1 / L 都必然失败。若用回溯,应检验光滑二次上界并另行说明步长规则;若 prox 仅近似求解,应控制内层误差;若 g 非凸,固定点一般只认证驻点。
推论与应用
三点不等式如何产生下降与速率
记 y = prox α h ( x − α ∇ g ( x ) ) 。最优性条件给出
s = x − y α − ∇ g ( x ) ∈ ∂ h ( y ) . 任取 z ∈ dom h ,次梯度不等式给 h ( y ) − h ( z ) ≤ ⟨ s , y − z ⟩ ;凸性给 g ( x ) − g ( z ) ≤ ⟨ ∇ g ( x ) , x − z ⟩ ;光滑性给 g ( y ) − g ( x ) ≤ ⟨ ∇ g ( x ) , y − x ⟩ + L ‖ y − x ‖ 2 / 2 。相加后,梯度内积消去,剩下 ⟨ x − y , y − z ⟩ / α + L ‖ y − x ‖ 2 / 2 。再用三点平方恒等式,得到
F ( y ) − F ( z ) ≤ ‖ x − z ‖ 2 − ‖ y − z ‖ 2 2 α − ( 1 2 α − L 2 ) ‖ y − x ‖ 2 . 先取 z = x 。前一个分式本身又贡献一份负平方项,所以
F ( y ) ≤ F ( x ) − ( 1 α − L 2 ) ‖ y − x ‖ 2 ≤ F ( x ) − ‖ y − x ‖ 2 2 α . 即使 α = 1 / L ,只要新旧点不同仍有严格下降。再取任一最小点 z = x ∗ ,舍去三点不等式末尾的非正项,从 j = 0 到 k − 1 求和便有
∑ j = 0 k − 1 ( F ( x j + 1 ) − F ∗ ) ≤ ‖ x 0 − x ∗ ‖ 2 − ‖ x k − x ∗ ‖ 2 2 α ≤ ‖ x 0 − x ∗ ‖ 2 2 α . 这一步首先控制的是一串函数值差的和。由于刚刚证明了函数值单调,每个求和项都不小于最后一项 F ( x k ) − F ∗ ,故
F ( x k ) − F ∗ ≤ ‖ x 0 − x ∗ ‖ 2 2 α k , k ≥ 1. 点收敛需要再走一步。三点不等式与 F ( x k + 1 ) ≥ F ∗ 表明,到每个最小点的距离不增,这称为 Fejér 单调性,因而序列有界。下降估计还给 ∑ k ‖ x k + 1 − x k ‖ 2 < ∞ ,故相邻差趋零。有限维有收敛子列 x k j → x ¯ ;prox 是连续映射,梯度也连续,固定步长更新与相邻差趋零便推出 x ¯ = prox α h ( x ¯ − α ∇ g ( x ¯ ) ) 。所以 x ¯ 是最小点。最后,到这个 x ¯ 的距离不增且有子列趋零,整个序列都趋于 x ¯ 。这一论证没有假设 h 在定义域边界连续。
计算成本与可以认证的对象
对 m × n 矩阵的平方损失,梯度通过 A x 和 A T ( A x − b ) 计算,连同残差与输出向量操作,稠密实数算术成本为 O ( 1 + m n + m + n ) ,在 m , n ≥ 1 时可简写为 O ( m n ) ;若稀疏表示实际保存 s 个条目(包括显式零和重复坐标),稀疏实现为 O ( 1 + s + m + n ) ,逐坐标软阈值另需 O ( n ) 。不必形成可能稠密的 A T A 。要报告新点处的残差与对偶证书,还需新点的矩阵向量乘法、转置乘法及 O ( 1 + m + n ) 的范数和内积检查;可以与下一轮梯度共享这些结果。
FISTA 在相应凸条件下将函数值界改善为 O ( 1 / k 2 ) ,但需新的加速权重与势函数,且通常不逐轮下降。函数值收敛、迭代点收敛与有限步识别支持集是不同结论;上述一般 O ( 1 / k ) 证明没有提供统计意义的变量恢复保证。Moreau 包络 公理库 Moreau 包络 Moreau envelope · Moreau-Yosida regularization 以二次 infimal convolution 将闭凸函数平滑化并保留其极小点的函数。 则对整个函数做平滑,虽然也出现 x − prox ( x ) ,仍应与这里的拆分子问题区分。
参考资料