arXiv:2608.09432cs.CLcs.AI2026-08

不用位置编码的轻量级语言模型,靠状态空间自动捕捉顺序信息。

ZetaGPT: A Reference Implementation of Positional--Encoding--Free State--Space--Attention Language Models

  • 用因果状态空间方程隐式编码位置信息,替代传统位置嵌入
  • 在不依赖显式位置编码下实现与Transformer相当的性能
  • 开源完整训练流程,适合研究、教学和快速原型验证

基于Transformer的语言模型依赖自注意力机制,其计算具有置换等变性,缺乏内在的词元顺序表示能力。现有架构通过学习的位置嵌入或手工设计的位置编码(如旋转位置编码RoPE)显式引入位置信息,将位置感知视为一种结构能力而非模型固有属性。本文提出一种无需显式位置编码的架构,利用因果状态空间方程在注意力计算前隐式编码序列信息。每个模型块在自注意力前应用因果状态空间方程,使递归状态动态将顺序信息融入词元表示,后续注意力层作用于已含位置感知的表示,无需额外位置编码,同时保持自注意力的建模能力。我们构建了名为 extsc{ZetaGPT}的紧凑混合语言模型,专为研究、快速原型设计、算法验证及教育应用而设。该模型提供完全开源的端到端训练流水线,涵盖数据集构建、分词器训练、预训练、监督微调、人类反馈强化学习(RLHF)以及纯强化学习驱动的思维链(CoT)推理。据我们所知, extsc{ZetaGPT}是首个无显式位置编码的小型开源语言模型,为无位置编码语言模型的开发与实证研究提供了紧凑且可复现的参考实现。

原文摘要 · Abstract (English)

Transformer-based language models rely on self-attention, whose computation is permutation-equivariant and therefore lacks an intrinsic mechanism for representing token order. Existing architectures address this limitation by explicitly incorporating positional information through learned positional embeddings or hand-crafted positional encodings, such as rotary positional encoding (RoPE), treating positional information as an architecturally acquired capability rather than an inherent property of the model. Motivated by the pursuit of positional-encoding-free architectures, this work explores a language model architecture that integrates causal state-space equations to implicitly encode positional information before attention computation. Specifically, each model block applies a causal state-space equation before self-attention, allowing recurrent state dynamics to encode sequential information into token representations. Consequently, subsequent attention layers operate on position-aware representations without requiring explicit positional encodings while retaining the expressive modeling capacity of self-attention. We present \textsc{ZetaGPT}, a compact hybrid language model designed for research, rapid prototyping, algorithm verification, and educational applications. In addition to the proposed architecture, \textsc{ZetaGPT} provides a fully open-source, end-to-end training pipeline encompassing dataset construction, tokenizer training, pretraining, supervised fine-tuning, reinforcement learning from human feedback (RLHF), and chain-of-thought (CoT) reasoning via pure reinforcement learning. To the best of our knowledge, \textsc{ZetaGPT} is the first open-source small language model without explicit positional encoding and establishes a compact, reproducible reference implementation for the development and empirical study of positional-encoding-free language models.

语言模型状态空间位置编码开源

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