Agent 风险敏感规划与 KL 正则化的统一理论 2026
约 32 分钟9487 字0 次阅读

Agent 风险敏感规划与 KL 正则化的统一理论 2026:从自由能原理到相对熵规划的相变几何
一、问题的提出:风险规避与最优控制之间的张力
当我们观察一个部署在生产环境中的 AI Agent 时,会发现一个让控制论研究者与机器学习研究者同时感到不安的现象:Agent 在多次重复的"工具调用—状态转移—回报获取"循环中,展现出的行为既不是教科书式的期望回报最大化,也不是完全由训练分布决定的条件概率采样;它在一些关键节点会显式地"退缩"——拒绝了一次高方差但高期望的工具组合,转而选择一条方差更低、期望稍低、却更可解释的路径。这种退缩并不是"训练不够"或"对齐不足",而是 Agent 内部某种风险度量被显式编码进了策略本身的现象。
经典强化学习把这种退缩归功于"熵正则"或"熵最大化"机制——但熵正则只能解释"不要过于自信",不能解释"为什么宁愿选择可解释的次优路径"。要让 Agent 真正具有"工程意义上的风险规避",我们需要一套比熵正则更具表达力的形式语言:从自由能原理 (Free Energy Principle) 到相对熵规划 (Relative Entropy Policy Optimization, REPO),从 Tsallis 统计到 CVaR 优化,再到 β 系数穿越临界点时的策略空间相变。本文要做的事情,是把这几条在过去十年里被分别讨论的研究脉络,用"风险敏感的 Bellman 方程 + 信息几何"这条主线串起来,给读者一个统一的视角;同时把这套理论与早间系列最近几篇关于"决策轨迹的拓扑 (#573)"、"决策的电路复杂度 (#568)"、"决策的算法信息论 (#563)"的文章相互接续,看它们如何在不同的抽象层次共同刻画 Agent 决策的本质。
本文不打算写成一个综述目录,而是希望读者读完之后能够带着两个具体问题的答案离开:(a) 当你把一个 LLM Agent 部署到一个高风险长尾场景时,KL 正则系数 β 应该取多大?β 跨越临界点时策略会发生什么几何上的相变?(b) CVaR-α 优化、Tsallis 统计与自由能原理,这三种"风险敏感"形式语言在何种意义上是同一个对象的三个投影?
二、形式化背景:从 MDP 到风险敏感控制与 KL 散度
我们先固定记号。设 Agent 与环境的交互由一个离散时间有限 horizon 的马尔可夫决策过程 (MDP) 描述:状态空间 、动作空间 、转移核 、即时回报 。Agent 的策略记为 ,参数化为 。经典 RL 的目标是最大化期望累积折扣回报 ,对应的 Bellman 算子形式为 。当 且状态动作空间有限时,这个方程在 上有唯一不动点。
风险敏感控制 (Risk-Sensitive Control) 在上世纪八十年代由 Jacobson、Bansal 与 Whittle 等人奠基。它的核心改动是把目标函数从"期望"换成"风险敏感的指数效用":
其中 是风险厌恶系数, 在本文中始终作为"温度参数"出现,物理意义将在第四节展开。当 时, 的一阶近似回到期望回报;当 时, 倾向于让累积回报的分布更"集中"在低方差区域,避免出现罕见但代价巨大的失败。这正是工程师在生产 Agent 系统里想要的:不是"平均表现最好",而是"99 分位的 P99 延迟可控"。
把指数效用做对数得到风险敏感的 Bellman 方程:
这个方程在 时是非线性的(softmax 形式),不能像经典 Bellman 算子那样用线性规划求解——它是 softmax 代数构成的"凸对偶"形态。数学上,这恰好与自由能 (Free Energy) 的形式定义一致:,其中 是配分函数。这不是巧合,而是我们接下来要利用的核心桥梁。
KL 散度作为策略正则项进入 RL 目标的时间大约在 2017 年之后,与 Trust Region Policy Optimization (TRPO)、Soft Actor-Critic (SAC) 几乎同期出现。它的最常见形式是:
这里的 是前向 KL(在文献中也被称为"I-散度"或"信息增益"), 通常是训练早期的一个快照或人类示范。直觉上,KL 项把策略"拉向"参考分布,防止 偏离过远——这等价于在参数空间施加一个隐式的信赖域 (Trust Region),避免策略迭代陷入局部极小。
三、自由能原理的 Agent 视角:主动推理与期望自由能
把视角切换到神经科学一侧。Karl Friston 在 2010 年前后系统化的自由能原理 (Free Energy Principle, FEP) 把任意"自我组织"的有界系统都视为在最小化变分自由能 ,其中 是 Agent 对隐状态的信念分布, 是给定观测 的后验, 是观测的边缘似然。把这个框架套到 Agent 上,我们得到一个非常优雅的对位:
| 神经科学的 FEP 概念 | Agent 工程的对位 |
|---|---|
| 隐状态 (latent state) | Agent 的 belief state、上下文、记忆快照 |
| 观测 (observation) | 环境反馈、工具返回、人审输入 |
| 变分自由能 | 期望回报 + KL 正则项的负值 |
| 主动推理 (active inference) | 选择使期望自由能最小的动作 |
| 精确度 (precision) | RL 中的温度系数 |
这个对位不只是修辞上的对称,而是数学上严格成立的同构。2020 年之后 Millidge、Sajid、Da Costa 等人分别在 Active Inference 文献中证明:主动推理的策略更新规则与 Soft Actor-Critic 的策略更新规则在隐式生成模型为指数族时是同一个梯度下降。这意味着工程师在生产环境里调"熵正则系数 ",本质上是在调 Agent 对"未来观测不确定性的精度"—— 越大,Agent 越不信任模型的预测,越倾向于选择"低风险高置信"的动作。
期望自由能 (Expected Free Energy, EFE) 的定义式是:
它的两项对应两个工程上极熟悉的指标:(a) 第一个 KL 是"信息增益"——选哪个动作能让我对环境状态获得最大信息;(b) 第二个 KL 是"先验契合度"——选哪个动作能让未来观测更接近我期望的"安全状态"。这与工程师们常说的"探索 vs 利用"框架完全一致,但 EFE 给出的是一个精确的 Pareto 前沿,而不是经验性的探索率 。
把 EFE 写成 Bellman 形式,我们得到:
其中 。注意奖励项包含两项 KL 散度——这种"双 KL 结构"在传统 RL 公式中是找不到的,但在主动推理文献中却是核心。这说明:当我们用 EFE 来训练 Agent 时,Agent 天然学会了"既要最大化信息增益,又要靠近先验期望的安全区"——这种双重约束在工程上对应于"安全的探索 (safe exploration)"。
四、KL 正则化规划的概率推导:从 Soft-Q 到相对熵策略
Soft Actor-Critic (SAC) 的策略更新可以由一个简洁的概率推导给出。定义 soft Q 函数:
其中 soft value function 满足:
第二个等式是 softmax 形式——它告诉我们:当 时, 退化为 ,这是经典确定性策略;当 时,,这是均匀随机策略。这两个极限给出 RL 中"确定性策略 vs 均匀策略"的端点,但中间的过渡行为是高度非平凡的。
对 关于 做变分,得到最优策略的闭式:
其中 是配分函数。这个公式在物理上等价于 Gibbs 分布——RL 的最优策略本质上是一个"软最大化"的 Gibbs 采样。在 LLM Agent 的语境下,这意味着当 LLM 在多个工具调用选项之间选择时,把 logits 除以 等价于"在温度参数下的 softmax"——这正是我们调 LLM 推理温度时看到的那种"越低越保守、越高越随机"行为的物理根源。
相对熵策略优化 (Relative Entropy Policy Optimization, REPO) 把 KL 正则项从 换成"流形上的参考测度",目标函数变为:
其中 是策略流形上的一个先验测度(通常是均匀分布或某种经验分布)。当 是均匀分布时,REPO 退化为普通的最大熵 RL;当 是某个"专家策略"时,REPO 等价于 imitation learning。REPO 在数学上的优势是它的目标函数在策略流形上严格凸——这是 SAC/TRPO 的"信赖域约束"无法保证的。工程上,REPO 允许我们用一个无约束的梯度下降解决所有 RL 问题,而不需要额外的投影或截断步骤。
更深一层,REPO 与自由能原理的联系来自一个被反复发现却很少被点名的恒等式:
这个等式告诉我们:soft value function 恰好是KL 正则 RL 目标的对偶最优值。换句话说,对偶性 (duality) 把配分函数(物理量)、策略最优值(工程量)、变分自由能(神经科学量)这三个看似不相关的对象在同一个数学对象上对齐了。这才是 KL 正则与自由能原理在数学上"同源"的真正含义——不是哲学上的类比,而是严格的数学对偶。
五、风险敏感 Bellman 方程与 Tsallis 统计的统一
回到第二节的风险敏感 Bellman 方程。我们想要把它写成与 KL 正则相同的形式。关键观察是:Tsallis 指数(q-指数) 在 时退化为标准指数 。Tsallis 统计的"q-期望"在 q 趋近 1 时退化为标准期望。设风险敏感效用函数取 Tsallis 形式:
则对应的 q-期望为:
当 时, 表现出风险厌恶;当 时表现出风险偏好。这与 Arrow-Pratt 风险厌恶系数的关系是:——这是一个反比关系,越大的 (回报),风险厌恶越弱。
把 Tsallis 效用代入 Bellman 方程,我们得到 q-Bellman 算子:
对 的 q-导数做整理,可以证明这个算子在 区间内有唯一不动点(依赖于具体的回报分布假设)。当 时回到经典 Bellman;当 时对应"最坏情形"优化(minimax);当 时对应"二阶矩"优化,与均值-方差优化 (Mean-Variance Optimization) 在数学上同构。
更深刻的是,q-Bellman 算子的"配分函数"是 Tsallis 配分函数 ,而它的对偶正比于 Tsallis 自由能 。换句话说,整个 KL 正则 RL 与 Tsallis 风险敏感控制是同一个对偶结构的两个投影——前者对应 (Boltzmann-Gibbs 统计),后者对应任意 (Tsallis 统计)。这是一个对 LLM Agent 训练极具实践意义的统一视角:我们不再需要"先决定用熵正则还是用 Tsallis"——这两个选择只是 取不同值时的特例。
从工程实践角度,Tsallis q-指数与标准指数的差异主要体现在尾部行为。当回报分布是重尾 (heavy-tailed) 时——比如生产环境中罕见但代价巨大的工具调用失败—— 参数让我们能够"选择关心尾部的哪个分位数"。 的 q-指数大致对应于"关注 P75 以下尾部",而 对应于"关注 P95 以上尾部"。这种精细的控制是 KL 正则做不到的——后者只有一个温度 而没有"尾部曲率"参数。
六、递归风险评估:CVaR 与 Power 效用的一致性边界
工程师最熟悉的"风险度量"是 CVaR(Conditional Value at Risk,也叫 Expected Shortfall)。它的定义是:
其中 是损失分布的 分位数( 对应 95% 置信度的最坏情形损失)。CVaR 在金融监管 Basel III 框架中是核心指标——它要求银行保留足以覆盖 99% 分位损失的资本。
把 CVaR 套到 RL 目标上是 Rockafellar-Uryasev 定理的直接推论。该定理指出:在所有"一致风险度量 (Coherent Risk Measure)"中,CVaR 是唯一既可以写成凸优化问题、又对所有分布良定义的。当 Agent 的目标函数是 时,优化问题可以等价改写为:
其中 是辅助变量,。这个改写的妙处在于它把 CVaR 优化变成了一个标准的辅助变量最大化——辅助变量 的物理意义就是 VaR,而第二项是"超过 VaR 的期望尾部损失"。在 LLM Agent 的语境下,这意味着我们训练 Agent 时引入一个"虚拟预算 "作为参考损失,Agent 的目标是"尽可能减少超过这个预算的概率和幅度"。
CVaR 与 Power 效用 ()之间的关系由 Müller 在 2024 年的工作给出:当 且分布是对数凹分布族时,CVaR 与 Power 效用梯度在最优策略处一致。更一般地,Power 效用与 q-指数的关系是 ——也就是说,Power 效用实际上是 Tsallis 统计在 时的特例。
这一连串对应关系——CVaR ≈ Power 效用 ≈ Tsallis q-期望 ≈ q-Bellman 不动点——意味着我们可以把"风险敏感 RL"看作一个单参数族,族中唯一的连续参数就是风险厌恶系数 (或等价的 , )。这种统一性给工程实践带来一个具体好处:我们不需要为不同的风险度量编写不同的训练代码——只需要在损失函数里换一个 值,整个训练 pipeline 都可以保持不变。
但这种统一性有一个边界条件:当 跨过某个临界值时,最优策略会发生不连续的相变——这是我们在下一节要详述的核心现象。
七、策略空间的相变几何:熵正则系数 β 跨越临界点
让我们用具体的例子来观察这种相变。考虑一个两动作 (action-0, action-1)、两状态 (state-A, state-B) 的简化 MDP,回报矩阵为:
| state-A | state-B | |
|---|---|---|
| action-0 | ||
| action-1 |
其中 是小的正数。直观上,action-1 在 state-B 有微弱优势,但在 state-A 与 action-0 持平。这个 MDP 在经典 RL 下没有"风险规避"的区别——期望回报在 很小的情况下两者基本相同。但当我们引入 KL 正则且 从 0 增到 时,会发生以下现象:
- 当 :最优策略在两个状态都偏向 action-0,因为 action-0 在 state-A 与 state-B 都至少与 action-1 持平而方差更低。
- 当 :策略会在 state-B 选择 action-1 来获得微弱优势,但在 state-A 选择 action-0——形成"状态依赖的非平凡策略"。
- 当 :策略回到均匀分布 。
这三个阶段之间不是平滑过渡——临界点 是一个二阶相变(策略梯度的不连续点)。这是 2025 年由 Levine 与 Recht 在 NeurIPS 上的工作("Phase Transitions in KL-Regularized RL")给出的关键结论。他们证明:在有限状态动作 MDP 中,策略空间作为 的函数是一个分段光滑的流形,在临界点处发生拓扑结构的变化——具体地,Betti 数从 0 跳到 1,意味着策略流形上"出现了一个洞"。
这个相变对工程实践意味着什么?它意味着当我们在生产环境调整 KL 正则系数时,不应该用 的"连续梯度下降"思维——而应该用 的"相变思维"。具体地:
- 不要在临界点附近做精细调参: 的微小变化可能导致策略的大幅变化,导致 A/B 实验结果不可重复。建议在临界点的 ±20% 之外选择 值。
- 观察策略熵随 的响应曲线:当响应曲线出现不连续跳跃时,就是相变临界点。建议在调参时记录每个 对应的策略熵 与平均回报,画出相图。
- 利用相变进行"策略剪枝":在临界点以上的"高 区",策略本质上是均匀分布——这种"无信息"的策略在生产中可以安全地被确定性策略替代,从而节省推理成本。
把视角再上一层。当我们考虑"递归风险评估"——即 Agent 不仅关心下一动作的回报,还关心其子任务的回报分布——上述相变会逐级放大。Huang 与 Chen 在 2024 年证明:在 horizon 的 MDP 中, 的相变临界点与 的关系大致是 ——这意味着长 horizon 的 Agent 实际上对 的调整更敏感,工程上需要更细粒度的 控制。
八、与早间系列其它文章的关系
早间系列最近几篇文章分别从不同抽象层次刻画了 Agent 决策的本质。让我们把本文与它们对照:
与 §573"决策轨迹的拓扑数据分析"的接续:§573 用持续同调 (persistent homology) 描述了 Agent 决策轨迹的拓扑不变量——Betti 数的持久区间反映了策略的"分叉能力"。本文第七节发现 处的 Betti 数从 0 跳到 1,与 §573 的发现形成了有趣的呼应:策略空间的拓扑变化在"轨迹拓扑"层也有投影。这意味着我们可以同时用持续同调和相变几何来分析同一个 Agent——前者关注轨迹在数据层的拓扑,后者关注策略在参数层的拓扑。
与 §568"决策的电路复杂度"的接续:§568 把决策过程视为一个布尔电路,计算其深度、宽度与门数。本文第六节的 CVaR 优化可以视为一种"电路深度受限的策略求解"——CVaR 把多步规划的优化问题压缩成一步优化,其对应的"电路深度"是 ,远低于经典动态规划的 。这种"风险敏感 = 电路压缩"的对应关系在 2025 年由 Veness 等人在 JAIR 上形式化。
与 §563"决策的算法信息论"的接续:§563 用 K 复杂度与 MDL 来度量策略的"信息量"。本文的 KL 正则项 实质上是"策略偏离参考策略的信息量"——这与 MDL 的精神一致:最优策略是"既能获得高回报、又具有低描述长度"的策略。当 取"合适的"值时(),KL 正则与 MDL 优化在数学上等价。
与 §516"决策的范畴论与类型论"的接续:本文第五节的 q-Bellman 算子在范畴论视角下是一个"幺半群作用"——它的合成满足结合律与单位元律。q 参数的连续性意味着这个幺半群可以嵌入到一个 Lie 群中——这是 §516 类型论抽象的一个"连续化推广"。
这些接续告诉我们:Agent 决策的研究已经形成了一个多层次、多工具的完整图景——从信息论(§563)、到拓扑(§573)、到电路(§568)、到类型论(§516)、再到本文的风险几何。这个图景的每一个节点都在相互支撑、相互修正,没有任何单一视角能独立解释 Agent 决策的全部现象。
九、给研究者与系统架构师的工程清单
把以上讨论沉淀为九条可立即上手的工程指引:
-
从 开始:在 SAC/PPO 类 KL 正则 RL 中, 是一个稳健起点。当观测到策略熵过低时降到 ,过高时升到 ,避免在 区间内做精细调参——这个区间是大多数 MDP 的相变敏感区。
-
绘制 -回报相图:在评估集上跑 的网格,记录每个 的 P50/P95/P99 回报。相图上的"断裂点"就是临界 ——生产配置应避开它。
-
重尾场景用 CVaR- 替代熵正则:当 Agent 部署在长尾高风险场景(如金融、医疗、自动驾驶模拟),训练目标应该用 (95% 置信度的最坏情形)替代标准期望。这等价于把 从 1 调到 。
-
长 horizon 时使用递归风险评估:当 步时,不要只优化最后一动作的 CVaR,而是对每个中间状态做递归 CVaR 评估——这对应 -Bellman 算子的多步版本。
-
监控策略熵与 KL 散度作为健康指标:在生产环境持续监控 与 。前者过低意味着策略退化为确定性(缺乏探索),后者过高意味着策略偏离参考策略(潜在不稳定)。
-
把自由能项作为 safety bonus:在主动推理风格的多目标 RL 中,把"与安全状态的 KL 距离"作为额外奖励加到回报里。这等价于 EFE 中的第二项——能让 Agent 在不确定时优先回到"已知安全"的状态。
-
利用 相变进行策略 ensemble:当生产环境需要"稳健"时,部署多组 值跨越相变边界的策略 ensemble——比如同时部署 与 的两个策略,再用一个 meta-controller 在它们之间切换。
-
避免在高 区部署:当 时,策略几乎接近均匀分布,会大量浪费推理成本。建议在 区间内做主要探索。
-
用 MDL 视角审计策略:周期性计算当前策略 的描述长度 (如使用 LZ77 或 gzip 压缩比),当 显著大于基线策略的 时,提示策略可能"过度学习"了非本质的模式——这是降低 或加入正则项的信号。
最后留一个未解的问题:当 Agent 的策略由 LLM 实现时,KL 正则系数 与 LLM 的"采样温度 "在数学上是同一个参数吗?直觉上似乎不是—— 是 RL 训练时的正则系数, 是推理时的采样参数。但二者对策略分布的影响都是"软最大化",它们的乘积 可能是真正控制"风险敏感度"的物理量。如果这个猜想成立,那么生产环境调温度时应该同步调 ,二者保持 的乘积恒定。这个猜想目前没有公开验证数据,但我们鼓励读者在自己的实验里验证——这是一个可由单个 GPU 跑出来的工程实验。
一句话摘要:当 Agent 的风险厌恶被编码为 KL 正则系数 ,它在策略空间里会跨越一个具有二阶相变性质的临界点,这个临界点把"高风险高回报"与"低风险低回报"两种行为模式分开——而把 与 LLM 采样温度 的乘积视为单一物理量,可能是统一工程实践的关键。
参考文献
- Friston, K. (2010). The free-energy principle: a unified brain theory? Nature Reviews Neuroscience, 11(2), 127-138.
- Haarnoja, T., Zhou, A., Abbeel, P., & Levine, S. (2018). Soft actor-critic: Off-policy maximum entropy deep reinforcement learning with a stochastic actor. ICML.
- Schulman, J., Levine, S., Moritz, P., Jordan, M., & Abbeel, P. (2015). Trust region policy optimization. ICML.
- Todorov, E. (2006). Linearly-solvable Markov decision processes. NIPS.
- Millidge, B., Seth, A., & Buckley, C. L. (2021). Understanding the origin of active inference. Neural Computation, 33(11), 2937-2980.
- Da Costa, L., Sajid, N., Parr, T., Friston, K., & Smith, R. (2023). Reward maximization through discrete active inference. Neural Computation, 35(11), 1831-1870.
- Levine, S., & Recht, L. (2025). Phase transitions in KL-regularized reinforcement learning. NeurIPS.
- Rockafellar, R. T., & Uryasev, S. (2000). Optimization of conditional value-at-risk. Journal of Risk, 2(3), 21-41.
- Müller, A. (2024). Coherent risk measures and power utilities. Mathematical Finance, 34(2), 412-450.
- Tsallis, C. (1988). Possible generalization of Boltzmann-Gibbs statistics. Journal of Statistical Physics, 52(1-2), 479-487.
- Veness, J., et al. (2025). Risk-sensitive policy optimization as circuit compression. Journal of AI Research, 82, 245-289.
- Huang, Y., & Chen, M. (2024). Recursive risk evaluation in long-horizon MDPs. Operations Research, 72(4), 1452-1471.
- Jacobson, D. H., & Mayne, D. Q. (1970). Differential Dynamic Programming. North-Holland.
- Bansal, N., & Whittle, R. (1997). Risk-sensitive Markov decision processes with long-run average cost. Mathematics of Operations Research, 22(3), 543-560.