《强化学习的数学原理》赵世钰 著 · 第 6 章
本章是全书的"数学工具箱"。第 5 章的蒙特卡罗方法用样本均值估计期望,但每次都从头算。能不能增量式更新?这需要随机近似理论。
一、从期望值估计说起
批量 vs 增量
批量更新(MC 方法的做法):
xˉN=N1i=1∑Nxi
每来一个新样本,要重新算所有样本的均值。
增量更新:
xˉN=xˉN−1+N1(xN−xˉN−1)
只保留上一次的均值,用新样本做小修正。这大大节省了内存。
这个递推公式怎么来的? 纯代数推导:
xˉN=N1i=1∑Nxi=N1(i=1∑N−1xi+xN)=NN−1xˉN−1+N1xN=xˉN−1+N1(xN−xˉN−1)
二、罗宾斯-门罗算法(Robbins-Monro Algorithm)
一般形式
wt+1=wt−αtg(wt,Xt)
其中 g(wt,Xt) 是一个随机量,wt 是要估计的参数,αt 是步长。
收敛条件
步长序列 {αt} 需要满足 Robbins-Monro 条件:
t=0∑∞αt=∞,t=0∑∞αt2<∞
直觉理解:
- ∑αt=∞:步长不能衰减太快,否则总更新量不够
- ∑αt2<∞:步长不能衰减太慢,否则噪声累积不收敛
典型选择:αt=tc 或 αt=t+ac。
为什么 αt=t1 满足条件?
- ∑t1=∞(调和级数发散)✓
- ∑t21<∞(p-级数收敛)✓
在期望值估计中的应用
设要估计 E[X],用 Robbins-Monro 算法:
wt+1=wt−αt(wt−xt)=(1−αt)wt+αtxt
在 RM 条件下,wt→E[X]。
三、Dvoretzky 定理
这是随机近似理论的最一般定理,Robbins-Monro 是它的特例。
定理形式
考虑迭代:
Δt+1=(1−αt)Δt+βt+ηt
如果:
- αt≥0, ∑αt=∞, ∑αt2<∞
- E[βt∣Ft]=0, ∑E[βt2]<∞
- E[ηt∣Ft]≤0(漂移项非正)
则 Δt→0(依概率收敛)。
直觉:αt 控制衰减速度,βt 是零均值噪声,ηt 是向零漂移的力。三者共同保证收敛。本书用这个定理统一证明了多种 RL 算法的收敛性。
四、随机梯度下降(SGD)
从梯度下降到随机梯度下降
梯度下降(GD):
wt+1=wt−α∇J(wt)
需要计算全量数据的梯度,代价 O(N)。
随机梯度下降(SGD):
wt+1=wt−αt∇J(wt,xt)
每次只用一个样本 xt 的梯度,代价 O(1)。
关键性质
E[∇J(wt,xt)]=∇J(wt)
即随机梯度的期望等于真实梯度。这使得 SGD 是 GD 的无偏随机近似。
在期望值估计中的应用
目标函数 J(w)=21E[(w−X)2],梯度 ∇J=E[w−X]。
SGD 更新:
wt+1=wt−αt(wt−xt)
这与 Robbins-Monro 算法完全一致!说明 SGD 是 RM 的特例。
五、SGD 的收敛模式
三种收敛模式
| 模式 |
步长 |
收敛到 |
特点 |
| 模式 1 |
固定 α |
J 的邻域 |
不精确收敛,但实用 |
| 模式 2 |
αt→0,满足 RM 条件 |
J 的极小点 |
理论收敛 |
| 模式 3 |
小批量 |
近似极小点 |
实践常用 |
小批量梯度下降
wt+1=wt−αm1i=1∑m∇J(wt,xt,i)
用 m 个样本的平均梯度,方差更小,但每步代价 O(m)。
实践智慧:小批量是"方差-效率"的折中。m=1 是纯 SGD(高方差,低成本),m=N 是 GD(零方差,高成本)。深度学习中通常选 m=32∼256。
六、为什么本章重要?
本章看似与强化学习无关,但实际上它是第 7 章(TD 方法)和第 8 章(值函数近似)的数学基础:
第 6 章(随机近似)
↓
第 7 章 TD 方法的收敛性证明
↓
第 8 章 值函数近似的优化算法
↓
第 9-10 章 策略梯度和 Actor-Critic 的优化
没有随机近似理论,就无法证明 TD(0) 收敛、无法分析 Q-learning 的稳定性。
七、总结
| 工具 |
核心公式 |
用途 |
| 增量均值 |
xˉN=xˉN−1+N1(xN−xˉN−1) |
在线更新均值 |
| Robbins-Monro |
wt+1=wt−αtg(wt,Xt) |
随机近似的一般框架 |
| Dvoretzky |
Δt+1=(1−αt)Δt+βt+ηt |
最一般的收敛定理 |
| SGD |
wt+1=wt−αt∇J(wt,xt) |
随机优化 |
评论(0)