Personal Knowledge Base

A long-term research and learning notebook for posts, notes, papers, projects, and research directions.

Skip to content
← Back to notes

强化学习 08:Value Function Approximation

用参数化模型表示大规模状态空间中的值函数,理解半梯度 TD、投影 Bellman 方程与深度 Q 网络。

4 min read

为什么需要函数近似

表格型方法为每个状态或状态动作对保存一个独立数值。当状态空间连续或维度很高时,不可能访问并存储所有条目。函数近似用共享参数表示价值:

V^(s;w)Vπ(s)\hat{V}(s;w)\approx V^\pi(s)

或:

Q^(s,a;w)Qπ(s,a)\hat{Q}(s,a;w)\approx Q^\pi(s,a)

共享参数使模型可以把已见状态的经验泛化到相似状态。

监督学习视角

如果有目标 UtU_t,可以最小化均方误差:

J(w)=12E[(UtV^(St;w))2]J(w) =\frac{1}{2}\mathbb{E} \left[ \left(U_t-\hat{V}(S_t;w)\right)^2 \right]

随机梯度更新为:

wt+1=wt+α[UtV^(St;wt)]wV^(St;wt)w_{t+1} =w_t+\alpha \left[ U_t-\hat{V}(S_t;w_t) \right] \nabla_w\hat{V}(S_t;w_t)

Monte Carlo 可以令 Ut=GtU_t=G_t;TD 方法则令:

Ut=Rt+1+γV^(St+1;wt)U_t=R_{t+1}+\gamma\hat{V}(S_{t+1};w_t)

半梯度 TD

TD 目标本身依赖参数 wtw_t,但半梯度方法只对当前预测求导,把目标暂时视为常数:

wt+1=wt+αδtwV^(St;wt)w_{t+1} =w_t+\alpha\delta_t \nabla_w\hat{V}(S_t;w_t) δt=Rt+1+γV^(St+1;wt)V^(St;wt)\delta_t =R_{t+1}+\gamma\hat{V}(S_{t+1};w_t) -\hat{V}(S_t;w_t)

这并不一定是某个普通均方损失的完整梯度,但在线性同策略设置下具有良好的收敛性质。

线性函数近似

令特征向量为 x(s)Rdx(s)\in\mathbb{R}^d

V^(s;w)=wx(s)\hat{V}(s;w)=w^\top x(s)

则梯度为 x(s)x(s),更新简化为:

wt+1=wt+αδtx(St)w_{t+1}=w_t+\alpha\delta_t x(S_t)

线性方法计算便宜、容易分析,并可结合 tile coding、Fourier basis 或人工设计特征。

投影 Bellman 方程

函数类通常无法精确表示真实值函数。TD 学习寻找的不是直接最小化每个状态的 Bellman 误差,而是近似求解:

V^=ΠTπV^\hat{V}=\Pi\mathcal{T}^\pi\hat{V}

其中 Π\Pi 表示把 Bellman 更新后的函数投影回可表示的函数空间。采样分布决定了投影使用的权重,因此训练数据覆盖范围会直接影响近似结果。

动作价值近似

控制问题中可以使用:

Q^(s,a;w)\hat{Q}(s,a;w)

半梯度 SARSA 的更新为:

wt+1=wt+α[Rt+1+γQ^(St+1,At+1;wt)Q^(St,At;wt)]wQ^(St,At;wt)w_{t+1} =w_t+\alpha \left[ R_{t+1}+\gamma\hat{Q}(S_{t+1},A_{t+1};w_t) -\hat{Q}(S_t,A_t;w_t) \right] \nabla_w\hat{Q}(S_t,A_t;w_t)

离散动作可以让网络一次输出所有动作的 QQ 值;连续动作中直接计算 maxaQ(s,a)\max_aQ(s,a) 更困难,通常需要额外的策略网络。

Deep Q-Network

DQN 使用神经网络逼近 QQ,并通过两项关键机制改善稳定性:

  • Experience Replay:随机采样历史转移,减弱连续数据的相关性并提高样本复用率。
  • Target Network:使用延迟更新的参数 ww^- 构造相对稳定的目标。

典型目标为:

yt=Rt+1+γmaxaQ^(St+1,a;w)y_t=R_{t+1}+\gamma\max_a\hat{Q}(S_{t+1},a;w^-)

损失为:

L(w)=E[(ytQ^(St,At;w))2]L(w)=\mathbb{E} \left[ \left(y_t-\hat{Q}(S_t,A_t;w)\right)^2 \right]

致命三角

以下三项同时出现时可能导致发散:

  1. 函数近似。
  2. 自举更新。
  3. 异策略训练。

这被称为 deadly triad。目标网络、双重 Q-learning、梯度裁剪、保守的学习率和更合理的数据分布,都是常见缓解手段,但不能替代对训练稳定性的持续监控。

评估要点

除平均回报外,还应检查 TD 误差分布、Q 值尺度、目标网络漂移、不同随机种子的方差以及 replay buffer 中的数据覆盖。函数近似带来了泛化能力,也让强化学习从求解表格固定点转变为不断移动的数据分布与优化目标之间的耦合问题。

Related Posts