arXiv Computation and Language
Oct 12

One Model to Loop Them All: Elastic Latent Recurrence for Language Models

The paper introduces Elastic Latent Recurrence (ELR), a training recipe that enables looped language models to match the accuracy of non-recurrent models at a single loop while allowing additional loops to improve accuracy with more compute. ELR achieves this through gated state updates with learned step sizes, mixed loop counts during training, and conditioning on loop index and budget. The method also reduces training and inference overhead in Mixture-of-Experts models and produces a single checkpoint, Koi‑4B, that performs competitively with state‑of‑the‑art models on multiple benchmarks.

By Faisal Ladhak, Shantanu Acharya, Santiago Akle Serano, Simeng Sun, Boris Ginsburg, Krishna C Puvvada
arXiv Machine Learning
Oct 8

LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superposition

LayerRoPE introduces a depth‑wise magnitude and angular superposition mechanism that replaces per‑layer normalization gains with a single shared vector and depth‑conditioned scalars, reducing parameters and computational cost. Across 16 pre‑trained large language models and a 100B+ token model ladder, LayerRoPE outperforms traditional Pre‑, Post‑, and Peri‑Norm schemes, achieving comparable or better performance with significantly less compute and improved convergence up to 512 layers. The method demonstrates that widening the residual stream, rather than shrinking it, enhances depth stability and learning‑rate sensitivity, and it transfers effectively to latent and vision transformer architectures.

By Shikhar Srivastava, Christopher Kanan
arXiv Machine Learning
5d ago

There Is No Free Loop: When Does Looping a Trained Model Pay for Its Inference Compute?

The paper investigates whether adding a loop—re‑applying a block of layers—to a trained language model is worth the extra inference cost. Experiments on Qwen3 and Qwen3.5 models from 0.8B to 9B parameters show that only the 0.8B looped model outperforms both a non‑looped model with the same parameters and a model with the same FLOPs. For larger models (4B and 9B), the looped version does not beat the baseline trained without loops, indicating that loops only pay off for small models with a fixed parameter budget.

By Rhea Sanjay Sukthanker, Pashmina Cameron