CISPO:裁剪概率比后,梯度还剩多少#
TL;DR#
本页内容6 个部分
1. 逐 token 裁剪与梯度归零#
DAPO 为正优势 token 放宽了上裁剪界。一旦当前概率与采样时概率的比值超过这个上界,任务目标仍会进入常数分支,该 token 的任务梯度归零。负优势也有对应的情况。比值跌破下界后,DAPO 会停止通过该 token 的任务项继续压低它的概率。[1]
CISPO(Clipped IS-weight Policy Optimization)让概率比超过上界时,token 仍能提供任务梯度。它把裁剪后的比值用作梯度权重,并保留当前策略的 log 概率参与求导。[2]
图中纵轴是乘在当前 token 的 log 概率梯度上的系数。上、下两栏分别把优势固定为 和 。两种方法共用 1.2 作为上界,0.8 是 DAPO 的下界。数字是为比较而设的手算配置。
2. 核心目标与机制#
对问题 的一个已采样回答,在第 个位置,令 表示相同的前缀, 是采样到的 token。正在更新的策略是 ,实际生成该 token 的策略是 。概率比为 ,表示当前策略相对采样策略给这个 token 多少倍的概率。用 表示该回答的优势估计,更新时固定 和采样概率。[2]
为单独看裁剪,两种方法使用相同的优势和 token,暂省共同的平均系数。设 和 为 DAPO 的下界和上界。DAPO 要最大化这个 token 的目标项:
当 时,较小值等于 。当 时,乘以负数会反转大小关系,较小值等于 。这两个分支决定了梯度在哪一侧归零。[1]
令 为 CISPO 的概率比权重上限。采用 MiniMax-M1 实验和 ScaleRL 使用的单边截断,CISPO 最大化的 token 项为:
表示停止梯度。每次前向计算用当前策略重新求 ,这次反传把截断结果当作常数。梯度通过 传播。MiniMax-M1 给出的通式可以设上下界,其实验采用上界截断。[2] [3]
3. 推导与计算#
采样概率固定,因此 。DAPO 的线性分支将 乘在这个 log 概率梯度上,常数分支的系数为零。CISPO 只对 log 概率求导,所以系数为 。在 DAPO 刚好触到裁剪折点时,左右导数可以不同。以下计算都取在折点之外。
正优势,上界之外。 假设 old 给 token 的概率是 1%,current 是 1.6%,于是 。取 、。DAPO 选中常数项 ,所以梯度系数为 。CISPO 选中权重 ,乘上优势后,任务梯度为 。
负优势,下界之外。 换一个 token,old 概率还是 1%,current 为 0.6%,于是 。取 、。DAPO 选中 ,梯度系数仍为 。单边 CISPO 保留权重 ,乘上负优势后,梯度系数为 。这条样本仍提供降低该 token 概率的训练信号。
4. 比较与边界#
当 DAPO 选中随 变化的分支,且 CISPO 也没有触及上界,两者给同一个 token 的梯度系数都是 。负优势在高概率比区间还有一种情况。DAPO 仍用 推低该 token 的概率,单边 CISPO 则把系数的绝对值限制在 。整批梯度还受样本筛选、优势和 Loss 平均 的影响。[1] [2]
权重上限改变了大概率比样本的相对贡献,会引入截断偏差。逐 token 概率比校正的是给定前缀下的条件概率,完整回答的换分布还涉及前缀本身的概率。实际参数步长取决于整批梯度及优化器。某个位置的优势为零或被训练 mask 屏蔽时,CISPO 的对应任务贡献也为零。[2]
参考资料#
- DAPO: An Open-Source LLM Reinforcement Learning System at Scale(2025)。非对称裁剪目标和 Clip-Higher 的定义。§2–3.1
- MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attention(2025)。CISPO 的目标、权重截断和梯度条件。§3.1,式 (3)–(7)
- The Art of Scaling Reinforcement Learning Compute for LLMs(2025)。单边截断形式及与 DAPO 的对照。§3.2,式 (4)