Read this paper write-up in English
RACE:跨轮动作似然驱动的推理压缩¶
独立核心机制 API 与真实 checkpoint 接口。短运行是 L1 诊断,非论文 benchmark 复测;尚未接入统一 Evolve 晋级。
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | arXiv 2610.12061 v1 |
| 公司/机构 | Renmin University of China(一作 Yiruo Cheng) |
| 首次公开日期 | 2026-10-08 |
| 原文开源代码 | 作者公布的仓库,截至 2026-10-10 仓库为空,尚无可下载实现 |
| Adapter | race(独立机制 API,非通用 reproduce adapter) |
| 本地复现代码 | src/auto_research/agent_research/race.py;共享后端 oct10_backend.py;scripts/run_oct10_agents.py |
原始论文总结¶
背景与主要改动¶
LoGiC 从第二轮逐步删除推理,检查本轮及所有后续原动作的平均 token 对数似然变化,已接受删除继续影响后续上下文。成功轨迹压缩后用于 SFT;RL 屏蔽可删推理并监督提前闭合。原文四个 Agent 环境推理 token 分别减少 32.3%、80.2%、46.4%、62.0%,未报告线上 A/B。
架构流程¶
graph LR
A[真实成功训练轨迹] --> B[逐轮移除推理]
B --> C[全部后续动作似然重算]
C --> D[接受删除或保留]
D --> E[压缩轨迹SFT]
D --> F[屏蔽策略项与闭合监督]
F --> G[真实on-policy工具轨迹]
原论文关键图¶
原论文 Figure 3(关键图):展示原论文的训练流程与关键优化环节。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
\(d_j=\ell(a_j\mid h_j)-\ell(a_j\mid\tilde h_j)\);所有 \(j\ge t\) 的 \(d_j\le\epsilon\) 才删除。RL 分母保留原全部策略 token 数;闭合项按全部回合数归一化。
论文离线与线上效果¶
上述数字属于原论文全文,不是本地结果,模型规模和指标协议不同不能直接比较。
本地复现¶
动作似然、成功轨迹 SFT 与 cover-aware actor 梯度接口已实现。真实输入必须提供公开环境、修订与 reason/action/observation 训练轨迹;不能把 QA 解题行伪装成工具轨迹。评估 Agent 不读取参考动作。论文四环境完整 on-policy RL 尚未复测;其端到端能力结果仍缺。
操作与数据协议¶
python -m pip install -e '.[post-training-gpu]'
PYTHONPATH=src python scripts/run_oct10_agents.py \
--method race --checkpoint /path/to/public/checkpoint \
--dataset /path/to/train.jsonl --device cuda \
--seed 42 --output runs/race/result.json
输入必须有 question、environment、source_revision、split=train、successful=true 与 turns;每轮含 reasoning、action、observation,参考轨迹只用于训练。也可加 --race-bounded 使用本仓库公开有界 calculator 环境:真实采样 A/B 工具动作并执行,环境私有判成功,随后重新采样 on-policy 轨迹训练;这是 L1 运行合同验证,不是官方 ScienceWorld。
验证与边界¶
2026-10-10,NVIDIA A100,SmolLM2-135M-Instruct,seed=42:公开 bounded-calculator-v1 中,模型真实采样动作、执行计算工具,再以成功轨迹做 SFT,并重新采样 64 条 on-policy 轨迹。11 条成功;cover-aware actor loss=0.02441,实际闭合监督项 6,原策略 token 分母 554,屏蔽推理 token 24。该环境只验证训练机制,不是 ScienceWorld,不作能力排名。
本地指标:L1 GPU 诊断。检查点修订、源码 commit 与执行命令以本批 GPU 回执为准。
对应 tests/test_race.py 检查公式和状态合同。真实运行保存诊断标识、seed、轨迹/似然或评测路径。单 seed、短响应与少量训练步数不能当稳定提升;失败和负结果保留。GPU 回执由本批集成统一登记,原始 runs 及权重不提交。CPU 可用于机制调试,CUDA 路径必须通过真实 NVIDIA 验证。
