Skip to content

算法Algorithm

随机近端梯度

Stochastic proximal gradient · Stochastic forward-backward method

对光滑项使用历史条件无偏梯度,对结构项精确做prox,以噪声方差和平均输出建立固定预算保证。

形式陈述 ​

设 F=g+h,g:Rn→R凸且为 L-光滑,L>0;h proper、闭凸,最小点 x∗存在。令历史 Fk包含当前 xk。随机oracle返回 vk,满足

(1)E[vk∣Fk]=∇g(xk),E[‖vk−∇g(xk)‖2∣Fk]≤σ2.

这些条件期望要求的是给定过去后的均值和方差,并不要求整条梯度序列相互独立。假定需要的随机量可积且更新可测。给定固定的 x0∈domh、整数预算 T≥1和固定 0<α≤1/(2L),做

(2)xk+1=proxαh(xk−αvk),x¯T=1T∑k=0T−1xk+1.

与确定近端梯度相同,prox本身精确求解;随机性只在光滑项oracle。本页证明

(3)E[F(x¯T)−F∗]≤‖x0−x∗‖22αT+ασ2.

输出平均点、预算、抽样方式与所用噪声上界。式(3)是固定预算的期望目标差,不是单次运行的可计算gap,也不是任选停时的高概率证书。

直觉

随机梯度保留光滑损失的平均方向,prox保留惩罚的尖角或约束。SGD的条件oracle结构在这里继续有用,但经过非线性prox后,“无偏梯度”不会变成“无偏下一点”。证明必须把随机误差与新点之间的依赖显式处理。

写 ξk=vk−∇g(xk)。ξk对旧历史均值为零,可是 xk+1已经使用了它,通常 E⟨ξk,xk+1−x∗⟩≠0。下面把这个内积拆成旧点项和本轮位移项:旧点项可条件消去,位移项由二次稳定项付账。这是随机复合分析中不可跳过的一步。

例子与边界

阈值之后为什么有偏 ​

取 g(x)=(x−1)2/2、h(x)=|x|/2,x0=0、α=1/2。每轮噪声取独立公平 ξ=±1,使用 v=x−1+ξ,因此满足式(1),L=1,σ2=1。

第一轮 v为 0 或 −2,梯度候选为 0 或 1;阈值为 α/2=1/4,所以 x1为 0 或 3/4,均值 3/8。若先用均值梯度 −1再做prox,结果为 S1/4(1/2)=1/4。3/8≠1/4说明不能交换期望和软阈值。

本问题最优点为 x∗=1/2、F∗=3/8。两个随机输出的目标分别是 1/2和 13/32,一步期望超额为 5/64。零输出不是统计意义上变量消失的证据,也不说明惩罚后的最优点为零。

若只掌握 ‖x0−x∗‖≤1,要求一个通用预算例,取 T=100、α=1/200<1/2,式(3)给期望超额不超过 2/100≈0.141421356。这只是所列上界;它不声称例中100轮实际误差等于这个数。

有限和的定标和完整检查 ​

对于不除以样本数的平方损失 g(x)=12∑i=1m(aiTx−bi)2,均匀有放回抽行 I时,正确oracle是 v=maI(aITx−bI)。漏掉 m会把目标的损失部分缩小,相对于固定 λ‖x‖1改变了问题。若改为平均损失,oracle才不用这个 m。

行梯度方差可能随 x增大,无约束Lasso不自动满足全程统一 σ2。要使用式(3),必须证明轨道所在区域的方差界,或将有界约束纳入 h并精确计算相应prox。只写“有限数据所以方差有界”不足以覆盖无限参数空间。

若需Lasso可行对偶证书,应在实际返回点计算完整 r=b−Ax和所有列的相关性;一个抽样梯度或抽样KKT残差不能证明所有对偶约束成立。这些完整检查的成本不包含在每次随机行查询中。

推论与应用

噪声如何进入三点不等式 ​

简写 x=xk,z=xk+1,ξ=ξk。精确prox给

s=(x−z)/α−∇g(x)−ξ∈∂h(z).

用其次梯度不等式、凸性和下降引理,对 u=x∗得到

F(z)−F∗≤‖x−x∗‖2−‖z−x∗‖22α−(12α−L2)‖z−x‖2−⟨ξ,z−x∗⟩.

因为 α≤1/(2L),位移平方的系数至少是 1/(4α)。拆开最后的内积,用

−⟨ξ,z−x⟩≤α‖ξ‖2+‖z−x‖24α,

消去这一位移项,剩下 −⟨ξ,x−x∗⟩。条件于历史,它的期望为零,而 E[‖ξ‖2∣Fk]≤σ2。求全期望并对 T轮求和,距离项望远镜消去,再由Jensen不等式处理平均点,就得到式(3)。没有把依赖噪声的新点冒充旧历史可测量。

如果距离有效上界为 D>0且 σ>0,在步长限制内平衡两项可选 α=min{1/(2L),D/(σ2T)};当第二项被选中,界为 2Dσ/T。噪声为零时可取 1/(2L)得到 O(1/T),但本证明的保守常数不替代确定近端页的更紧版本。

不同误差需要不同预算 ​

每轮有一次随机oracle、一次精确prox和 O(n)平均累加;T轮的prox总成本必须另报。独立同点小批量把方差上界缩小为 σ2/b,却使用 b次oracle;相关批量不能自动除以 b。随机小批量与内层近似prox也不是同一个误差来源,后者会再加一份确定或可控残差项。

FISTA的无噪声加速势对增长权重很敏感。将式(1)oracle直接代入它,不能宣称仍有无噪声 O(1/T2)。本页以平均输出换取清楚的随机保证;最后一点、强凸加速和高概率停止是另需条件的任务。

两道短自测 ​

  1. D=1,σ=2,T=200,L=1,取 α=D/(σ2T),它有效吗,式(3)是多少?答案:α=1/40≤1/2,两项各0.1,总界0.2。
  2. 不归一化的10行平方损失,只抽到第3行,oracle应为哪一项?答案:10a3(a3Tx−b3);若只用原行梯度,均值是目标梯度的十分之一,固定惩罚下改变了目标。

随机复合镜像下降使用可能不光滑的损失、原始对偶二阶矩与更新前平均,并为结构项首尾差付账;本页使用光滑项中心化方差、欧氏 prox 与更新后平均。相同形式的更新不使两份保证互换。需要减少昂贵 prox 次数时,批量随机近端加速另以放大曲率和带权噪声势证明最后主点界,给出8次 prox、242次样本的预算;它不是直接继承无噪声 FISTA。

参考资料
关系图谱15 个相邻概念 · 3 类关系

拖动节点调整位置。

显示关系

显示:依赖

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