2014年,深度学习刚在图像识别上大杀四方的时候,没人会想到十年后大语言模型还会在一个看似简单的问题上栽跟头:让AI一步一步推理,走上十几步、几十步,最后得出正确答案。
这事儿听起来不难。人类做数学证明、做逻辑推理,不也是一步接一步来的吗?可当研究者们真的让AI去挑战一个叫做Andrews-Curtis猜想的数学难题时,问题暴露了。这是一个群论里的经典难题,任务是通过一系列合法的符号变换,把一个复杂的群表示逐步化简成最简单的"平凡表示"。每一步变换单独看都没毛病,但走着走着,AI就迷路了。
这篇来自弗吉尼亚理工、威斯康星大学麦迪逊分校和达特茅斯学院研究团队的论文,把这个"走着走着就迷路"的现象拆解成了两种具体的病症,并且开出了药方。
**推理这件事,为什么走得越远越容易崩**
先说清楚这篇论文研究的到底是什么场景。
长程推理*:指的是AI需要连续做出很多步决策才能得到最终答案的任务,比如复杂数学证明、多步代码调试、长篇逻辑推理,区别于那种"一步到位"的简单问答。
这类任务有一个天生的麻烦:中间过程往往没人告诉你对不对,只有走到终点才知道成败。这在强化学习里叫"稀疏奖励"。
稀疏奖励*:指的是训练信号只在任务完成时才出现,中间步骤没有反馈,AI没法知道自己走到第五步的时候是不是已经偏离了正轨。
想象一下你蒙着眼睛在迷宫里走,只有走到出口才会有人告诉你"到了",走错了也没人拍你肩膀提醒。你唯一能做的就是凭感觉往前挪,等真的撞墙了才知道错,可这时候已经走出去二十步了,回头都不知道该退到哪一步。这就是稀疏奖励下长程推理的真实处境:错误发生得早,暴露得晚,纠正得更晚。
论文里用Andrews-Curtis问题(后文简称AC问题)做了一个特别形象的例子。AI在处理一个群表示的时候,每一步可以选择反转某个关系式、把两个关系式相乘、或者用某个生成元做共轭变换。这些操作单独看都是"合法"的,就像下棋时每一步都符合规则,但只有极少数走法能真正把局面导向简化,大多数合法走法只是原地打转甚至越搅越乱。
研究团队把这背后的根本困境归结为两种偏差,也是全文的核心诊断。
第一种叫探索偏差(exploration bias)。随着推理步数增加,可能的路径数量是指数级爆炸的,但能通向成功的路径只占极小一部分。AI训练时倾向于选择那些"看起来容易采样、局部合理"的分支,而不是那些"虽然罕见但真正能走通"的分支。用书里的话说,多数合法的走法都是"局部看似靠谱、结构上却站不住脚"的死胡同。
第二种叫复合偏差(compounding bias)。因为只有终点才有奖励信号,早期一个不起眼的偏离没法被及时纠正,它会随着步数累积,越滚越大,最后彻底葬送整条推理链。论文里给了一个特别扎心的技术解释:在KL正则化的训练框架下,当终点奖励信号太弱,训练的梯度更新会被KL项主导,逐渐退化成"尽量维持原模型的行为不变",这意味着模型压根没有动力去修正那些早期埋下的小毛病。
KL正则化*:一种强化学习训练技巧,让新训练出的模型不要和原始参考模型偏离太远,防止训练跑偏,但副作用是当奖励信号太弱时,模型会"懒得改变",直接抄参考模型的作业。
这两种偏差不是空想出来的概念,研究者专门用数学工具去刻画了它们,这个工具叫Symbolic Closure Analysis,简称SCA。
**SCA:给推理空间画一张"可行区域"地图**
SCA的核心思路,是把整个推理过程看作一棵不断分叉的树,每一步走一个分支,走到底才算完成一次推理。SCA定义了一个叫"可行闭包"(feasible closure)的概念。
局部可容许性*:指某一步操作在语法或规则层面是"合法的",不代表这条路最终能走到正确答案,就像下棋时每一步都符合规则,不代表这盘棋能赢。
局部可容许性有一个重要的数学性质,叫前缀封闭。
前缀封闭*:如果一条推理路径的某个前段就已经违反了局部规则,那么不管后面怎么走,这整条路径都注定不可行,这就像盖楼时地基歪了,上面盖得再整齐也是危楼。
基于这个性质,研究者证明了一个关于探索偏差的关键结论:可行路径占全部路径的比例,大致等于每一步"可行分支数"与"总分支数"比值的连乘积。如果每一步的可行分支远少于总分支,这个连乘积会指数级缩小。换句话说,如果不加干预,模型随机撒网式地探索,几乎注定会把大部分精力浪费在不可行区域,这在数学上是几乎必然的,不是运气不好。
论文接着用一个方差分解命题(Proposition 3.4)说明了这件事对训练梯度的实际影响:如果采样落在可行区域和不可行区域两边,会引入额外的"跨区域方差",这部分方差会拖慢训练、让梯度信号变得嘈杂。反过来说,如果能让采样集中在可行区域内,这部分噪声就能直接消掉。
而对于复合偏差,论文给出了一个更硬核的定理(Theorem 3.5),证明了在终点奖励极度稀疏的情况下,KL正则化训练得到的最优策略和参考策略之间的总变差距离,被一个和"成功轨迹在参考策略下出现概率"成正比的式子严格限制住了。翻译成人话:如果参考模型本来就很少能碰巧走对,那么单靠终点奖励去训练,模型几乎没法真正偏离原来的行为模式,也就没法学会避开那些早期就该纠正的小毛病。
这就好比你想训练一个孩子学做饭,但你只在他端出最终菜品的时候打分,中间切菜、调味的每一步都不管。如果这孩子十次里九次做出来的菜都难吃,那他压根积累不了"哪一步切错了"的经验,只能靠瞎蒙。SCA这套理论就是在数学上证明了这种"只看结果不看过程"的训练方式,在长程任务里天生就是低效的。
**SAGE:给推理装上两个"导航仪"**
诊断清楚了病灶,接下来就是开药方。研究团队提出的方案叫SAGE,全称是Structural Admissibility-Guided Exploration(结构可容许性引导探索)。
SAGE的思路不是去暴力搜索或者堆更多标注数据,而是往训练过程里注入两种结构性的引导信号,分别对症探索偏差和复合偏差。
第一味药叫代数稀疏化(algebraic sparsification),专门对付探索偏差。
它的做法是这样的:在推理的每一步,AI面前的当前状态会残留一些"没解决的结构信息",论文里叫残差(residual)。比如在AC问题里,残差可能是还没化简的生成元数量、关系式长度这些代数结构信息。对于每一个候选操作,系统会计算这个操作能不能"命中"这些残差,命中程度高的操作会被优先采样。
具体的数学实现是把残差投影到每个操作对应的一个子空间上,算一个投影长度占残差总长度的比例,比例越高说明这个操作越"对症"。论文还专门证明了一个恢复保证(Theorem C.1),借用信号处理里经典的正交匹配追踪(OMP)理论,说明只要候选操作对应的子空间之间不太"纠缠"(专业说法叫互相关系数足够小),这套贪心式的匹配算法理论上能准确找到真正相关的操作。
这就好比你在修一台故障的机器,机器上亮着好几个警示灯,分别对应不同的部件。有经验的师傅不会随手拿起扳手乱拧,而是先看哪个警示灯最亮,然后专挑对应那个部件下手。如果不这么做,直接随机试螺丝,理论上也可能修好,但大概率是在浪费时间拧那些根本没问题的螺丝,机器修好的概率会大幅降低,这正是探索偏差在实际操作里的样子。
第二味药叫双曲结构引导(hyperbolic structural guidance),专门对付复合偏差。
这里用到了一个有点反直觉的数学工具:双曲几何。
双曲空间*:一种负曲率的几何空间,和我们平时熟悉的平直空间(欧几里得空间)不同,它的特点是能用有限的空间容纳指数级增长的分支结构,天然适合表示树状、层级状的数据。
为什么用双曲空间而不是普通空间?因为长程推理天然长得像一棵树,从起点开始不断分叉,越往深处分支越多。如果用普通的平直空间去表示这棵树,越靠外围的节点会挤得越来越密,几乎没法区分。而双曲空间的负曲率特性正好相反,越往外走空间反而"越宽敞",特别适合装下这种指数膨胀的结构。
具体做法是把每一步的推理状态,通过一个叫Poincaré嵌入(Poincaré embedding)的方法映射进双曲空间里,然后计算当前状态和目标状态之间的双曲距离,距离越近,说明这一步越"朝着正确方向走",就给一个奖励信号。这个信号不需要等到最后一步才出现,每一步都能算,这就直接解决了"只有终点才有反馈"的问题。
打个比方,这就像你开车去一个陌生城市,如果导航只在你到达终点时告诉你"到了"或"没到",中间开错了路你完全不知道,那真的到达目的地的概率会很低。而双曲结构引导相当于给你装了一个实时导航,哪怕你还在半路,它也能持续告诉你"现在离目的地是变近了还是变远了"。如果没有这套实时反馈,早期一个错误的转弯要等到几十步之后才被发现,那时候想回头已经太晚了,这正是复合偏差最致命的地方。
论文把这两个信号合起来,构造了一个叫ΨSAGE的综合结构势能,用来调整训练时的采样概率,同时也融入奖励计算。整个机制在训练阶段起作用,模型学到的偏好会被吸收进策略本身,推理阶段直接用训练好的模型走一遍,不需要额外的搜索或过滤开销。这一点很关键,意味着SAGE不是靠推理时"多算几遍再挑最好的"取巧,而是真正让模型学会了更聪明的走法。
论文还给出了一个理论保证(Proposition 4.2),证明只要结构势能能把可行区域和不可行区域的势能值区分开一个足够大的间隔,增大引导强度参数λ,就能让不可行区域的概率质量相对可行区域呈指数级压缩。这个结论和前面的方差分解命题呼应上了,说明SAGE不是拍脑袋设计,而是有严密数学逻辑支撑的。
**实验结果:从2B到35B,SAGE都在涨分**
理论说得再漂亮,最后还得看实验数据说话。研究团队在7个模型家族、12个基准测试上做了系统评估,涵盖三大类任务:封闭形式的数学推理(比如MATH、AIME24、Putnam数学竞赛题)、开放式的自然语言推理(比如MMLU-Pro、GPQA、BBH-H)、以及最硬核的真实世界长程推理任务,也就是前面提到的AC问题。
数学推理这块,数据很直观。在9B规模的Qwen3.5模型上,基线模型平均准确率是29.01%,加上SAGE之后涨到47.95%,这中间差了将近19个百分点。在35B规模上,从45.21%涨到64.86%。更让人意外的是,35B的SAGE版本(64.86%)居然超过了参数量翻倍的Llama-3.3-70B-Instruct旗舰模型(38.48%),差距超过26个百分点。这说明结构引导带来的收益,某种程度上比单纯堆参数更划算。
自然语言推理这块也没落下。9B模型在BBH-H基准上从44.04%提升到45.31%,ARC-C从39.73%提升到42.04%;27B模型的BBH-H达到60.25%,35B模型达到69.07%。这些都是在没有过程标注、没有人工写好的推理步骤监督的情况下拿到的,全靠训练时的结构引导信号。
真正的重头戏在AC问题上。论文用了两个指标衡量表现,一个叫AC Validity(衡量每一步操作是否语法和逻辑上有效,反映局部精度),另一个叫Lean-Verified Proofs(衡量最终证明能否通过编译器验证,是端到端严谨性的黄金标准)。
结果显示,SAGE在AC Validity上带来了19.2%到26.0%的提升,在Lean-Verified指标上更是在所有测试架构里都带来超过13%的提升,其中Qwen3模型达到了接近8倍的增长。这个数字是全文最亮眼的成绩,也是标题里"8倍改进"的来源。
论文还专门做了消融实验,把代数稀疏化(ΨP)和双曲结构引导(ΨH)分开测试。结果显示,去掉任何一个组件,性能都会下降,证明这两个信号是互补的,不是重复劳动。研究者还设计了两个对照实验来排除"只是加了个密集奖励就管用"这种简单解释:一个是把双曲距离换成普通欧几里得距离,效果变差;另一个是把目标锚点打乱,效果大幅下降。这两个对照实验说明SAGE的效果来自于"对准目标"的结构引导,而不是随便加点密集信号就能奏效的。
研究团队还和一种叫GRPO-PRM的方法做了对比,这是一种依赖学习出来的过程奖励模型(process reward model)的方案。结果显示GRPO-PRM确实比原始GRPO有提升,证明密集的过程反馈确实有用,但SAGE依然更强,尤其在AC问题上优势明显,说明SAGE的收益不是靠简单堆密集奖励,而是靠真正对齐目标结构的引导信号。
**写在后面**
读完这篇论文,最触动我的不是那些漂亮的准确率数字,而是研究者把"AI推理为什么会失败"这件事,拆解得如此精确。很多论文遇到长程推理的问题,会本能地想到"加更多数据""加更多标注""训更大模型",但这篇论文选择先停下来问一个更根本的问题:失败到底是怎么发生的,发生在哪一步,为什么会发生。
这种诊断先于治疗的思路,其实值得所有做工程的人借鉴。不管是训练AI还是排查一个复杂系统的故障,直接上手改代码往往比先搞清楚问题根源要诱人得多,但也往往治标不治本。
另一个让我印象深刻的地方是双曲几何的引入。这不是第一次有人用双曲空间处理层级结构数据,但把它用来给稀疏奖励的长程推理"实时导航",这个组合确实巧妙。它提醒我一件事:几何直觉有时候比堆算力更能解决结构性问题,前提是你得先认清问题的结构长什么样。
论文里还留了一个没完全说透的问题:在数学和自由文本推理这些非符号化的任务里,残差和目标锚点都是"学出来的代理",不是像AC问题那样有精确的符号规则可依。这意味着SAGE在这些场景下的理论保证是打了折扣的,效果好不好很大程度依赖这些代理指标学得准不准。这算是一个诚实的局限,也是留给后续研究的一个缺口。
如果AI推理真的要走得更远,或许答案不在于让它跑得更快,而在于先给它画一张更靠谱的地图。
Q&A
Q1:SAGE是什么,主要解决什么问题?
A:SAGE是弗吉尼亚理工等机构提出的一种训练框架,全称结构可容许性引导探索,专门用来缓解大语言模型在长程推理任务中的探索偏差和复合偏差,通过代数稀疏化和双曲结构引导两种信号在训练阶段优化模型行为,不增加推理时的额外开销。
Q2:SAGE在Andrews-Curtis问题上效果如何?
A:论文显示SAGE在AC Validity指标上提升19.2%到26.0%,在Lean-Verified Proofs指标上所有测试模型都提升超过13%,其中Qwen3模型达到接近8倍的增长,这是该框架在真实世界长程推理任务上最亮眼的成绩。
Q3:探索偏差和复合偏差具体指什么?
A:探索偏差指模型在指数级膨胀的推理路径中,容易被局部看似合理但实际不可行的分支吸引;复合偏差指在稀疏奖励下,早期一个小的推理偏差因缺乏中间反馈而不断累积,最终导致整条推理链失败。







