Quiet-STaR: Language Models Can Teach Themselves to Think Before Speaking
Quiet-STaR:从普通文本学习有助于预测的内部 reasoning
在普通文本的多个位置生成中间 reasoning,用其对未来文本预测的帮助训练 model,将自生成 reasoning 学习扩展到问答数据之外。
Classification & RSI Relation 1 Variant · Editorial Assessment
主题标签用于检索;以下按实际 Experiment / Variant 判断更新对象与证据。Not Demonstrated 表示这项研究未提供相应证据。
Rationale Policy Training
- Update Target
- Model Evolution
- Loop Role
- Solver · Experience Generator
- Persistence
- Across Training Updates
- Recursive Reuse
- Demonstrated
- Evidence
- Task Gain · Recursive Reuse
- System Boundary
- Model Weights & Reasoning Policy;以该论文实际可更新组件为边界,固定部分见下方判定。
- Feedback
- 真实后续文本的 log-likelihood 衡量自生成 reasoning 的帮助,policy gradient 据此更新 reasoning 生成相关参数。
- Evidence Scope
- 预测后文更好不等于中间解释在逻辑上真实,model 仍以外部文本提供的信号学习。并行采样与额外 reasoning 增加计算需求;实验不表明系统能创造新训练算法,也不证明持续扩大 reasoning 长度会获得无界改进。 已核验固定版本的相关方法与主要结果;未独立复现。
训练中更新的 Rationale Policy 继续为后续训练生成 Thought,属于 Training-update 尺度的 Generator 反馈复用。该判断描述数据依赖,未证明训练算法自修改或更好的通用 Improver。
检查原文设置 ↗原论文明确提出或报告的内容
相关文献中已有的知识与结论
基于证据的解释、重建或教学推演
待验证的猜测、实验计划或新研究提案
Idea Reconstruction 是从已知背景出发的推演,不代表作者真实心理过程。Follow-up 是研究提案;相关工作比较不等于已证实 Novelty。教学示例与原文案例在正文中区分。
Contents · 12 Questions
Prior Work & Research Gap
前人做到哪一步,缺口在哪里?
STaR 通过最终答案筛选 Rationale,需要问题与正确答案。Quiet-STaR 利用文本本来就有的后续 Token 作为信号,学习哪些潜在思考改善预测,并处理并行生成、思考初始化和未来多 Token 评分等困难。
Idea Reconstruction
从已有知识与失败模式出发,如何走到这个想法?
在读一句话时,人会推断省略的原因、意图或中间计算;这些推断通常能帮助预期下文。若一段内部文字让 Model 更好预测真实后续文本,就可把它视为候选有用思考。初始思考可能很差,所以需要一种平滑机制保留原有预测能力,再逐步学习何时使用思考。
Core Intuition
用一个清楚的视角抓住方法本质。
中间 reasoning 是否有用,可以通过它改变后续文本概率的效果来衡量。这样,未写出的解释成为可学习的隐变量,监督来自真实后文;重点是学会产生对预测有帮助的 reasoning,而非要求 reasoning 文本模仿人工标准答案。
Method & Worked Example
沿着输入、处理、输出走完 Pipeline。
在多个文本位置并行采样 reasoning,以可学习的开始和结束标记界定思考。mixture head 组合原预测与思考后的预测,再用真实后续文本的预测改善作为 REINFORCE 信号,同时更新标记及 model weights。
沿用原文的一般文本训练设置作演示:给定关于某人忘带雨伞的前缀,Model 在该位置采样短 Rationale,再预测后续文本。比较有无思考的预测质量,好的思考得到较高训练信号。实际训练会在多个位置并行处理,并用 Mixing Head 混合预测。这个天气句子是教学构造,不是论文报告的样本。
Mathematical Foundations
从符号、直觉与简单例子理解理论。
核心信号是 Thought 对未来文本 Log-Likelihood 的帮助。离散 Rationale 无法像连续向量那样直接求导,因此使用 REINFORCE 风格估计:高于基线的奖励提高该 Rationale 的生成概率。Mixing Head 学习在有无 Thought 的 Logits 之间插值,避免初期低质量 Thought 破坏原能力。下面是简化估计,不包含原文所有并行与非短视细节。
R(r) ≈ log p(future | prefix,r) ∇J ≈ (R(r)−baseline) · ∇log p(r|prefix) Mixed logits = α·logits_thought + (1−α)·logits_base
把 Rationale 当作一个会被采样的动作 r,就能理解 Policy Gradient。它如果让真实后续文本更容易预测,R(r) 会较高;减去 Baseline 后得到相对好坏的信号。乘上 ∇log p(r|prefix),正信号会提高这类 Thought 的采样倾向,负信号则降低它。Logit 是 Softmax 前的分数,Mixing Head 学习的 α 控制 Thought 路径与普通路径的混合。这里列的是教学近似;论文使用跨后续位置的学习信号,不能将单个 Token 概率提升当作已经学会一般推理的证据。
Experiments & Claims
Research Question → Experiment → Answer
Question:普通文本中的潜在思考训练能否改善预测并迁移到推理?→ Experiment:训练含 Thought 的 Model,检查语言建模及推理任务,分析 Thought 长度和相关组件。→ Answer:作者报告收益,但产生的文字是否忠实反映内部计算,需要另外的干预证据。
摘要报告,继续 pre-training 后 GSM8K zero-shot 成绩从 5.9% 升至 10.9%,CommonsenseQA 从 36.3% 升至 47.2%,且困难词元的预测改善。这些任务没有单独 fine-tuning,但结果仍限于所研究的 model、语料和训练规模。
Takeaways
这篇论文改变了哪些判断?
One-Week Reproduction
用一周检验一个最小且明确的命题。
Day 1–2:建立小型可执行算术文本,限制 Thought 长度。Day 3–5:比较普通训练、随机 Thought 与奖励加权 Thought。Day 6–7:测未见运算组合,并将 Thought 替换或打乱,判断收益是否依赖其内容。
Counterexample Design
如何构造有辨识力的反例?
训练文本中固定用某种措辞对应答案,测试时打乱措辞与运算关系。如果 Thought 只重复文体线索,训练 Likelihood 提高但结构迁移失败,就能拆开预测捷径与推理能力。
Follow-up Research
从缺陷与需求推导新的研究问题。
研究潜在 Thought 的功能可识别性:对 Thought 进行语义保持改写与逻辑破坏干预,找出哪些内容真正决定后续计算。相对 STaR 的终点筛选,这要求因果验证中间过程。目标是建立可迁移计算的判据,而非增加思考 Token 数量。
Sources & Verification
阅读已固定版本的原文 Background、方法与主要结果,结合下列相关文献撰写。Paper Claim 表示作者报告,未独立复现;Inference 包括数学教学推演与 Idea Reconstruction。One-Week Reproduction、Counterexample 和 Follow-up 均为待执行提案,Novelty 尚需更全面的文献与实验检验。
- [1] Quiet-STaR: Language Models Can Teach Themselves to Think Before Speaking · v2 ↗原文 Background、Method / Formalization 与主要实验或理论结论。数字沿用原文设置;未独立运行完整实验。
- [2] STaR: Bootstrapping Reasoning With Reasoning ↗相关方法与研究背景;用于说明具体差异,不能替代对本篇结论的验证。
