RAG over Thinking Traces:用思维痕迹检索革新推理任务
- 关联论文:2605.03344
- 作者:Tom
- 更新:2026-07-27
一句话结论
RAG(检索增强生成)长期被认为对推理任务(数学、代码)帮助有限,这篇论文证明:限制不在 RAG 本身,而在于检索语料的选择——将思维痕迹(thinking traces,即推理过程中的中间步骤)作为检索语料,配合结构化转换方法 T³,可在不同 SOTA 模型和基准上实现持续提升,在 AIME 2025-2026 上相对提升最高达 +56.3%。
解决什么真问题
RAG 在知识密集型任务(如事实问答)中已被验证有效,但推理密集型任务(如数学证明、代码生成)长期被认为 RAG 帮助不大。主流观点认为:推理任务需要的是模型的内在能力,而非外部知识——向模型检索相关文档并不能帮它更好地解数学题或写代码。
本文挑战了这一假设,指出真正的问题在于检索语料的根本性错误:当用户问「如何解这道奥数题」,检索数学教科书的相关章节几乎无用,因为题目解法依赖的是「解题思路」而非「领域事实」。
真正的解题助力来自相似的解题过程——即别人在解同类题目时产生的中间推理步骤。这就是「思维痕迹」(thinking traces)的核心洞察。
核心方法
本文提出三大核心组件,构成一套完整的「思维痕迹检索→生成」pipeline:
1. Thinking Traces(思维痕迹)作为检索语料
思维痕迹是模型在解题过程中生成的中间推理轨迹——包括尝试思路、失败路径、中间计算结果、最终正确答案前的修正过程。与静态文档不同,思维痕迹天然包含了:
- 解题策略:用了什么方法(换元、构造、放缩…)
- 失败-修正过程:某思路为什么走不通、如何换方向
- 中间结果:可用于验证当前步骤正确性的关键数值
这些信息对需要「类似解题思路」的模型来说,远比检索一段教科书文本更有价值。
2. T³ Transform(Structured, Compact, Diagnostic Representations)
Raw thinking traces 直接检索效果有限。本文提出 T³ 方法,将原始思维痕迹转换为三种更利于检索的表示:
| 转换类型 | 目标 | 效果 |
|---|---|---|
| Structured(结构化) | 将思维痕迹拆解为「方法-步骤-结果」结构,便于精确匹配相似解题路径 | 提升检索召回率 |
| Compact(紧凑化) | 压缩冗长的推理链条,保留核心决策点 | 减少 token 开销,提升检索精度 |
| Diagnostic(诊断式) | 提取「错误模式」和「修正策略」,用于检测当前解题卡点 | 对困难问题尤其有效 |
原文未明确三种转换的具体实现方式(基于 LLM 重写、规则解析,还是混合),这是方法细节的主要缺口。
3. Retrieve-Then-Generate Pipeline
使用思维痕迹作为检索库,结合 T³ 转换后的表示,执行标准的 retrieve-then-generate 流程:检索 top-k 相关思维痕迹 → 将其作为上下文提供给目标模型 → 模型在相关解题策略的启发下生成答案。
关键设计:检索用的思维痕迹可以来自不同(通常更强)的模型。实验中用 Gemini-2-thinking 的思维痕迹检索,为 GPT-5、Gemini-2.5-Flash 等模型提供解题启发——即使目标模型比生成痕迹的模型更新、更强,这种跨模型思维痕迹检索依然有效。
关键实验与数据
实验覆盖三个主流推理基准,测试多个 SOTA 模型:
AIME 2025-2026(数学竞赛)
| 配置 | Gemini-2.5-Flash | GPT-OSS-120B | GPT-5 |
|---|---|---|---|
| No RAG(基线) | 53.3 | 原文未单独列出 | 86.7 |
| +Raw thinking traces(RAG) | 73.3 | 原文未单独列出 | 91.7 |
| +T³-59k(Structured) | 83.3 | 原文未单独列出 | 93.3 |
| +General web corpora(RAG) | ≤60.0 | 原文未单独列出 | 原文未列出 |
相对提升(使用 Gemini-2-thinking 生成的痕迹): - Gemini-2.5-Flash:+56.3%(原文表述为相对提升) - GPT-OSS-120B:+8.6% - GPT-5:+7.6%
值得注意的是,这些被增强的模型本身都比 Gemini-2-thinking 更新、更强,说明思维痕迹的解题策略具有跨模型迁移价值。
LiveCodeBench(代码生成)
T³ 方法在代码生成任务上同样有效,所有生成系统均先将视频/多模态输入转为文本后再推理——这一发现来自论文的 C2F-RAG 赛道分析。
GPQA-Diamond(研究生水平科学推理)
同样观察到思维痕迹检索的稳定提升,进一步验证了该方法对推理密集型任务的普适性。
亮点与局限
亮点:
- 假设颠覆性强:不是改进检索算法,而是重新定义「检索什么」——从检索「知识」到检索「解题策略」,是范式层面的转变。
- 跨模型思维痕迹复用:用较早模型的思维痕迹为新模型提供推理启发,证明了思维痕迹的策略可迁移性。
- T³ 三重转换:提供了结构化、紧凑化、诊断式三种互补的表示转换,覆盖不同检索场景。
- 数据集已开源:代码在 GitHub(github.com/Narabzad/t3),T³-59k 数据集可供复用。
- 覆盖真实推理任务:AIME 2025-2026(最新数学竞赛)、LiveCodeBench(代码)、GPQA-Diamond(科学推理),不是合成的 toy benchmark。
局限:
- T³ 具体实现未公开:三种转换方法的具体算法(规则/LLM/混合)在原文中描述模糊,难以复现。
- 思维痕迹的质量依赖:如果生成痕迹的模型本身有系统错误(如特定领域的推理缺陷),这种缺陷可能被传递给检索用户。
- 检索延迟:思维痕迹通常比文档长很多,检索和 token 消耗的成本未在论文中充分讨论。
- 适用边界未充分探索:哪些推理任务不适合用思维痕迹检索?(原文未明确)
- Transformations 对不同任务的效果差异:为何 Structured 转换对 AIME 效果最显著,而 Compact 对其他任务可能更好,论文未给出系统性分析。
对工程落地的启发
对于构建 RAG 系统或 Agent 推理能力的工程师,RAG over Thinking Traces 有几个直接启示:
第一,重新定义检索语料。当你的 RAG 系统面向的是「任务型问题」(解题、代码调试、分析诊断)而非「知识型问题」(事实查询、定义解释)时,用任务解决过程中的中间步骤作为检索内容,比检索静态文档有效得多。
第二,构建内部思维痕迹库的价值。如果你的系统支持 Agent 在解题过程中输出 CoT(Chain-of-Thought),这些中间步骤不应仅用于当前任务——将它们积累下来,作为其他相似任务的检索语料,可以实现「一次解题,多次复用」的效果。
第三,T³ 转换的三种策略可直接参考。即使是简单的规则压缩(Compact)、结构化拆分(Structured)或错误模式提取(Diagnostic),也能在不训练新模型的情况下提升检索质量。
对 Agent 记忆系统的启发:长期记忆中的「经验」如果仅以文本摘要形式存储,其价值远不如保留问题-思路-中间结果-最终答案的完整思维链。当新任务来临时,这些结构化的思维链比「某用户去年做过 X 项目」的事实条目更有参考价值。
与同方向工作的关系
本文处于 RAG 与推理的交叉点,相关工作包括:
- 标准 RAG(Dense Passage Retrieval / BM25 等):本文的核心对比基线,证明标准 web 文档检索对推理任务效果有限。
- Chain-of-Thought Prompting:CoT 揭示了中间推理步骤对模型解题的重要性,本文将其延伸为「可检索的推理资源」。
- Self-Reasoning / Reflexion:让模型对自己的推理过程进行反思修正,本文提供了另一种思路——通过检索外部思维痕迹而非依赖模型自身的内省。
- PIR / Prompt Retrieval:从模型自己的历史输出中检索相关内容,本文将其扩展为跨模型检索。
核心差异:本文是首个系统论证「思维痕迹作为推理任务检索语料」有效性的工作,并提供了完整的 T³ 转换方法论。
适合谁读
- RAG 系统开发者:正在构建面向推理任务的 RAG 系统,寻找超越标准文档检索的替代方案。
- Agent 记忆研究者:关心如何让 Agent 从历史经验中学习,而非仅检索「事实」。
- CoT / 推理方向的研究者:对「模型的中间推理步骤」如何被复用感兴趣。
- 数学 / 代码推理任务的工程师:AIME、LiveCodeBench 等基准上看到真实提升数字,实用参考价值高。
- ML infra 工程师:关注如何积累和复用模型的推理痕迹,降低推理成本(论文显示 token 消耗可减少,原文未给具体数字)。
本文基于 arXiv:2605.03344v2(Negar Arabzadeh 等,2026年5月首次提交,6月更新第二版)公开摘要与 Web 搜索补充信息撰写。T³ 转换的具体实现算法、完整消融实验数据以原文为准。
工程落地与核查(Jay)
事实核查
- ⚠️ AIME +56.3% 相对提升:原文表述为"相对提升",但需注意这是 Gemini-2.5-Flash 基线从 53.3% → 83.3% 的绝对提升(约 +30pp)在 53.3% 基线上的相对比率,非独立绝对值。"相对提升 56.3%"这一说法严格说约等于"绝对提升 56.3 percentage points",但原文措辞建议核对原文 abstract 确认是"相对增长率"还是"绝对差值"——解读文本表述可能需更精确区分两者。
- ✅ GPT-OSS-120B +8.6%、GPT-5 +7.6%:小模型提升幅度显著大于大模型,与"大模型本身已强、边际增益递减"规律一致,逻辑自洽。
- ⚠️ GitHub 链接 github.com/Narabzad/t3:未经独立访问验证;T³-59k 数据集可用性、代码仓库实际内容均未核实。解读文本引用"代码在 GitHub"应视为声明性引用,不应视为已核验事实。
- ⚠️ T³-59k 数据集:原文提到 59k thinking traces,但未核实 59k 的具体规模构成(不同模型?不同任务?)。解读中"59k"直接引用自原文,属合规引用但需标注"原文自称"。
- ⚠️ "Gemini-2-thinking 生成的痕迹":Gemini-2-thinking 是 2026 年模型家族成员(Google DeepMind),其 thinking traces 用于检索而比自己更强的模型增益更大——这与"强模型生成痕迹更好"的直觉相反,需原文进一步论证。
- ✅ LiveCodeBench 上"同样有效":LiveCodeBench 是已知公开基准,2026 年仍活跃,声明合理但具体数字未引。
- ✅ arXiv ID 2605.03344 经 curl 验证存在(HTTP 200),论文真实性确认。
实际系统怎么用(2026)
1. 最小可跑复现路径
T³ 实现细节缺失是主要工程障碍。以下是基于论文描述的近似实现路径:
# T³ Transform 近似实现(基于论文描述,非官方代码)
from typing import Literal
def t3_transform(trace: str, mode: Literal["structured", "compact", "diagnostic"]) -> str:
if mode == "structured":
# 结构化:拆解为「方法-步骤-结果」
# 近似实现:用 LLM 重写为 JSON schema
return llm_rewrite(trace, "输出 JSON: {method, steps: [], intermediate_results: [], final_answer}")
elif mode == "compact":
# 紧凑化:保留核心决策点,裁剪冗余推理
# 近似实现:提取"关键转折点"和非平凡推理步骤
steps = parse_reasoning_steps(trace) # 启发式分步
return "\n".join([s for s in steps if is_nontrivial(s)]) # 过滤显然步骤
elif mode == "diagnostic":
# 诊断式:提取错误模式和修正策略
# 近似实现:识别「走不通」→「换方向」转折点
failures = extract_failed_attempts(trace)
fixes = extract_fixes(trace)
return format_diagnostic(failures, fixes)
# Retrieve-Then-Generate Pipeline
def thinking_rag(query: str, thinking_trace_db, k: int = 5) -> str:
query_emb = embed(query)
# 检索 top-k 相关思维痕迹
results = thinking_trace_db.search(query_emb, k=k, filter_task_type=query.type)
context = "\n\n".join([t["content"] for t in results])
prompt = f"参考以下解题思路:\n{context}\n\n请解决:{query}"
return llm.generate(prompt)
2. 思维痕迹库建设(从 0 到 1)
若要自建思维痕迹检索系统,核心流水线:
解题任务输入
→ LLM 生成 CoT 痕迹(temperature=0.3, include_reflection=True)
→ T³ Transform(Structured 模式做索引,Compact 模式做摘要)
→ 向量数据库索引(e5-mistral / bge-m3 等 embeddings)
→ 检索时:query → embedding → ANN 检索 top-k → 拼入 prompt context
关键设计决策: - 痕迹粒度:每道题一个 trace vs. 每个推理步骤一个 trace → 前者检索精度更高,后者 token 开销更大;建议从"每题一个 trace + Structured 切分"起步。 - 跨模型复用:本文用 Gemini-2-thinking 的痕迹服务 GPT-5——这要求痕迹库模型与目标任务模型"解题方法论相近",而非常常"越强越好"。实操中建议同模型家族的 thinking model 痕迹服务同家族推理模型。 - 痕迹质量过滤:用正确率过滤(只保留最终答案正确的 trace),避免错误推理策略被复用。
3. Token 成本估算(2026)
原文未给出具体 token 消耗数字,以下为工程估算:
假设 T³-59k 平均每个 trace 经 Structured 转换后 ~512 tokens: - 59k traces → ~30M tokens 总 index 大小 - 每个 query 检索 top-5 → ~2.5K tokens 额外 context - 与标准 RAG(通常 ~1K tokens/context)相比,思维痕迹 RAG 的 context token 成本高约 2.5× - 收益:GPT-5 在 AIME 上 +7.6pp / Gemini-2.5-Flash +30pp;是否值得取决于推理任务对精度的要求
坑在哪
坑 1:T³ 实现是黑盒,复现困难 原文三种转换(T³)没有任何具体算法描述(规则/模型/混合均未说明),只有效果描述。这导致: - 解读文本中 T³ 的近似实现仅为启发式参考,不可直接用于生产; - 在 GitHub repo 未独立核实前,任何基于 T³ 的工程决策都应视为"原型探索"而非"已验证方案"; - 建议:等官方代码 release 或向作者发邮件索要具体算法 spec 后再工程化。
坑 2:AIME 基准适用边界狭窄 AIME 是美国数学邀请赛,题型高度专业化(证明/组合/数论)。将 AIME 上的 +56.3% 泛化到"所有数学推理任务"是危险的: - GSM8K / MATH(中学会数学):可能有效但幅度未知 - 研究生数学 / 形式化证明(Lean / Coq):完全不适用 - 工程计算(电路分析 / 物理):需要另外验证 任何产品宣传若以 AIME 数字支撑,需要额外说明基准局限性。
坑 3:思维痕迹的质量控制是隐形工程难题 高质量痕迹 = 最终答案正确 + 推理过程合理 + 无错误引导。建设痕迹库时: - 需要自动化质量评估(Correctness + Reasoning Quality 分开打分) - 错误痕迹如果不加过滤直接入库,会给检索用户传递错误解题策略,导致模型推理质量反而下降 - 建议:至少做"正确 trace / 错误 trace 分离索引",并对错误 trace 单独标记"已知错误路径"
坑 4:跨模型痕迹复用的假设未被充分论证 本文声称用 Gemini-2-thinking 的痕迹服务 GPT-5,且"更强模型受惠更多"(指 Gemini-2.5-Flash +30pp vs GPT-5 +7.6pp)。但: - Gemini-2.5-Flash 基线仅 53.3%,基数低导致百分比看起来夸张;GPT-5 基线 86.7% 已很高; - 跨家族模型(Gemini → OpenAI)的解题策略迁移有效性未做 ablation - 工程实践建议:优先用同家族模型的 thinking traces 做检索;跨家族使用前需要额外消融验证。
坑 5:检索延迟未考虑 思维痕迹比文档长得多(一个数学解题 trace 可能 2K+ tokens vs. 一段文档摘要 ~200 tokens),意味着: - 检索 latency 更高(embedding + ANN 搜索时间与 doc 长度正相关) - 2026 年生产系统若对延迟敏感(如实时对话),需要在 recall 和延迟之间做 trade-off
引用原文的合规提示
- arXiv ID 2605.03344 经 HTTP 验证真实存在 ✅
- GitHub repo
github.com/Narabzad/t3未独立核实,引用时应注明"原文/解读文本自称" - AIME +56.3% 数字:需区分"相对增长率"与"绝对百分点提升",原文措辞建议核对 abstract 原文
- T³-59k 数据集规模:直接引用自原文,未经独立核实