arXiv:2604.12946cs.LG2026-04被引 32

提出稳定循环语言模型架构,用数学约束提升训练稳定性与推理效率。

Parcae: Scaling Laws For Stable Looped Language Models

  • 将循环结构建模为时变动态系统,通过控制参数谱范数防止梯度爆炸。
  • 在固定参数量下,训练时提升算力可使困惑度降低6.3%,测试时计算量可指数扩展。
  • 适合追求高效推理、算力优化的部署场景,尤其在资源受限条件下表现突出。

传统固定深度架构通过增加参数量来提升质量,但代价是更高的内存或数据开销。循环架构则通过反复传递激活值增加计算量,更具潜力。然而现有方法常因残差爆炸和损失突增而训练不稳定。本文将循环视为残差流上的非线性时变动力系统,通过线性近似发现不稳定性源于注入参数的过大谱范数。为此提出Parcae,采用负对角参数化离散化,约束谱范数,实现稳定训练。实验显示,相比先前大规模循环模型,其验证困惑度降低最多6.3%。在训练中,我们推导出在固定参数量下按幂律增加算力的可预测规律;在测试时,算力可按饱和指数衰减规律扩展。当扩展至1.3B参数时,在固定参数与数据预算下,相较于强基线模型,CORE与Core-Extended得分分别提升2.99与1.18点,达到两倍大模型相对质量的87.5%。

原文摘要 · Abstract (English)

Traditional fixed-depth architectures scale quality by increasing training FLOPs, typically through increased parameterization, at the expense of a higher memory footprint, or data. A potential alternative is looped architectures, which instead increase FLOPs by sending activations through a block of layers in a loop. While promising, existing recipes for training looped architectures can be unstable, suffering from residual explosion and loss spikes. We address these challenges by recasting looping as a nonlinear time-variant dynamical system over the residual stream. Via a linear approximation to this system, we find that instability occurs in existing looped architectures as a result of large spectral norms in their injection parameters. To address these instability issues, we propose Parcae, a novel stable, looped architecture that constrains the spectral norm of the injection parameters via discretization of a negative diagonal parameterization. As a result, Parcae achieves up to 6.3% lower validation perplexity over prior large-scale looped models. Using our stable looped architecture, we investigate the scaling properties of looping as a medium to improve quality by increasing FLOPs in training and test-time. For training, we derive predictable power laws to scale FLOPs while keeping parameter count fixed. Our initial scaling laws suggest that looping and data should be increased in tandem, given a fixed FLOP budget. At test-time, we find that Parcae can use looping to scale compute, following a predictable, saturating exponential decay. When scaled up to 1.3B parameters, we find that Parcae improves CORE and Core-Extended quality by 2.99 and 1.18 points when compared to strong Transformer baselines under a fixed parameter and data budget, achieving a relative quality of up to 87.5% a Transformer twice the size.

循环架构稳定性优化算力扩展语言模型

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