Skip to content

一致稳定性泛化界

Uniform stability generalization bound · Stability generalization theorem

将学习算法对单点替换的一致稳定性转化为期望与经典高概率泛化间隙界。

假设与结论

设确定性学习算法 A 接受 m 个 IID 样本 SDm,损失 (A(S),z)[0,M]。若任意只相差一个样本的 S,S 与任意测试点 z 都满足

|(A(S),z)(A(S),z)|β,

则称 Aβ-一致稳定的。记泛化间隙

Φ(S)=RD(A(S))R^S(A(S)).

其期望满足

|ESΦ(S)|β.

此外,直接使用有界差分可得一个经典、但未必最紧的高概率版本:以至少 1δ 的概率

Φ(S)β+(2mβ+M)log(1/δ)2m.

若采用“删除一个点”而非“替换一个点”的稳定性定义,或对双边绝对间隙取界,常数会改变;应从相应定义重新计算,而非拼接不同版本。

期望界的换一法

S=(Z1,,Zm),再取独立副本 Zi,令 S(i)Zi 替换为 Zi。由于 (S(i),Zi)(S,Zi) 同分布,

EΦ(S)=1miE[(A(S),Zi)(A(S),Zi)]=1miE[(A(S),Zi)(A(S(i)),Zi)].

每一项绝对值至多 β,故结论成立。这里没有对整个假设类取上确界;控制来自算法输出随单点变化不大。

从期望到尾界

S 的第 i 点替换后,总体风险项至多改变 β。经验风险的其余 m1 项各至多改变 β,平均后不超过 β;被替换的那一项还可能改变 M/m。因此

|Φ(S)Φ(S(i))|2β+Mm.

m 个坐标应用 有界差分不等式,再代入 EΦβ,就得到上述高概率式。若 β=c/m,尾项仍是 O((c+M)log(1/δ)/m);现代稳定性尾界能在某些条件下更好地保留 β 的尺度,但不应冒充这条朴素 McDiarmid 推导。

具体例子:强凸正则化 ERM

考虑

A(S)=argminw{1mi=1m(w,Zi)+λ2w22},

(,z)w 凸并 L-Lipschitz。两个相邻样本对应的目标都为 λ-强凸;比较彼此最优性并相加,可得解的距离为 O(L/(λm))。再用损失的 Lipschitz 性,得到

β=O(L2λm).

图像是:正则项把目标函数底部做成有曲率的碗,单个样本只占 1/m 权重,无法把极小点推得太远。参数距离本身并不是稳定性;必须再由 Lipschitz 损失把参数变化转换为任意测试点上的损失变化。

边界

β 不随 m 衰减,期望间隙不会被迫趋零。无界损失也使经典尾界失去 M。随机算法还需明确稳定性是对同一随机种子耦合、对内部随机性取期望,还是联合高概率陈述;三者不能互换。本路线与 Rademacher 泛化界的区别在于,前者评价具体算法的敏感性,后者评价整个函数类在样本上的波动。

参考资料
  • Olivier Bousquet and André Elisseeff, “Stability and Generalization,” JMLR, 2002.
  • Vitaly Feldman and Jan Vondrák, work on high-probability generalization bounds for uniformly stable algorithms.