形式陈述
考虑复合问题
min x ∈ R n F ( x ) = g ( x ) + h ( x ) , 其中 g 可微,h 为 proper、下半连续凸函数且其 prox 可计算。给定 x 0 、步长 α k > 0 、容差与预算,近端梯度法迭代
x k + 1 = prox α k h ( x k − α k ∇ g ( x k ) ) . 第一部分是梯度下降 公理库 梯度下降法 Gradient descent method · Steepest descent method 反复沿当前负梯度方向取步以降低可微目标的基础一阶算法。 的显式前向步,第二部分是近端算子 公理库 近端算子 Proximal operator · Proximity operator 在降低凸函数值与保持靠近输入点之间取得精确平衡的单值算子。 的隐式后向步,因此也称 forward-backward splitting。算法定义只规定更新与输出状态;若要保证凸全局收敛,另需 g 为$L$-光滑凸函数 公理库 光滑凸函数 Smooth convex function · L-smooth convex function 同时具有凸性与全局 Lipschitz 梯度的函数类,其曲率被零与有限上界夹住。 、h 凸、解存在,并选择例如 0 < α k ≤ 1 / L 。
定义近端梯度映射
G α ( x ) = 1 α ( x − prox α h ( x − α ∇ g ( x ) ) ) . 更新即 x + = x − α G α ( x ) 。由 prox 的最优性条件,G α ( x ) = 0 当且仅当
0 ∈ ∇ g ( x ) + ∂ h ( x ) , 这正是复合凸问题的一阶最优性条件 公理库 一阶最优性条件 First-order optimality condition · Variational inequality optimality condition 以梯度和所有可行方向的非负内积充要刻画可微凸问题的全局极小点。 。停止时应报告 ‖ G α ( x ) ‖ ,不能在 h 不可微时误报不存在的 ‖ ∇ F ( x ) ‖ 。
直觉
若直接对 g + h 做梯度步,h 的尖角可能没有梯度;若每轮对整个 F 做 prox,又可能把本来容易求梯度的 g 放进昂贵子问题。近端梯度恰好按计算结构拆分:先用 g 的局部线性信息提出候选,再让 h 的 prox 在降低惩罚与靠近候选之间精确权衡。算法利用的是“一个梯度便宜、另一个 prox 便宜”,而不只是代数上能写成两项。
在 h = 0 时,prox 是恒等映射,算法退化为梯度下降;在 h 是集合指标函数时,prox 变成投影。这个统一视角也解释了软阈值:ℓ 1 惩罚不会在显式梯度中产生精确零,但其 prox 能把一整段输入直接送到零。步长仍由光滑项 g 的曲率控制,而不是由整个非光滑目标的某个虚构 Hessian 控制。
例子与边界
取一维复合目标
F ( x ) = 1 2 ( x − 3 ) 2 + | x | . g ( x ) = ( x − 3 ) 2 / 2 的 L = 1 ,h ( x ) = | x | 的 prox 是软阈值。用 α = 1 / 2 、x 0 = 0 ,有
x 1 = S 1 / 2 ( 0 − 1 2 ( − 3 ) ) = S 1 / 2 ( 1.5 ) = 1 , x 2 = S 1 / 2 ( 1 − 1 2 ( − 2 ) ) = S 1 / 2 ( 2 ) = 1.5 , x 3 = S 1 / 2 ( 2.25 ) = 1.75 , 其中 S τ ( z ) = sign ( z ) ( | z | − τ ) + 。函数值依次为 4.5 , 3 , 2.625 , 2.53125 ,趋向唯一解 x ∗ = 2 的 F ( x ∗ ) = 2.5 。若取紧步长 α = 1 ,无论初值为何,梯度候选都是 3 ,一次软阈值即到 2 ;这是一维特殊结构,不代表一般问题一步收敛。
边界来自拆分而非记号。若 h 的 prox 只能由昂贵内层迭代近似,必须控制内层误差;若 g 非凸,固定点通常只表示复合驻点而非全局极小。低估 L 可能破坏充分下降,可用回溯实际检查由下降引理 公理库 下降引理 Descent lemma · Quadratic upper-bound lemma 以 Lipschitz 梯度常数给出函数相对一阶模型的全局二次上界。 给出的二次模型。把两个都非光滑的项硬指定一个为“光滑项”,或把约束集合非凸时的多值投影当单值 prox,均超出标准理论。
推论与应用
在凸假设与 α ≤ 1 / L 下,更新的最优性不等式和光滑二次上界合并可得
F ( x k + 1 ) − F ( x ∗ ) ≤ ‖ x k − x ∗ ‖ 2 − ‖ x k + 1 − x ∗ ‖ 2 2 α . 求和后得到 F ( x k ) − F ∗ = O ( 1 / k ) ;若复合目标强凸,可进一步得到几何率。加上 Nesterov 权重便形成 FISTA 的 O ( 1 / k 2 ) 率,但主序列可能不单调,证明需要新的势函数。这里控制的是复合函数值或梯度映射,不应偷换成每个坐标恢复正确支持集。
Lasso、稀疏逻辑回归、总变差模型和带简单凸约束的经验风险最小化都可使用这一结构。Moreau 包络 公理库 Moreau 包络 Moreau envelope · Moreau-Yosida regularization 以二次 infimal convolution 将闭凸函数平滑化并保留其极小点的函数。 也通过 x − prox ( x ) 产生光滑梯度,但它对整个函数取 infimal convolution;近端梯度只对 h 取 prox。二者公式相似而优化对象不同,诊断时应先写清子问题。
参考资料
Amir Beck, First-Order Methods in Optimization , SIAM, 2017,Ch. 10,proximal gradient methods。
Neal Parikh and Stephen Boyd, Proximal Algorithms , Foundations and Trends in Optimization 1(3), 2014,§4.2,proximal gradient and splitting interpretation。
Amir Beck and Marc Teboulle, “A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems,” SIAM Journal on Imaging Sciences 2(1), 2009, 183–202,§§2–4,ISTA and FISTA rates。