论文

YANchor-4B:线性时间、常量状态的长程推理语言模型

YANchor-4B: Effective Long-Horizon Reasoning in O(N) Time with O(1) Memory

模型架构模型推理上下文与知识递归架构推理加速上下文工程

摘要

长期推理需要以可管理的生成成本访问早期信息。全历史注意力会导致存储和计算不断增长,而循环压缩可能会丢失精确的细节。因此,我们提出了 YANchor-4B,这是一种通用的循环模型,它将关键的记忆保留为 ANchors,以便在后续推理过程中进行检索。除了 $O(N)$ 时间生成和 $O(1)$ 内存之外,YANchor 通过其多维内存机制实现有效的长视野推理。例如,在具有挑战性的数学问题上,它在 AIME 2024--2026 上达到了 82.93% 的平均 pass@1 成绩,在 HMMT 上达到了 63.64% 的成绩,大大优于线性时间、恒定状态的对应模型,包括较大的模型。它还提供比 H100 上的 Transformer 和混合基线高几倍的批量长代吞吐量。此外,数十个基准测试的评估证明了YANchor在通用能力方面的优势。

YANchor-4B:线性时间、常量状态的长程推理语言模型:论文原图
图 3:YANchor-4B 的架构。循环层与局部注意力和记忆模块交织在一起。插图显示了如何写入显式记录、如何将其保留为内存锚点以及如何通过独立的读取路径进行检索。所有历史路径都有固定的容量。图片来源:美国宇航局。