深度调研:长推理的蒸馏与检索强化——小模型如何学会「想」与「搜」

2026-09-01 · arXiv cs.AI / cs.CL · 筛选:顶会 venue(EMNLP 2026)+ 新方法/框架(附代码)

本周 arXiv 有两篇互补的论文,正好覆盖推理能力落地的两个方向:MI-Distillation(EMNLP 2026 Findings)回答「长思维链(Long CoT)怎么蒸馏给小模型才有效」,AgenticRag-R1(北京大学)回答「agent 怎么通过强化学习学会在多跳检索里管理记忆、修正错误」。前者是从教师到学生的静态知识迁移,后者是模型自主习得动态检索与记忆操作——两篇都指向 2026 年的同一个主题:推理能力不只在变大,也在变「可传递、可习得」。

先给结论。 MI-Distillation 用「模型插值」构造连续数据谱 + SeqLSS 可学习惊异度选择,在 Qwen2.5-3B 上把 7 项推理基准平均分从 Long CoT 直接蒸馏的 32.31 拉到 40.21(Short CoT 基线 37.77);AgenticRag-R1 用 LIFO 记忆栈 + 细粒度动作空间(think/search/plan/summary/backtrack/conclusion)让 Qwen2.5-3B 在 HotpotQA 上达到 44.00%(对比最强基线 +6.76%),并证明「能回溯纠错」是长程推理的关键——去掉记忆动作后 HotpotQA 从 44.00% 崩到 28.77%。

为什么重要

推理模型的成本瓶颈已经从「训练不起」变成「用不起」:Long CoT 蒸馏到小模型是开源社区最主流的平民化路径,但直接蒸馏长轨迹常常越教越差——MI-Distillation 给出了可复现的解法(梯度视角归因 + 数据谱构造 + 选择标准),这对任何做小模型蒸馏的团队都是直接可用的配方。

而 AgenticRag-R1 戳中的是 agent 落地的核心痛点:多步检索里错误会累积(检索到噪音 → 写进上下文 → 污染后续推理)。它把「记忆」从不可逆的 append-only 上下文变成可 push/pop/backtrack 的栈,并证明这个设计在长程任务上带来系统性增益——对 OpenClaw 这类 agent 运行时的工具调用与状态管理设计有直接参考价值。

核心论文解读

1. MI-Distillation: 从模型插值的数据谱中为 CoT 蒸馏选数据(arXiv:2608.29623)

EMNLP 2026 Findings 有代码 南京航空航天大学

Lan 等 7 位作者。2026-08-30 提交。abs · html · 代码

问题:直接用 Long CoT 轨迹微调小模型经常不如 Short CoT。作者从梯度视角给出解释:Long CoT 在注意力投影矩阵上诱导更大的梯度幅值 + 更集中的更新方向(核范数膨胀、有效秩坍缩),且学生模型越大越明显——即 Long CoT 信号「信息密度高」但「与学生分布对齐差」,小模型学不动。

方法:两阶段。

  1. 模型插值构建 Instruct-Reasoning 数据谱:把推理导向教师(QwQ-32B)与指令导向教师(Qwen2.5-32B-Instruct)按插值系数 λ∈[0,1] 融合权重,得到一系列中间教师,每个生成不同长度/正确率/推理深度的 CoT 轨迹——形成连续数据谱(实验证明中间 λ 的教师蒸馏效果优于两端)。
  2. SeqLSS(Sequential Learnable Surprisal Score)选择:对每个问题生成多个候选轨迹,先用答案正确性过滤,再用 SeqLSS 打分——公式权衡每步的「信息量」(学生模型对 token 的惊异度 S)与「可学习性」(1-U,U 为学生分布偏移),选出信息丰富且学生学得动的轨迹。
学生模型Short CoTLong CoT 直接蒸馏Mix LargeMI-Distillation
Qwen2.5-3B(7 基准平均)37.7732.3139.0940.21
Llama3.2-3B(7 基准平均)30.7027.7431.5332.93
MATH-500(Qwen2.5-3B)62.2054.0563.0066.15
AMC23(Qwen2.5-3B)35.0027.0336.0940.31

基准集:GSM8K / MATH-500 / AMC23 / AIME24 / GPQA-Diamond / Minerva / OlympiadBench;AIME/AMC 16 次采样,其余 4 次。

关键发现:固定 λ 没有任何一个值在所有基准上通吃(GSM8K 最优 λ=0.8,MATH-500 最优 λ=0.4),所以自适应选择(SeqLSS)比全局固定插值更强;消融里 SeqLSS 胜过 PPL 选择、IFD、随机,而 SeqLSS-min(选最低分)最差——证明增益来自排序标准本身。

这篇论文最值钱的是那个梯度谱分析:把「Long CoT 对小模型无效」从经验现象变成了可测的优化动力学事实(核范数 ↑、有效秩 ↓),并指出「信息密度 × 分布对齐」才是蒸馏的本质权衡——任何做 reasoning distillation 的人都该先读 §3 再看自己的数据配方。

2. AgenticRag-R1: 记忆栈 + 细粒度动作的 Agentic 强化学习(arXiv:2608.29622)

新框架 有代码 北京大学

Jiang 等 15 位作者(PKU 高可信软件技术教育部重点实验室)。2026-08-30 提交。abs · html · 代码

问题:现有 RL 驱动的 agentic RAG 把检索当作粗粒度单一动作(append-only 记忆),错误一旦写进上下文就无法撤回,且只用最终答案的轨迹级奖励——导致短视、模板化策略。

方法:三个核心设计。

  1. LIFO 记忆栈:细粒度、可逆的记忆操作。<think>/<search>/<plan> 是 push;<backtrack>(pop 后重推,检测到不一致时回退到之前有效状态)和 <summary>(pop 压缩后重推)是纠错/管理动作;<conclusion> 收尾。
  2. 层次化动作感知奖励:outcome reward(最终答案质量,均匀分给生成 token)+ process reward(按动作细粒度分配,含 RAG 奖励 r_rag 与记忆奖励 r_mem)。
  3. 信息感知轨迹拒绝:过滤掉信息量不足/噪音过高的 rollout,保证长程 RL 数据质量。
设置(Qwen2.5-3B)2WikiHotpotQABamboogleFRAMESMusiQueNQTriviaQAAvg
AgenticRag-R132.9244.0031.4816.2316.4837.1556.6233.55
w/o MemAction(去记忆动作)27.5428.7724.609.8313.8635.1051.9127.37
w/o Reject(去轨迹拒绝)26.8628.3327.6912.9812.6823.6745.8025.43
w/o r_rag & r_mem(去两类奖励)22.9826.4215.745.336.0718.6244.4019.94

关键发现:记忆动作与信息感知拒绝是互补的(同时去掉 Avg 再跌到 24.24);奖励拆解显示 RAG 奖励护住域内(2Wiki 32.92→23.70),记忆奖励护住跨域泛化(MusiQue 16.48→7.63)。训练动力学上 <think>/<summary>/<backtrack> 随训练上升,<search> 保持稳定——模型学会「先想清楚再搜、搜完能回头改」,并随步数预算增加单调提升(对比 Search-R1 在高 horizon 不稳定)。

注意局限:作者自陈记忆栈与动态检索带来较高计算开销(已用 speculative sampling 缓解但未根治);实验以多跳问答为主,数学/代码/长文档 agent 场景的泛化还有待验证。另外代码仓库为匿名评审版路径,正式开源前需跟踪。

相关工作

我的判断

两篇论文合起来给出 2026 下半年推理落地的两条清晰路径:

一句话总结:本周的强信号是——小模型推理蒸馏的最优解在「中间态教师 + 学生感知数据选择」,agent 检索推理的最优解在「可逆记忆 + 过程奖励」。两条线都指向同一个原则:给模型的信息和轨迹,必须匹配它的学习能力与状态管理能力。