形式陈述
在有限维空间 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 ) ,仍应与这里的拆分子问题区分。
把复合残差转换成可核验的停止条件
梯度映射 G α ( x ) 检查的是旧点 x 的固定点偏差;要在新点 z = prox α h ( x − α ∇ g ( x ) ) 构造真正的目标次梯度,可用
v = G α ( x ) + ∇ g ( z ) − ∇ g ( x ) ∈ ∂ F ( z ) , ‖ v ‖ ≤ ( 1 + α L ) ‖ G α ( x ) ‖ . 第一式来自prox最优性给出的 G α ( x ) − ∇ g ( x ) ∈ ∂ h ( z ) ,再加上 ∇ g ( z ) 。若已知 ‖ z − x ∗ ‖ ≤ R ,次梯度下界和Cauchy–Schwarz给 F ( z ) − F ∗ ≤ R ‖ v ‖ 。若整个 F 具有$\mu$-强凸性 理路 强凸性 Strong convexity · Strongly convex function 函数在一阶凸下界之外还保留统一二次增长量的曲率性质。 ,对其二次下界最小化,则得到 F ( z ) − F ∗ ≤ ‖ v ‖ 2 / ( 2 μ ) 。二者都需要明确的距离或曲率信息;仅有一个很小的残差数不够。
例如 F ( x ) = μ x 2 / 2 ,在 x = δ / μ 处梯度为 δ ,目标差却为 δ 2 / ( 2 μ ) ,可在 μ 很小时很大。若只看位移,则 F ( x ) = x 2 / 2 , x = 1 , α = 10 − 12 给约 10 − 12 的位移,目标差仍约 1 / 2 。步长缩小能够制造小位移,不能制造同等精度的最优性。
对Lasso,优先使用可行原对偶gap 理路 Lasso 的最优性与对偶间隙 Lasso optimality conditions · Lasso duality gap · Lasso primal-dual certificate 从残差相关性核验 Lasso 的零与非零坐标,并用可行对偶值认证剩余优化误差。 直接上界目标差,不需要猜最优点距离。坐标下降 理路 Lasso 坐标下降 Lasso coordinate descent · Lasso cyclic coordinate minimization 用部分残差和列范数逐坐标精确最小化 Lasso,并保持残差缓存与可行对偶证书一致。 和FISTA 理路 FISTA 复合加速法 FISTA · Fast iterative shrinkage-thresholding algorithm 以近端主点、外推查询点和递推权重组成复合凸加速,并用势函数而非逐轮下降证明函数值速率。 都可以产生候选,再使用同一证书;安全筛除 理路 Lasso 的间隙安全筛除 Gap Safe screening for Lasso · Lasso safe sphere screening 把可行原对偶间隙转成最优对偶点的安全球,再以严格列判据证明某坐标在所有 Lasso 最优解中为零。 进一步将gap转成可永久删列的严格判据。这里的停止接口仍完整保留,方法选择不会改变证书含义。
若prox只有近似解,需改用内层残差预算 理路 近似近端梯度的误差预算 Inexact proximal gradient · Proximal subproblem residual budget 用可核验的近端子问题次梯度残差分配内层精度,并把累计误差传到外层平均目标差。 ;若梯度带条件无偏噪声,需采用随机近端的平均输出保证 理路 随机近端梯度 Stochastic proximal gradient · Stochastic forward-backward method 对光滑项使用历史条件无偏梯度,对结构项精确做prox,以噪声方差和平均输出建立固定预算保证。 。这两种误差不能仅凭“每轮看起来下降”省略。
结构项还可以作用于整组变量或矩阵奇异方向。组 Lasso 理路 组 Lasso 的块收缩与对偶证书 Group Lasso proximal certificate · Block soft thresholding · 组软阈值 对不相交变量组推导径向近端收缩,以整组残差相关性建立KKT和可行对偶间隙,并在耦合设计上认证停止。 把本页的欧氏近端子问题变成块收缩,并以整组对偶球认证;带核惩罚的矩阵补全 理路 矩阵补全的优化间隙与唯一性证书 Matrix completion certificate · Nuclear norm completion dual certificate · 核范数补全证书 以掩码近端梯度求解带核惩罚的观测损失,用支持受限的谱范数对偶点认证误差,再以切空间严格证书证明精确补全唯一性。 把光滑梯度换成观测掩码残差,保留未观测位置的当前猜测,再做奇异值软阈值。两者都逐项检查本页的凸性和固定步长条件,误差证书分别需要全组相关性或观测支撑内的谱范数约束。
参考资料