arXiv:2607.24471cs.CL2026-07中稿 · COLM被引 1

研究变压器如何在不改变提示的情况下自动切换求解策略。

Grounding latent algorithm routing in transformer reasoning

论文配图:Grounding latent algorithm routing in transformer reasoning
图 1 · 摘自论文原文
  • 通过潜变量路由机制,让模型根据数据生成方式切换不同求解偏好。
  • 306M模型路由准确率达84.1,接近理想路由水平的80.9%。
  • 结果在自然语言、乱序支持等干扰下仍稳定,适合研究模型内部推理机制。

当前上下文学习研究的核心问题是:变压器能否围绕不同归纳偏置家族组织案例级适应?本文通过潜变量路由,在受控环境下研究该问题——即在保持提示形式不变的前提下,求解家族偏好随潜在数据生成模式变化,且对干扰因素鲁棒,可通过针对性激活干预调控,同时保持较高答案质量。我们提出 ROUTEBENCH 基准,其不同任务范式分别偏好全局收缩、稀疏性、鲁棒性与局部性,由岭回归、Lasso、Huber、kNN 等代表实现。在从头训练的密集解码器变压器(参数量44M-612M)上,306M模型关闭了80.9%的最优路由差距,路由F1达84.1。该效果在自然语言表述、支持集乱序、词汇改写及统一四路路由设置下依然显著。更强的自适应方案(如输入条件软混合、无监督Gumbel路由)虽缩小差距,但在路由F1与跨域性能上仍低于306M与612M模型。探针控制与匹配激活修补实验进一步表明,路由相关内部方向可被解码且功能上参与一致输出行为。这些结果提供受控证据,证明密集变压器在ROUTEBENCH上可发展出类路由内部变量,但不支持预训练语言模型中存在通用路由或任意自然语言推理中的广泛适用性。

原文摘要 · Abstract (English)

A central question in the in-context learning literature is whether transformers can organize episode-level adaptation around different inductive-bias families. We study this question in a controlled setting through latent algorithm routing: route-like behavior in which the solver-family preference changes with the latent data-generating regime while prompt form is held fixed, remains stable under nuisance perturbations, and is selectively influenced by targeted activation interventions without large losses in answer quality. We introduce ROUTEBENCH, a diagnostic benchmark whose regimes differentially favor global shrinkage, sparsity, robustness, and locality, operationalized by ridge-like, lasso-like, Huber-like, and kNN-like family representatives. Across dense decoder-only transformers trained from scratch at 44M-612M parameters, a 306M model closes 80.9 percent of the oracle-routing gap and achieves route F1 of 84.1. The effect remains substantial under natural-language renderings, shuffled supports, lexical paraphrases, and a unified four-way routing setting. Stronger adaptive alternatives, including an input-conditioned soft mixture and an unsupervised Gumbel router, narrow the gap but remain below the 306M and 612M models on route F1 and OOD performance. Probe controls and matched activation-patching controls further show that route-relevant internal directions are decodable and functionally involved in solver-family-consistent output behavior. These results provide controlled evidence that dense transformers trained on ROUTEBENCH can develop route-like internal variables, but they do not establish universal routing in pretrained language models or unrestricted natural-language reasoning.

模型推理潜变量路由变压器机制基准测试

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。