arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.24809cs.LGcs.AI

Nautile-370M:在小推理模型中融合频谱记忆与注意力

Nautile-370M: Spectral Memory Meets Attention in a Small Reasoning Model

Maixent Chenebaux

AI总结:

Nautile-370M是一款3.7亿参数的小语言模型,通过融合频谱记忆与注意力机制,在有限参数和推理预算下实现高效推理。

摘要:

我们提出了Nautile-370M,一款3.7亿参数的小语言模型,旨在在严格的参数和推理预算下实现高效的推理。Nautile-370M采用混合主干结构,其中两个SeqCond Attention(SCA)层和一个线性时间频谱序列操作符(受SeqCondenser启发)交替使用一个Transformer层。这种设计旨在保留结构化序列模型的长上下文效率和状态跟踪优势,同时保持注意力机制的token到token路由能力。该模型在Google TPU Research Cloud(TRC)计划提供的单个Cloud TPU v4-64 pod slice上进行训练;随后的强化学习阶段在单个NVIDIA DGX Spark上进行。我们证明SCA读取机制可以精确检索前缀摘要中的任何单个token,并且可以将softmax注意力的任何输出作为特殊情况重现,从而证明SCA在连续极限下至少与完全自注意力一样具有表现力。我们还描述了训练数据管道,并概述了一个专门用于推理、验证和响应质量的强化学习阶段。

英文摘要:

We present Nautile-370M, a 371-million-parameter small language model designed for efficient reasoning under strict parameter and inference budgets. Nautile-370M uses a hybrid backbone in which two SeqCond Attention (SCA) layers, a linear-time spectral sequence operator inspired by SeqCondenser, alternate with one transformer layer. This design aims to retain the long-context efficiency and state-tracking benefits of structured sequential models while preserving the expressive token-to-token routing of attention. The model was trained on a single Cloud TPU v4-64 pod slice provided through the Google TPU Research Cloud (TRC) program; the subsequent reinforcement learning stage was carried out on a single NVIDIA DGX Spark. We prove that the SCA readout mechanism can exactly retrieve any individual token from the prefix summary and can reproduce any output of softmax attention as a special case, establishing that SCA is at least as expressive as full self-attention in the continuous limit. We also describe the training data pipeline and outline a reinforcement learning stage specialized for reasoning, verification, and response quality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑