This article is a structured English summary of a Chinese technical deep-dive on Looped Language Models (LoopLM), centered on ByteDance Seed's Ouro model.
Key points
1. Core architecture innovation
- Parameter-shared looping: Instead of independent layers, a stack of N shared-weight layers is applied multiple times ("loop steps"), decoupling compute depth from parameter count. Ouro's 1.4B/2.6B versions use 4 loop steps (R4) and match standard 4B/8B Transformers on nearly all benchmarks (2–3x parameter compression).
- Latent-space iterative computation: Iterative refinement of hidden states forms a "latent chain of thought," avoiding the context-window bloat, latency, and faithfulness issues of explicit text-based CoT.
- Compatibility: Decoder-only Transformer with RoPE (with per-loop angular offset θ_i + t×Δθ), SwiGLU, RMSNorm (sandwich normalization for loop stability).
- Entropy-regularized exit gates: Training adds a KL(p_exit || p_uniform) term; a learned gate computes p_exit after each loop; inference stops when cumulative exit probability exceeds threshold θ.
- Deployment results (Ouro-1.4B): 50% of queries finish within 2 loops, 75% within 3, only ~5% need all 4. Average compute is 35–40% lower than matched standard Transformers on typical mixed workloads.
- Depth–complexity correlation: average loops ≈ 1.3 (TriviaQA), 2.1 (SQuAD), 3.2 (GSM8K), 3.5 (ReClor); depth is instance-adaptive.
- Depth extrapolation: trained at T=4, performance still improves at T=6 and saturates around T=8.
- Ouro was pretrained on 7.7T tokens (≈73% Nemotron-CC, 13% MAP-CC, 7.5% OpenCoder, 4.1% MegaMath-web, 2% Ultra-FineWeb-zh) via a six-stage pipeline (warmup, two 3T stable phases, 1.4T annealing, 20B-token 64K long-context, 300B mid-training), with an upcycling split into 1.4B/2.6B branches.
- Compared with post-hoc CoT fine-tuning, pretraining-internalized latent reasoning shows better out-of-distribution generalization, higher causal faithfulness (ARCHE analysis), and zero extra context consumption.
- Safety: on HEx-PHI, harmfulness scores consistently decrease as loop depth increases.
- Benchmarks: Ouro-2.6B outperforms Qwen3-8B on MMLU-Pro (55.73 vs 53.72), GSM8K (85.3 vs 81.5), MATH500 (90.85 vs 62.30); Ouro-1.4B scores 82.40 on MATH500 vs 59.60 for Qwen3-4B (+38% with 35% of the parameters).
- Knowledge-use hypothesis: control tasks show equal storage capacity (~2 bits/parameter, Capo task) but markedly better knowledge *manipulation* (Mano modular-arithmetic trees, multi-hop QA).
- Theory: looped models can solve graph reachability in O(log²D) loop steps vs O(D) sequential steps; latent iteration scales as O(T×d²) independent of vocabulary size and avoids CoT's KV-cache growth.
- Diminishing returns per loop: +15–20% (1→2), +8–12% (2→3), +3–5% (3→4), saturating beyond.
- Capability function extends from C ≈ f(P, D) to C ≈ f(P, T, D), where loop depth T is a *runtime-adjustable* scaling dimension ("runtime scaling").
- Deployment scenarios: T≤2 for edge real-time (≈0.3x cost), T=4 cloud standard, T=6–8 extrapolation for deep reasoning. INT4-quantized 1.4B fits under 1GB for on-device NPU inference.
- Enables compute-based pricing for MaaS (charging per loop depth) and higher data efficiency (~60% of samples needed on Mano).
- Throughput: 2,400 tok/s vs 1,800 (+33%); P50 latency 45ms vs 78ms (−42%); ~55% lower energy per 1K tokens; ~65% less memory; ~1.8x throughput on mixed workloads; training throughput at 85–90% of standard Transformers via hierarchical parallelism.
- Research directions: multi-scale/hierarchical looping, integration with external memory (iterative retrieval–reasoning), combining loops with MoE-style conditional computation.
- Theory: a three-dimensional scaling law L(N, D, T) and new tools for dynamic compute.
- Ecosystem: Ouro is open-sourced at http://ouro-llm.github.io with vLLM and SGLang integration.