Prime Intellect 推出 IPO 掩码方案,修复 IcePop 在 SWE RL 中的崩溃

nrehiew_ · x · 2026-10-20

Prime Intellect 发布异步 RL 训练的新 token 掩码方案 IPO(INTELLECT Policy Optimization),并用它在困难设置(GLM-4.5-Air 上的 agentic SWE RL、64 步异步 off-policy)中做了验证。

问题背景:异步 RL 中 trainer 与推理引擎对同一 token 的概率永远有偏差。现有主流方案 IcePop 在概率比值超出固定区间时掩掉该 token,但比值判据会误判分布两端——稀有 token 从 0.01 涨到 0.06 是 6 倍比值却被掩(实际变化很小),高概率 token 从 0.90 跌到 0.20 比值仅 0.22 却被保留(实际策略变化巨大)。

IPO 做法:改用概率的绝对差而非相对比值作为掩码判据:|πtrain − πinfer| > ε。掩码是双边的、不依赖 advantage 符号、只有一个超参;判据源自 DPPO(Qi et al., 2026),但 IPO 把它用于 trainer–inference 失配而非 PPO 信任域,并替换掉了之前的 DPPO 默认 loss。

实测结果:IcePop 在约 1.2k 步时因 trainer–inference KL 暴涨超 20 倍而训练崩溃(solve rate 从 0.8 跌落);IPO 全程稳定训练,仅掩掉约 0.25% 的 token。

所属事件:Prime Intellect 发布 IPO:修复异步 RL 尾部误判(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →