Skip to content

算法Algorithm

随机坐标下降与坐标曲率

Randomized coordinate descent · Coordinate Lipschitz sampling

按坐标曲率抽取一个精确偏导更新,以条件期望核算进展,并分清单坐标、整块与同时更新的步长。

形式陈述 ​

设 f:Rn→R 凸可微,最小点存在。对每个坐标给定 Li>0,使

(1)|∂if(x+tei)−∂if(x)|≤Li|t|,x∈Rn, t∈R.

这里 ∂if 是梯度的第 i 分量。令 S=∑iLi、pi=Li/S,初始化 x0。每轮在观察本轮坐标之前确定 xk;相对于历史 Fk,以固定概率 pi 抽取 ik,只查询该坐标的精确偏导,再执行

(2)xk+1=xk−∂ikf(xk)Likeik.

抽样要求 Pr(ik=i∣Fk)=pi,后续证明通过历史条件期望实现;一轮内无放回遍历不自动满足它。输入还包括访问预算和误差目标;输出点、访问数、抽样协议及独立可用的停止证书。期望率是固定预算保证,本身不是某次运行已经达标的证明。

若 f 另外为欧氏 μ-强凸、μ>0,下面将证明

(3)E[f(xk)−f∗]≤(1−μ/S)k(f(x0)−f∗).

这里处理光滑目标;可分非光滑惩罚的坐标prox更新会在后文说明,但不直接继承式(3)的光滑证明。

直觉

全梯度的Lipschitz常数控制所有方向;Li只控制第 i 坐标单独移动时同一偏导的变化。对 f(x)=xTQx/2−cTx、Q⪰0,可取 Li=Qii;最小全局欧氏常数却是 λmax(Q)。对有耦合的矩阵,它们不能互换。

由沿坐标线段积分,式(1)给

f(x+tei)≤f(x)+t∂if(x)+Lit2/2.

把右侧二次式最小化,就得到式(2),且下降至少为 [∂if(x)]2/(2Li)。高曲率坐标单次步子小,但按 Li 更频繁地访问它,恰好让期望中的分母相消。这是对变量索引抽样;SGD通常对样本或梯度oracle抽样,随机性的来源不同。

例子与边界

两种曲率与两种抽样 ​

取 f(x)=(x12+9x22)/2、x0=(1,1)。L1=1,L2=9,S=10,μ=1,初始目标为 5。抽到坐标1会把它精确置零,留下目标 9/2;抽到坐标2会留下目标 1/2。按 p=(1/10,9/10) 抽样,一步期望目标为 9/10;均匀抽样的一步期望为 5/2。

由于某个坐标一旦访问就归零,k 次访问后的精确期望是

Ef(xk)=12(9/10)k+92(1/10)k.

式(3)的上界 5(9/10)k 比实际值保守。取 k=2,实际为 0.45,上界为 4.05。这既核对抽样,也说明最坏界不能代替实例计算。若初值改为 (1,0),均匀抽样更快消掉唯一误差;按曲率抽样优化的是一类统一保证,不是每个初值的最佳策略。

单坐标安全,不代表同时更新安全 ​

令 f(x)=(x1+x2+x3)2/2。每个 Li=1,但全局光滑常数为 3。从 (1,1,1) 开始,单独更新一个坐标给 (−2,1,1),目标降至零。若三个坐标都用旧点偏导 3、各自按步长1同时更新,结果为 (−2,−2,−2),目标从 9/2 升至 18。把三条安全的单坐标规则并行执行,会重复使用同一份未更新的耦合信息。

若把变量分成不相交块 Ib,应给块梯度的常数 Lb:‖∇bf(x+Ubd)−∇bf(x)‖≤Lb‖d‖。每轮只选一块,沿该块按 1/Lb 更新,前述证明把标量平方换成块范数平方即可。二次问题的 Lb=λmax(Qbb),通常不是该块对角元的最大值。多个块同时更新还需新的耦合上界,本页不为它提供串行率。

坐标曲率为零时不能使用除法。若该方向为常数,可固定不动;若为非零线性函数,在无约束空间上没有有限最小点,与本页假设冲突。也可以选严格正的保守上界,但必须明确它是否真正满足式(1)。

推论与应用

从单步下降到两个速率 ​

条件于 Fk,xk固定。按式(1)的下降求期望:

(4)E[f(xk+1)∣Fk]≤f(xk)−∑ipi2Li[∂if(xk)]2=f(xk)−‖∇f(xk)‖22S.

强凸性给 f(z)≥f(x)+⟨∇f(x),z−x⟩+μ‖z−x‖2/2。将右侧对整个空间最小化,得到 f∗≥f(x)−‖∇f(x)‖2/(2μ)。代入式(4)再取全期望,得到误差每轮至少收缩因子 1−μ/S,归纳即式(3)。

没有强凸性时,若已知整个初始下水平集到某个最优点 x∗ 的距离至多 0<R0<∞,逐次下降保证迭代留在其中。凸性给 f(xk)−f∗≤R0‖∇f(xk)‖。记 ak=E[f(xk)−f∗],用 EZ2≥(EZ)2 得

ak+1≤ak−ak22SR02.

若某步 ak=0,目标误差已经几乎必然为零;否则 ak+1≤ak,故 1/ak+1−1/ak≥1/(2SR02)。求和得到 ak≤2SR02/k。最小点存在并不自动保证所需下水平集有界,必须单独检查这个条件。

可分惩罚的接口与成本 ​

对 F=f+∑ihi(xi),可把式(2)换成 xi+=proxhi/Li(xi−∂if(x)/Li)。它精确最小化这一坐标的二次上模型,因而保证 F不增;要建立随机复合率,需保留惩罚差和近端映射,不能把式(4)里的普通梯度范数直接照抄。Lasso坐标下降是平方损失加绝对值惩罚的具体执行方案,给出真实残差缓存和循环证明。

一次偏导未必便宜:若查询器先计算完整梯度,随机坐标节省的只是更新算术。平方损失维护残差后,访问第 i 列需 O(1+nnz(Ai));初始化残差、曲率表、抽样表以及最终全量证书都另计。固定抽样分布可用 O(n)预处理建立累计概率表,每次二分抽样为 O(log⁡n);这些费用也在预算内。报告 k 次坐标访问不能直接与 k 次全梯度比较。

两道短自测 ​

  1. f=(4x12+x22)/2,从 (1,1)按曲率抽样,一步期望目标是多少?答案:概率为 (4/5,1/5),两种剩余目标为 1/2,2,期望 4/5。
  2. 对 f=(x1+x2+x3)2/2,把三坐标合为一个块时能否取块常数1?答案:不能;该块Hessian为全1矩阵,最大特征值3,常数至少3。对角元1只适用于单坐标移动。
参考资料
  • Stephen J. Wright, Coordinate Descent Algorithms,§3.2 的分量曲率、§3.3 的条件期望分析、§3.7 的可分正则扩展。本文完整推导按 Li 抽样的版本;原文 Theorem 1 展示均匀抽样与共同步长版本。
关系图谱11 个相邻概念 · 2 类关系

拖动节点调整位置。

显示关系

显示:依赖

  1. 前置三跳
  2. 前置二跳
  3. 前置一跳
  4. 当前条目
  5. 后续一跳
  6. 后续二跳
  7. 后续三跳
文字版关系按与当前条目的最短距离分组
类型化关系