ADRS:回报相关的自蒸馏奖励塑形¶
Fidelity:核心机制复现。本页把原论文结论、本地机制验证和未复刻部分分开陈述。
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | Agentic Reinforcement Learning with Self-Distilled Reward Shaping(arXiv 2608.03223) |
| 公司 / 机构 | University of Science and Technology of China / Alibaba Group |
| 首次公开日期 | 2026-08-04(arXiv v1) |
| 原作者代码 | 已开源:gitrxh/ADRS-arxiv |
| 本地 adapter / 方法键 | adrs |
| 本地复现代码 | src/auto_research/post_training/ |
原始论文总结¶
背景与主要改动¶
privileged teacher 的高置信并不必然与真实任务回报一致。ADRS 在每个交互 step 内标准化教师分数,以教师置信与 realized return 的相关性形成 TVA gate,再把 gated token signal 写入原生 reward-to-advantage 路径,推理时无需技能。
flowchart LR
A["学生 / Agent rollout"] --> B["训练目标与教师视图"]
B --> C["可审计的目标或状态更新"]
C --> D["公共 mini-suite 评测"]
原论文关键图¶
原论文 Figure 2(关键图):展示原论文的训练流程与关键优化环节。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
\[
z_{i,t}=(s_{i,t}-\mu_t)/(\sigma_t+\epsilon),\quad g_t=[\operatorname{corr}(z_{\cdot,t},R)]_+,\quad A'_{i,t}=A_i+\lambda g_tz_{i,t}.
\]
论文离线与线上效果¶
三个交互基准、多个 RL backbone、低数据和未见任务上均持续提升;摘要未给统一单值,且无生产 A/B。
本地复现¶
实现 step 内中心化、return association、TVA gate 与原生 REINFORCE advantage 注入。
Arithmetic candidate suite、120 steps、seed 42:accuracy 0.2344 → 0.6250(+166.67%)。诊断字段完整记录在固定指标文件中。
auto-research post-train --algorithm adrs --dataset arithmetic-smoke --maximum-examples 256 --steps 120 --seed 42
auto-research evolve --model post-training --dataset arithmetic-smoke --direction "组合 adrs 与已安装论文算子" --generations 2 --population 4
固定指标见 ../../experiments/p0-p1-closed-audit-20260808-seed42.json。
复现边界¶
本地使用确定性公共 mini-suite 验证核心状态更新和公平预算,不等同于原论文大模型、多卡 RL、私有环境或完整 benchmark;本地相对变化不得与原文提升混写。
