Read this paper write-up in English
ReCal:为 OPD 恢复校准结构化剪枝¶
已实现真实 FFN 通道删除、教师 SFT 和学生轨迹 OPD;当前四题诊断包含负结果,不代表论文 benchmark 收益。
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | ReCal: Calibrating Structured Pruning for On-Policy Distillation Recovery |
| 公司 / 机构 | Zhongguancun Academy / Tencent / University of Science and Technology of China(第一作者 Houcheng Jiang) |
| 首次公开日期 | 2026-10-08(arXiv v1) |
| 原作者代码 | 已开源 |
| 本地 Adapter | recal |
| 本地复现代码 | src/auto_research/post_training/recal.py;入口 scripts/run_recal_checkpoint.py |
原始论文总结¶
背景与主要改动¶
先剪枝再蒸馏并不保证可恢复:损坏的学生轨迹会限制 OPD 教师信号。ReCal 先用剪枝 probe 测量教师轨迹上损伤,再用损伤重加权原始未剪枝模型的校准统计,重新选择通道;目标宽度和恢复流程保持一致。
flowchart LR
原始教师 --> 教师轨迹 --> 剪枝probe损伤
剪枝probe损伤 --> 重加权校准 --> 实体剪枝 --> SFT --> 学生轨迹OPD
原论文关键图¶
原论文 Figure 3(关键图):从教师轨迹定位剪枝损伤,回到未剪枝模型重加权校准,再实体剪枝并进行 SFT/OPD 恢复;不是在 probe 剩余通道中再次选择。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
损伤由 \(D_{KL}(p_T\|p_{probe})\) 测量;实现保留教师 Top-16 及其余词表的总概率桶,而非只归一化 Top-16。每条响应的损伤权重归一至单位质量,零损伤退化为均匀权重。Wanda-SP 用权重范数乘加权激活 RMS 选通道。
论文离线与线上效果¶
原文在数学比较中最高提高 AIME 16.7 个百分点,但并非所有模型或任务都改善;IFBench 比较存在下降。未报告线上 A/B。
本地复现¶
Qwen3-4B/A100,FFN 宽度从 9728 真正减到 7296。对照与 ReCal 使用相同目标宽度和教师 SFT 响应,2 轮 SFT、4 步 OPD;SFT/OPD 问题分开,验证 4 题。见运行说明。
数据协议与结果¶
教师生成截断时停止而非悄悄接受不完整目标;学生推理不接收答案。具体阶段成绩、非零梯度和参数位移见逐阶段指标,不要把四题满分解释为稳定能力。使用 GPU 回执 对照执行命令。
代码映射与复现边界¶
prune_gated_ffn 删除 gate/up 的行及 down 的列,不是把 dense 参数置零。恢复使用 q/v LoRA,未复现原文全参数 FSDP、四领域数据或多 seed。原始短跑里 ReCal OPD 后 exact match 从 1 降到 0.75,对照仍为 1;保留负结果,不声称提升。旧切分声明与重跑原因见运行审计。尚未接入 Evolve。
