← 返回 2026-03-18

递归语言模型遇上不确定性:用于长上下文的自反思程序搜索的惊人有效性 Recursive Language Models Meet Uncertainty: The Surprising Effectiveness of Self-Reflective Program Search for Long Context

Keivan Alizadeh, Parshin Shojaee, Minsik Cho, Mehrdad Farajtabar 📅 2026-03-07 👍 13 2026-07-13 08:36
不确定性估计 程序化交互 自反思 递归语言模型 长上下文推理

通过不确定性感知的自反思机制选择最优程序轨迹,无需显式递归即可超越RLM

前置知识

递归语言模型

RLM是一种将长上下文作为编程环境中的外部变量来处理的方法,让模型生成程序来查询、切片和递归地与上下文交互。通过外部化上下文交互为程序执行,RLM能够扩展模型的有效推理范围,超越通常的提示词限制。

本文的方法建立在RLM框架之上,理解RLM的工作原理对于理解SRLM如何改进和超越RLM至关重要

不确定性估计

不确定性估计是指量化模型对其预测的确信程度的技术。在语言模型中,不确定性可以通过多种方式估计,包括基于采样的一致性检查、模型口头表达的置信度分数,以及行为信号如推理轨迹长度等。

SRLM的核心创新就是利用多种不确定性信号来指导程序搜索,理解不同类型的不确定性估计方法是理解本文方法的关键

程序化上下文交互

程序化上下文交互是指将长上下文视为可编程对象,通过编写和执行程序来查询、操作和分析上下文内容,而不是直接将整个上下文作为输入喂给模型。这种方式允许模型以结构化、迭代的方式与上下文进行交互。

这是本文研究的基础框架,SRLM和RLM都是在这个框架下工作的,理解这个概念对于理解整个方法的动机和实现很重要

研究动机

长上下文处理仍然是语言模型的核心挑战,即使有了扩展的上下文窗口,模型经常无法可靠地提取、推理和使用长上下文中的信息。现有的Recursive Language Models通过代理方式将长上下文分解为递归子查询来解决这个问题,虽然很有前景,但RLM的成功关键取决于如何选择上下文交互程序的轨迹,这个问题尚未被探索。模型在每个步骤都需要决定要检查哪个上下文片段、如何制定中间自查询、提出什么子问题以及如何聚合这些编程步骤和部分结果,因此最终预测高度敏感于在上下文交互推理期间实例化的特定程序轨迹。尽管如此,RLM目前主要依赖固定的递归方案,缺乏评估和选择替代推理轨迹的原则性机制。

本文的目标是本文的目标是研究递归本身是否是长上下文推理的关键因素,还是真正的瓶颈是在不确定性下如何选择候选交互程序。本文引入Self-Reflective Program Search for Long Context框架,通过不确定性感知的自反思来增强基于编程的上下文交互,利用三个内在信号作为模型内部不确定性的互补指标,使模型能够在不需要外部监督的情况下评估和比较候选上下文交互程序。

与已有工作不同的是,本文的独特切入角度是将长上下文推理重新定位为一个程序搜索问题,重点放在如何选择和评估候选上下文交互程序轨迹上,而不是递归本身。与RLM依赖固定的递归方案不同,SRLM通过模型自身的内部不确定性信号来指导程序轨迹的选择,这是一个尚未被充分探索的研究方向。本文挑战了递归是RLM性能主要驱动因素的假设,并提出了一个更简单但更有效的替代方案。

核心方法

SRLM的整体思路是将长上下文推理视为一个程序搜索问题,通过模型自身的内部不确定性信号来指导候选上下文交互程序的选择。给定查询和长上下文,SRLM从模型策略中独立采样K个候选程序,每个候选程序代表在上下文上的不同推理轨迹。然后利用三个互补的不确定性信号来评估这些候选程序的质量,包括基于采样的不确定性(自一致性)、语义不确定性(口头置信度)和行为不确定性(推理轨迹长度),最后选择最优程序输出最终预测。

SRLM的核心创新点在于利用模型自身的内部不确定性信号来实现自反思的程序搜索,而不需要任何外部监督或奖励模型。与RLM依赖显式的递归子调用不同,SRLM解耦了上下文交互质量与递归结构,将长上下文推理改进的重点转移到候选上下文交互程序轨迹的选择机制上。通过结合自一致性、口头置信度和推理轨迹长度这三个互补的不确定性信号,SRLM能够在不需要显式递归或自查询机制的情况下,匹配甚至超越RLM的性能。

方法步骤详情

SRLM方法的第一步是候选程序采样,从模型策略中独立采样K个候选程序,每个候选程序代表不同的推理轨迹。第二步是自一致性过滤,计算每个候选答案的经验频率,保留与 plurality answer 一致的候选程序集合。第三步是口头置信度计算,在每个中间生成步骤要求模型报告置信度分数,并在对数空间中聚合。第四步是轨迹长度计算,计算每个程序的推理轨迹总长度。第五步是联合不确定性评分,将口头置信度与轨迹长度相乘得到联合不确定性分数,在一致候选集中选择最优程序,最终输出最优程序的预测结果。

技术新颖性

SRLM的技术新颖性在于它首次系统地研究了程序化上下文交互框架中的轨迹选择问题,并提出了一个完全基于模型内部不确定性信号的自反思解决方案。与依赖外部监督或奖励模型的方法不同,SRLM利用三个完全源自模型自身生成过程的不确定性信号,不需要verifier、奖励模型或外部标注数据。这种自反思的设计使得SRLM能够在不改变底层模型的情况下,显著提升长上下文推理性能。此外,SRLM解耦了上下文交互质量与递归结构,挑战了递归是RLM性能主要驱动因素的假设,这是一个重要的概念创新。

Overview of SRLM, a framework that augments programmatic context interaction reasoning with uncertainty-aware self-reflection.
Figure 1: Overview of SRLM, a framework that augments programmatic context interaction reasoning with uncertainty-aware self-reflection.

实验结果

核心发现包括:在BrowseComp+、OOLONG和LongBench-v2 CodeQA三个基准测试上,SRLM一致地优于最先进的基线,在相同的wall-clock时间预算下比RLM最高提升22%。使用Qwen3-Coder-480B-A35B骨干模型时,SRLM在LongBench-v2 CodeQA上达到64.9%准确率,比RLM的59.8%提升5.1个百分点;在BrowseComp+上达到59.7%准确率,比RLM的37.1%提升22.6个百分点;在OOLONG上达到51.8%准确率,比RLM的45.7%提升6.1个百分点。使用GPT-5骨干模型时,SRLM在LongBench-v2 CodeQA上达到68.9%准确率,比RLM的59.5%提升9.4个百分点;在BrowseComp+上达到92.4%准确率,比RLM的86.0%提升6.4个百分点;在OOLONG上达到65.5%准确率,比RLM的53.0%提升12.5个百分点。更重要的是,递归本身不是RLM性能的主要驱动因素,简单的自反思程序搜索可以在不需要自查询或显式递归机制的情况下匹配或超越RLM。RLM的递归过程对上下文长度变化更敏感,当上下文长度在模型的原始上下文窗口内时,递归RLM推理通常会相对于基线模型降低性能,而SRLM在短上下文和长上下文上都提供更稳健和一致的改进。RLM在语义密集型任务上效果较差,其中启发式程序搜索是不够的,需要更广泛的上下文理解,而SRLM中的自反思提供了更好的语义信号来指导这些具有挑战性的长上下文场景中的推理。

Performance comparison of SRLM against baselines on long-context benchmarks from [75].
Table 1: Performance comparison of SRLM against baselines on long-context benchmarks from [75].
Performance across context lengths on OOLONG and LongBench-v2 Full datasets.
Figure 2: Performance across context lengths on OOLONG and LongBench-v2 Full datasets.
Accuracy versus cost pareto comparison of RLM and SRLM (no sub-call) on long-context settings of benchmarks under GPT-5 and Qwen3-Coder-480B.
Figure 3: Accuracy versus cost pareto comparison of RLM and SRLM (no sub-call) on long-context settings of benchmarks under GPT-5 and Qwen3-Coder-480B.
Comparison of SRLM, RLM, and Base LLM across LongBench-v2 domains (averaged across backbone models).
Figure 4: Comparison of SRLM, RLM, and Base LLM across LongBench-v2 domains (averaged across backbone models).
Results of ablation experiments across SRLM's variants (averaged across backbones and recursive/nonrecursive runs).
Figure 5: Results of ablation experiments across SRLM's variants (averaged across backbones and recursive/nonrecursive runs).
查看结构化数据
任务指标本文基线提升
LongBench-v2 CodeQA accuracy (%) 68.9 (GPT-5) 59.5 (RLM, GPT-5) +9.4 pp
BrowseComp+ (1K documents) accuracy (%) 92.4 (GPT-5) 86.0 (RLM, GPT-5) +6.4 pp
OOLONG (131K tokens) accuracy (%) 65.5 (GPT-5) 53.0 (RLM, GPT-5) +12.5 pp
LongBench-v2 CodeQA accuracy (%) 64.9 (Qwen3-Coder-480B) 59.8 (RLM, Qwen3-Coder-480B) +5.1 pp
BrowseComp+ (1K documents) accuracy (%) 59.7 (Qwen3-Coder-480B) 37.1 (RLM, Qwen3-Coder-480B) +22.6 pp
OOLONG (131K tokens) accuracy (%) 51.8 (Qwen3-Coder-480B) 45.7 (RLM, Qwen3-Coder-480B) +6.1 pp

局限与改进

作者承认的局限性包括:本文采用了相对简单的自反思形式,基于内在不确定性信号来指导程序化上下文交互,虽然有效,但这代表了本文方法的一个局限性,并为未来研究留下了空间。本文在实验中使用了固定的候选程序数量K等于8,没有系统地研究这个超参数对性能的影响,也没有探索自适应的候选程序生成策略。口头置信度的elicitation可能会受到模型校准问题的影响,如果模型对其能力过度自信或过度谦虚,这可能会影响自反思信号的质量。本文主要在代码QA、浏览和合成任务上评估了SRLM,在更多真实世界任务上的泛化性能还需要进一步验证。我自己的观察包括:SRLM需要运行多个候选程序并评估它们,这增加了计算开销,虽然通过并行执行可以控制wall-clock时间,但总的token使用量仍然显著增加。自反思信号的质量很大程度上依赖于模型自身的元认知能力,对于能力较弱的模型,这些信号可能不够可靠。

独立分析的弱点

独立分析的弱点包括:在高度结构化和局部化的任务上,递归方法可能仍然具有优势,因为它们可以利用明确的子问题分解策略,而SRLM的自反思信号可能在这些任务上不够精细。SRLM的不确定性信号可能受到模型固有偏见的影响,例如模型可能倾向于生成较短的推理轨迹,这可能使得轨迹长度信号不够可靠。口头置信度的elicitation可能会干扰模型的正常推理过程,特别是在要求模型在每个步骤都报告置信度的情况下,这可能会影响推理质量。SRLM没有考虑候选程序之间的相互影响,每个候选程序是独立评估的,但实际上不同程序之间可能存在互补性或相互冲突。改进方向包括:探索更精细的递归与自反思结合机制,既利用显式的子问题分解又利用自反思信号;研究自适应的不确定性信号加权机制,根据任务类型和上下文特征动态调整不同信号的权重;探索候选程序的协同评估机制,考虑程序之间的互补性和相互关系;开发更robust的口头置信度elicitation方法,减少对正常推理的干扰。

未来方向

作者提出的未来工作方向包括:探索更丰富的内在自反思形式,在程序化上下文交互框架中超越显式递归子调用;设计与自反思信号集成的决策机制,实现推理的更早终止和改进的token使用控制。基于本文成果可延伸的研究方向包括:将SRLM的不确定性感知自反思机制扩展到其他agent框架和任务领域,如多agent协作、工具使用等;研究如何将自反思信号与强化学习或奖励模型结合,形成更强大的程序搜索策略;探索在训练过程中显式优化模型的自反思能力,提高不确定性信号的质量;研究如何将SRLM应用到更广泛的长上下文任务,如长文档摘要、长对话管理、长代码库理解等;探索在多模态场景中应用自反思程序搜索,处理长序列的图像、音频等多模态数据。

复现评估

复现评估方面,论文提供了相对详细的实验设置,包括使用的骨干模型、数据集、超参数设置和评估方法。然而,论文没有提供源代码或实现细节,这使得完全复现实验结果存在一定困难。GPT-5和Qwen3-Coder-480B都是大型商业模型,需要API访问权限,这增加了复现的成本和难度。论文使用的数据集大多是公开可用的,但BrowseComp+的特定子集和评估协议可能需要额外的工作来复现。从算力需求来看,由于SRLM需要运行多个候选程序并评估它们,总体的计算开销较大,但通过并行执行可以控制wall-clock时间。总体而言,复现本文实验的难度为中等偏高,主要挑战在于访问大型模型API和实现细节的缺失。