RLVR 中采样器失配下熵漂移的符号定律
A Sign Law for Entropy Drift under Sampler Mismatch in RLVR
Zhiyuan Wang · Jacob Steeves · Chang Liu
中文摘要
在可验证奖励强化学习(RLVR)中,熵的变化取决于本应中立的设置,如超长响应是掩码还是计为失败。这源于采样器失配,即 rollout 使用 top-k/top-p 截断或学习者未应用的温度。证明表明,常数优势偏移会在一阶期望上改变 softmax 策略的熵,符号由采样器决定:在似然比随 token 概率增加的采样器(如 top-k, top-p, min-p, 温度<1)下,正偏移使策略变锐;在更平坦的采样器下则变平。在基于组的 RLVR 中,基线控制的漂移由 token 加权净优势决定,掩码与失败处理可使其符号相反;剩余部分为优势与采样器锐度差的协方差。在 2B 模型上,截断采样下熵斜率与净优势偏移线性相关,精确采样下不可区分,token 级重要性比率保留 98% 的漂移。100 步运行中,零净优势基线仅延迟精确采样未显示的熵爆炸,固定正偏移进一步延迟,锐度差加权基线在不改变学习者的情况下保持熵稳定。在 9B 推理模型 10 步中,未校正规则降低留出集准确率,而零化净优势或重归一化学习者分别提升 8.2 和 8.8 分,精确采样下无影响。
关键要点
- 01指出 RLVR 中熵漂移源于采样器失配,如 top-k 截断或温度不匹配。
- 02证明常数优势偏移改变熵的符号由采样器决定,正偏移在锐采样器下使策略变锐。
- 032B 模型实验显示截断采样下熵斜率与净优势线性相关,精确采样下无漂移。
- 049B 模型实验表明零化净优势或重归一化可提升留出集准确率 8.2 至 8.8 分。
解读
在 GitHub 查看内容由 AI 生成,仅供参考。如发现错误,欢迎反馈。
原始英文摘要
arXiv:2610.34044v1 Announce Type: new Abstract: Entropy in reinforcement learning with verifiable rewards (RLVR) moves up or down depending on choices that should be neutral, such as whether over-length responses are masked or counted as failures. We trace this to sampler mismatch, where rollouts are drawn with top-k/top-p truncation or at a temperature the learner does not apply, and prove that a constant advantage shift then changes the entropy of a softmax policy, at first order and in expectation, with a sign fixed by the sampler: positive shifts sharpen the policy under any sampler whose likelihood ratio to the policy increases with token probability, including top-k, top-p, min-p, and temperatures below one, and flatten it under flatter samplers. In group-based RLVR with token-level losses, the baseline-controlled part of this drift is set by the token-weighted net advantage, which masking and failing over-length responses can push to opposite signs; the remainder, a covariance between advantages and the sampler's sharpness gap, can feed back over long horizons. On a 2B model, the entropy slope is linear in shifts of the net advantage under truncated sampling and indistinguishable from flat under exact sampling, and token-level importance ratios leave 98% of the drift. In 100-step runs, a net-zero baseline only delays an entropy explosion that exact sampling does not show, a fixed positive shift delays it further, and a baseline weighted by the sharpness gap keeps entropy stable without changing the learner. On a 9B reasoning model over ten steps, the uncorrected rule loses held-out accuracy, whereas zeroing the net advantage or renormalizing the learner gains 8.2 and 8.8 points over it, with no held-out effect under exact sampling.
1. 问题与价值
这篇论文解决的是强化学习可验证奖励(RLVR)训练中一个隐蔽但影响巨大的问题:策略熵的漂移方向,会被采样器与学习器之间的不匹配所决定。具体来说,当推理引擎用 top-k、top-p、min-p 截断采样,或者用学习器未采用的温度采样时,一个在精确采样下无关紧要的常数优势平移,会在截断采样下以固定符号推动熵上升或下降。
为什么重要:熵崩溃会导致策略停止探索、准确率平台化;熵爆炸会导致回复质量下降甚至训练发散。而触发这两种相反灾难的开关,可能只是一个看似与熵无关的实现选择——比如对超长回复是「掩码」还是「判失败」。
谁受益、怎么受益:所有跑 RLVR 训练大模型的团队。他们可以直接获得一个诊断量——token 加权净优势
N,以及一个修正方案——在学习器侧重归一化采样支持集或中心化分数。论文在 9B 模型上显示,仅把N清零,10 步后留出集准确率就比未校正规则高 8.2 个百分点;重归一化学习器高 8.8 个百分点(原文只报相对未校正规则的点差,没有给绝对准确率)。这意味着训练团队可以避免因采样器不匹配而浪费整轮训练算力。2. 已有工作的不足
在本文之前,最好的相关做法分几类:
具体哪一步不够:缺少一个把「采样器相对于策略的锐度」与「熵漂移方向」联系起来的可证明符号定律,也缺少把 group-based RLVR 中组内优势求和为零这一事实与「常数平移从何而来」连接起来的分解。
3. 重建作者的思考路径
从旧方法出发。on-policy 理论说,常数优势平移对熵没有一阶效应,因为
E_{a∼π}[Φ(a)] = 0,其中Φ(a)是熵极性。但实践中掩码和判失败明明把熵推向相反方向。如果是我面对这个矛盾,第一步会怀疑:是不是采样分布不是 π?先检查最简单的偏离:截断采样。假设学习器在 softmax 参数化下做普通梯度步,采样 token
a带优势A,梯度是A(e_a − π)。熵对 logits 的偏导是−u_v,其中u_v = π_v(log π_v + H)。于是单步熵变化是ΔH = −ηAΦ(a),其中Φ(a) = u_a − E_{v∼π}[u_v]。在精确采样下E_{a∼π}[Φ(a)] = 0,这解释了旧理论。但在采样器μ下,E_{a∼μ}[ΔH] = −ηc δ(μ),其中δ(μ) = E_{a∼μ}[Φ(a)]。到这里会撞上第一个问题:
δ(μ)的符号是什么? 如果它没有固定符号,那常数平移的方向就无法预测。我会先试最简单的截断:截断到最可能 token 的集合S,重归一化。这时δ = π(S^c)(E_π[u∣S] − E_π[u∣S^c])。直觉上,S里的 token 概率更高,而u_v = π_v(log π_v + H)在π_v ≥ e^{−H}时非负。所以S里的u均值应该大于补集里的均值,δ > 0。这给出一个猜想:更锐的采样器有正 gap,更平的采样器有负 gap。但「更锐」怎么定义?如果只是截断,那很好办。但温度采样、min-p、top-p 的序列组合怎么办?这里需要一个能覆盖所有情况的定义。一个自然的想法是看似然比
μ_v/π_v:如果这个比值随π_v单调递增,那μ比π更锐;单调递减则更平。这个定义下,截断、温度小于 1、min-p 都是更锐;温度大于 1、与均匀分布混合都是更平。然后可以用层蛋糕表示把任意更锐采样器写成截断的混合,从而把符号推广到一般情况。到这里,单位置的符号定律有了。但 group-based RLVR 里每个组的优势求和为零,常数平移从哪来? 这是第二个关键分叉。我会先写出 token 级损失:
J(θ) = Σ_i ω_i A_i Σ_t log π_θ(y_{i,t} ∣ s_{i,t})。每个响应i的 token 数L_i不同,所以响应i对梯度的贡献被ω_i L_i加权。那么组内优势的加权平均——token 加权净优势N_g = Σ_i ω_i L_i A_i / Σ_i ω_i L_i——就充当了那个常数平移。这个量在组均值基线下等于Cov(r, L) / L̄,因为长响应更容易失败,所以判失败时N_g < 0,掩码时N_g > 0。这解释了图 1d 的现象。但这里会撞上第三个问题:
N = 0时熵为什么还在涨? 100 步实验里 FAIL+NZ 只是把熵爆炸推迟了 17–18 步。这说明还有第二个不依赖基线的项。回头看分解,除了D_net还有D_cov = −η Σ_g W_g Cov_g(A, δ̄)。这个协方差项来自失败响应访问更不自信的前缀,所以 gap 更大。它不依赖基线,且会自我反馈:如果策略被推离更锐的采样器,丢弃质量x增大,而ẋ ∝ x²(1−x),在x < 2/3时加速。这就是 Proposition 2 的非对称反馈。最后的分叉是:怎么修? 清零
N只去掉一项;GAPW 基线让两项对消;学习器侧重归一化等于分数中心化,直接消除整个不匹配漂移;而 token 级重要性比率只是把漂移乘以保留质量π(S),符号不变,所以几乎无效。与论文实际路径的差异:我猜的路径是从「截断的 gap 符号」直接跳到「一般采样器的层蛋糕表示」,论文确实走了这条路。但我没有预料到他们会把
D_cov的反馈做成一个可精确求解的两层模型(Proposition 2),并从中推出「推离更锐采样器在有限时间内起飞、拉向它则饱和」这种非对称动力学。我原以为D_cov只是一个需要数值估计的次要项,但论文把它提升为解释 100 步熵爆炸的核心机制。另外,我也没有想到他们会把「支持集重归一化等于分数中心化」这个恒等式作为 Proposition 3c 单独列出,并用数值实验验证到浮点精度——这更像是一个工程洞察被提升为理论陈述。4. 核心 intuition
用一个日常类比:一个只在高分学生里抽样的老师,会把「表扬」和「批评」变成两种完全不同的教学动作。
想象一个班级里学生水平参差不齐。老师有两种抽样方式:一种是从全班随机抽学生回答问题(精确采样);另一种是只从成绩最好的前 20 名学生里抽(截断采样,更锐的采样器)。
现在老师决定给所有回答正确的学生加一个固定的奖励分数
c(常数优势平移)。在随机抽样下,这个固定奖励对所有学生一视同仁,长期来看不会改变班级的成绩分布——好学生和差学生被抽到的概率一样,固定奖励不会系统性地偏向任何一方。但在只抽前 20 名学生的模式下,情况完全不同。如果
c > 0(正平移),老师每次抽到的都是好学生,给他们加分,这会让好学生的成绩更高,班级分布变得更集中、更「锐」——熵下降。如果c < 0(负平移),老师每次抽到的都是好学生却给他们扣分,这会把好学生往下拉,班级分布变得更平均——熵上升。这个类比对应到机制上:截断采样器
μ只在最可能 token 的集合S上重归一化,所以常数平移c的期望梯度是c(μ − π)。当c > 0,梯度把 logits 推向μ,即把概率质量更集中到S上,熵下降;当c < 0,梯度把 logits 推离μ,概率质量从S外溢,熵上升。而「更锐」的定义——似然比μ_v/π_v随π_v递增——正是说采样器比策略更集中在大概率 token 上,所以这个机制对 top-k、top-p、min-p、低温采样都成立。5. 方法 pipeline 举例
编一个具体输入:一个数学题 prompt
x,组大小G = 8,采样器是 top-k 20 加 top-p 0.95,学习器温度 1。假设 8 个响应中 6 个正常完成,2 个达到生成长度上限(超长)。第一步:采样与奖励。 8 个响应
y_1, ..., y_8被采样。6 个完成响应中 4 个正确(r = 1),2 个错误(r = 0);2 个超长响应没有可验证答案。第二步:计算基线。 组均值基线
r̄ = (4×1 + 2×0 + 2×0) / 8 = 0.5。每个响应的优势A_i = r_i − r̄:4 个正确的A = +0.5,2 个错误的A = −0.5,2 个超长的A = −0.5。第三步:应用规则。 假设用 FAIL 规则:超长响应得零奖励但仍参与训练。假设 prompt-mean 聚合,所有 token 共享权重
ω_i ∝ 1/Σ_j L_j。设 6 个完成响应的长度分别为 200、300、250、400、350、280 token,2 个超长响应都达到 2048 token 上限。总长度Σ L_j = 200+300+250+400+350+280+2048+2048 = 5876。第四步:计算 token 加权净优势
N_g。N_g = Σ_i L_i A_i / Σ_i L_i。分子:200×0.5 + 300×0.5 + 250×0.5 + 400×0.5 + 350×(−0.5) + 280×(−0.5) + 2048×(−0.5) + 2048×(−0.5)。先算前四项:100 + 150 + 125 + 200 = 575。后四项:−175 − 140 − 1024 − 1024 = −2363。分子575 − 2363 = −1788。N_g = −1788 / 5876 ≈ −0.304。这是负的,所以在更锐采样器下,净优势项D_net会推高熵。第五步:计算 gap。 对每个采样前缀
s_{i,t},学习器计算δ_{i,t} = E_{μ}[u] − E_{π}[u]。假设超长响应的前缀平均 gap 是完成响应的 1.5 倍(论文 4.2 节报告了这个比例)。完成响应的平均 gap 约4.0×10^{−3},超长响应约6.0×10^{−3}。第六步:分解漂移。
D_net = −η W_g N_g δ̄_g。因为N_g < 0且δ̄_g > 0,D_net > 0,即熵上升。D_cov = −η W_g Cov_g(A, δ̄)。失败响应(包括超长)的优势为负,且它们的 gap 更大,所以Cov_g(A, δ̄) < 0,D_cov > 0,也推高熵。两项同向,FAIL 规则下熵快速上升。第七步:对比 MASK 规则。 如果改为 MASK:超长响应在计算基线时得零奖励,但被排除在损失之外。基线变为
r̄_M = (4×1 + 2×0) / 6 = 0.667。6 个训练响应的优势:4 个正确的A = +0.333,2 个错误的A = −0.667。N_g = Σ_{i∈M} L_i A_i / Σ_{i∈M} L_i。分子:200×0.333 + 300×0.333 + 250×0.333 + 400×0.333 + 350×(−0.667) + 280×(−0.667)。前四项:66.6 + 99.9 + 83.25 + 133.2 = 382.95。后两项:−233.45 − 186.76 = −420.21。分子382.95 − 420.21 = −37.26。分母200+300+250+400+350+280 = 1780。N_g = −37.26 / 1780 ≈ −0.021。但论文指出 MASK 还额外加一项(n_mask/G) r̄_M = (2/8)×0.667 ≈ 0.167,所以N_g ≈ −0.021 + 0.167 = +0.146。符号翻转为正,D_net < 0,熵下降。这与论文报告的 2B 上 MASK 的N ≈ +0.30、9B 上N ≈ +0.16方向一致(具体数值因数据分布不同而异)。6. 核心数学推导
目标:推导单位置熵变化公式(Equation 3)和期望下的符号定律(Equation 4)。
符号解释:固定前缀
s,logits 为z,策略π = softmax(z),词汇表V。熵H = −Σ_v π_v log π_v。定义u_v = π_v(log π_v + H)。采样 tokena带优势A,学习率η。第一步:熵对 logits 的偏导。 由 softmax 性质,
∂π_w/∂z_v = π_w(1[w=v] − π_v)。于是:∂H/∂z_v = −Σ_w (log π_w + 1) ∂π_w/∂z_v展开:
= −Σ_w (log π_w + 1) π_w(1[w=v] − π_v)= −π_v(log π_v + 1) + π_v Σ_w π_w(log π_w + 1)= −π_v(log π_v + 1) + π_v(1 − H)= −π_v(log π_v + H) = −u_v第二步:梯度步对 logits 的改变。 损失是
A log π_a,其对z的梯度是A(e_a − π),其中e_a是 tokena的 one-hot 向量。所以 logits 更新为Δz = ηA(e_a − π)。第三步:熵的一阶变化。 用链式法则:
ΔH = ⟨∂H/∂z, Δz⟩ + O(η²)= −ηA Σ_v u_v(1[v=a] − π_v) + O(η²)= −ηA(u_a − Σ_v π_v u_v) + O(η²)= −ηA(u_a − E_{v∼π}[u_v]) + O(η²)定义
Φ(a) = u_a − E_{v∼π}[u_v],得到 Equation 3:ΔH = −ηA Φ(a) + O(η²)。第四步:精确采样下的期望。
E_{a∼π}[Φ(a)] = E_{a∼π}[u_a] − E_{v∼π}[u_v] = 0。所以常数优势c在精确采样下对熵的一阶期望效应为零。第五步:采样器不匹配下的期望。 设采样器为
μ,常数优势为c:E_{a∼μ}[ΔH] = −ηc E_{a∼μ}[Φ(a)] + O(η²)= −ηc(E_{a∼μ}[u_a] − E_{v∼π}[u_v]) + O(η²)定义 sharpness gap
δ(μ) = E_{a∼μ}[u_a] − E_{v∼π}[u_v],得到 Equation 4:E_{a∼μ}[ΔH] = −ηc δ(μ) + O(η²)。第六步:符号定律(Theorem 1 的证明思路)。 对截断到上集合
S的采样器q_S = π 1_S / π(S):δ(q_S) = E_{q_S}[u] − E_π[u]= (1/π(S)) Σ_{v∈S} π_v u_v − E_π[u]= (1/π(S))(E_π[u] − π(S^c) E_π[u∣S^c]) − E_π[u]= π(S^c)(E_π[u∣S] − E_π[u∣S^c])因为
S是上集合,S里的 token 概率不低于补集里的。而u_v = π_v(log π_v + H)在π_v ≥ e^{−H}时非负。通过比较S与嵌套集合{v : π_v ≥ e^{−H}},可以证明E_π[u∣S] > E_π[u∣S^c],所以δ(q_S) > 0。对一般更锐采样器,用层蛋糕表示μ = ∫ π(S_s) q_{S_s} ds,其中S_s = {v : μ_v/π_v ≥ s}是上集合,δ对μ线性,所以δ(μ) > 0。更平采样器同理,符号翻转。一句话总结:这个推导在算「常数优势平移
c在采样器μ下对策略熵的一阶期望改变」,结果是−ηc δ(μ),其中δ(μ)的符号由采样器相对于策略的锐度决定——更锐则正,更平则负。7. 实验设计
数据集:MATH 训练集 7,496 道题,math-verify 判分。750 题留出,其中 300 题构成留出评估集,64 题构成熵探针集。
模型:Qwen3.5-2B(非思考模式)、MiMo-V2.6-Distill-Qwen-9B、OLMo-3-7B-Instruct 作为第二模型家族。
基线:默认训练器用组大小 8、组均值基线(无标准差归一化)、prompt-mean 聚合、AdamW 学习率
10^{−6}、梯度裁剪 1、无 KL 或熵项、每批一步优化。规则包括 MASK、FAIL、FAIL+NZ、FAIL+GAPW。采样器包括截断(top-k 20 加 top-p 0.95)、top-p 0.99、top-p 0.8、精确采样、温度 0.7 和 1.2。指标:探针集熵斜率(每步 nats)、留出集准确率、固定集准确率、四个基准的平均准确率、训练批次的平均 gap。
消融:剂量-响应实验(对优势加
−λN_g平移,λ从 −1 到 2);修正方案对比(支持集重归一化、token 级重要性比率、裁剪目标内比率、学习器应用采样温度、多 minibatch PPO、SGD 对照、标准差归一化、sequence-mean 聚合);100 步长程运行;熵初始值踢高实验。实验能否支撑结论:能支撑核心符号定律和分解。剂量-响应实验直接验证了
D_net的线性性(截断下斜率−106.8±0.7 ×10^{−3}nats/步/单位N,精确采样下+1.1±2.7,与零不可区分)。跨 6 个采样器的 MASK−FAIL 差异跟随 gap 符号,温度 1.2 时翻转。100 步运行分离了反馈机制。9B 准确率实验给出了实际收益。缺什么:一是 9B 实验只有 10 步,且从已训练 20 步的未修正 fork 出发,部分收益可能是从既有漂移中恢复,而非长期稳定性的证明。二是 100 步运行中 GAPW 和 per-batch shift 的净优势会随训练增长,100 步无法区分「增长」与「水平」的贡献。三是第二模型家族 OLMo 的熵响应只有 Qwen 的约八分之一,说明「漂移到熵的转换率」依赖模型,但论文没有系统研究这个依赖的来源。四是所有实验都在数学推理任务上,其他 RLVR 任务(代码、工具调用)的泛化性未验证。
8. take-aways
N决定。9B 批次上D_net占 FAIL 和 PEN 规则下不匹配漂移的 66–78%(三个批次分别为 76%、78%、66%)。N不够:100 步运行中 FAIL+NZ 只是把熵爆炸从 FAIL 的第 17 步推迟到第 34–35 步,最终熵仍达 4.5–7.3 nats。gap 协方差项D_cov会自我反馈,推离更锐采样器时在有限时间内起飞(Proposition 2)。π(S),符号不变。实验中保留质量均值 0.99,所以留下 98% 的漂移;四个裁剪 minibatch 更新反而把差异放大七倍。9. 最脆弱的假设
最脆弱的假设是:一阶 softmax 参数化分析能迁移到训练好的神经网络在 Adam 优化器下的实际行为。
论文的分析全部在 softmax 参数化、普通梯度步、一阶近似的框架下进行。但实际系统用的是 AdamW,其对角预条件器会改变每个参数的更新方向。论文自己承认「Adam 的对角预条件器可以在个别前缀改变符号」(Section 3.3 末尾),并且在剂量-响应实验中发现 Adam 下响应在大漂移处饱和,而 SGD 下响应与漂移成正比(Section 4.2)。这说明 Adam 引入的非线性是一阶理论没有覆盖的。
论文给出的证据:一是纯平移探针在 Adam 和 SGD 下符号相同(Table 5),二是 35 个运行的熵斜率与 logged drift 相关性 0.97(Figure 5),三是 9B 准确率实验在真实 Adam 训练下复现了理论预测的方向。但这些证据都是「符号一致」层面的,不是「幅度可预测」层面的。如果 Adam 的预条件器在某个特定权重初始化或训练阶段系统性地翻转了符号,那么符号定律的实践指导价值就会大打折扣。论文没有给出 Adam 下符号翻转的系统性研究,只在一个段落里提到这种可能性。
10. 一周最小复现
目标:复现最核心的结论——截断采样下 MASK 与 FAIL 的熵斜率差异跟随 gap 符号,且支持集重归一化消除该差异。
硬件:1 张 24GB 消费级显卡(如 RTX 4090 或 A5000)。
模型:Qwen3.5-2B 或更小的 1.5B 模型(如 Qwen2.5-1.5B-Instruct),非思考模式。
数据:从 MATH 训练集抽 2,000 题,留出 100 题做熵探针(每步前对 64 个固定 prompt 各采样 1 个响应,teacher forcing 计算全 softmax 熵)。
训练配置:组大小 8,每步 8 组(共 64 个响应),生成长度上限 2048 token,AdamW 学习率
10^{−6},梯度裁剪 1,无 KL 项,每批一步优化。采样器用 vLLM 或 Hugging Face transformers 的 top-k 20 加 top-p 0.95。运行三个 arm:MASK、FAIL、FAIL+NZ,各 15 步,每个 arm 2 个种子。预期结果:MASK 的熵斜率为负(约
−9×10^{−3}nats/步),FAIL 为正(约+12×10^{−3}nats/步),FAIL+NZ 居中(约+2×10^{−3}nats/步)。MASK−FAIL 差异约−21×10^{−3}nats/步。再加一个 arm:FAIL 加支持集重归一化(学习器对采样支持集重归一化 log 概率),15 步 2 个种子。预期 MASK−FAIL 差异降到截断值的约 4%,即约
−0.8×10^{−3}nats/步。时间分配:第 1 天搭环境和数据管线;第 2–3 天跑三个 arm(每个 arm 约 2–3 小时);第 4 天跑重归一化 arm;第 5 天分析熵斜率、计算
N和 gap;第 6–7 天写报告。如果时间充裕,加一个温度 1.2 的 arm 验证符号翻转(预期 MASK−FAIL 差异变为正,约+32×10^{−3}nats/步)。预期指标:熵斜率用最小二乘拟合探针熵对步数的斜率。
N用每步训练批次的 token 加权净优势均值。gap 用学习器对每个采样前缀计算的δ_{i,t}均值。11. 反例设计
核心主张:更锐采样器下,正优势平移降低熵,负优势平移提高熵;且这个符号由采样器的锐度决定。
反例实验:构造一个非单调似然比的采样器,它既不是更锐也不是更平,但论文的符号定律如果被过度推广,可能会预测错误。具体来说,用温度
T > 1后接截断的采样器。论文自己承认这种采样器「既不是更锐也不是更平,gap 的符号取决于 π」(Section 3.2 末尾)。判据:找一个具体的策略分布
π,使得温度 1.5 后接 top-p 0.9 的采样器μ的 gapδ(μ)为负,但常数正优势平移c > 0却降低熵(即E_{a∼μ}[ΔH] < 0)。如果这个现象在真实 2B 模型的探针集上被观测到,且幅度超过统计噪声(比如超过精确采样下效应的 6 倍,即论文报告的截断与精确采样之比),那就说明符号定律不能推广到「既非更锐也非更平」的采样器,论文的核心主张「采样器固定符号」需要加上「仅当采样器有序时」的限定。更强的反例:如果能在更锐采样器(如 top-k)下找到一个反例——正优势平移提高熵——那将直接推翻 Theorem 1。这需要构造一个策略分布,使得截断集合
S的E_π[u∣S] < E_π[u∣S^c]。但 Lemma 1 证明这在S为上集合时不可能。所以更强的反例在数学上已被排除,实验上不太可能出现。实验设计:用 2B 模型,固定一个前缀集合,手动构造 logits 使得策略分布有一个「中等概率平台」(很多 token 概率接近),然后应用温度 1.5 后接 top-p 0.9,计算 gap 的符号,再做一个纯正平移探针(
A ≡ +1),测量探针集熵的一阶变化。如果 gap 为负但熵变化也为负,反例成立。12. 落地判断
能否用于「抢占式 GPU 自托管 27B 模型做多智能体脚手架」:能,而且应该用。
怎么用:这个场景下,27B 模型跑多智能体脚手架,通常用 vLLM 或 SGLang 做推理,采样器几乎必然带 top-p 或 top-k 截断(为了控制延迟和显存)。如果训练时学习器不重归一化,就会引入本文描述的不匹配漂移。具体落地:
log π(y) − log π(S)。这个改动只需要在损失计算处加几行代码,不需要改推理引擎。N和平均 gap。如果N的绝对值持续大于 0.05 且 gap 大于10^{−3},说明漂移显著,需要修正。成本量级:支持集重归一化的额外计算是每个采样 token 做一次 kept set 的 log-sum-exp,对于 27B 模型、每步 32 组 × 8 响应 × 平均 2000 token,约 512,000 个 token 位置,每个位置对 top-k 20 的 kept set 做约 20 次 logit 查表和一次 log-sum-exp。这在单张 24GB 卡上约增加 1–2% 的训练步时间。内存开销为零(不需要存储额外状态)。
最大的坑:推理引擎与学习器的 kept set 不一致。论文报告约 0.06% 的 token 落在学习器重计算的 kept set 之外,原因是引擎与学习器之间的数值差异(浮点舍入、logit 精度)。如果这个比例在你的系统里更高(比如因为引擎用了不同的 top-p 排序实现),重归一化会把这些 token 的 log 概率变成
log π(y) − log π(S),其中y ∉ S,导致 log 概率被错误地压低。对策:实现时把采样 token 强制加入 kept set(论文 Section C 提到「kept set joined with that token」),这样即使 token 不在重计算的 kept set 里,也不会出现log π(S)不包含π(y)的数学错误。另一个坑是异步管线:如果推理引擎和训练器之间有 rollout lag,重要性比率对旧策略可以修正 staleness,但不能修正截断(论文 Section G),所以重归一化仍然必要。