Tiny Local Recurrence with Adaptive Computation
Implementation & benchmark of arXiv:2608.14019 — Emergent Models: Intelligence from Tiny Substrates
Source paper: Emergent Models: Intelligence from Tiny Substrates arXiv:2608.14019 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a stack of independently parameterized residual or MLP blocks with a small latent grid or vector repeatedly updated by one shared transition rule. Let the number of updates depend on the current latent state, so easy examples terminate early while hard examples receive more computation, potentially improving parameter efficiency and extrapolation.
Formulas
Mathematical statement
The paper's central computation is the latent dynamical system $x\to E(x)\to s_0\xrightarrow{f^T}s_T\to D(s_T)\to y$, where $x$ is the external input, $E$ is an encoder, $s_0$ is the latent state, $f$ is one fixed transition map reused at every microstep, $T(s_0)$ is an input-dependent number of iterations, $D$ is the decoder, and $y$ is the output. The trainable supervised objective is $\theta^\star\in\arg\min_\theta\mathbb{E}[\ell(\hat y,y)]$ with $\hat y=\Phi_\theta(x)$. Implement $s_{t+1}=f_\theta(s_t)$ using a local convolutional rule or shared low-rank MLP, and estimate a halting probability $h_t=\sigma(w_h^\top\operatorname{pool}(s_t)+b_h)$; the expected computation is $T$ obtained by accumulating halting mass. The paper's latent-universality claim says that, for some substrates, holding $f$ and the interface fixed while varying the initial condition can realize any partial computable function. This does not prove trainability, but motivates placing task information in $E(x)$ and keeping the recurrent rule extremely small and reusable.
Implementation notes
Integrate this as a replacement for a conventional stack of Transformer feed-forward blocks or residual blocks, leaving the surrounding embedding and output head unchanged. Use a latent tensor $s\in\mathbb R^{C\times H\times W}$ for the simplest experiment, or $s\in\mathbb R^d$ for a vector version. The encoder writes the token or example representation into designated channels or positions, and the decoder reads the final state. Define one shared transition as s = s + alpha * local_rule(norm(s)), where local_rule is a depthwise 3x3 convolution followed by a pointwise two-layer MLP; all weights are reused at every microstep. Pseudocode: s=E(x); mass=0; state_acc=0; for t in range(Tmax): s=f(s); h=sigmoid(wh@pool(s)+bh); delta=min(h,1-mass); state_acc += delta*s; mass += delta; if mass >= 1-eps: break; y=D(state_acc + eps*s). Compute the task loss plus the ponder penalty $\lambda T$ shown above. The paper contributes the fixed-rule recurrence, interface separation, and adaptive $T$ interpretation; the halting probability and penalty are an explicit engineering adaptation. First test on MNIST, CIFAR-10, and a synthetic arithmetic extrapolation task using a 6-layer residual MLP baseline with matched parameter count. Measure validation accuracy, average microsteps, wall-clock time, FLOPs, and extrapolation accuracy outside the training range. Success means equal accuracy with at least 2x fewer learned parameters, or lower average computation on easy examples without a worse hard-example error curve.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significant win ✓
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built a shared adaptive recurrence classifier and a six-block residual MLP baseline with fixed seeds and CUDA fallback. The toy checks confirmed the predicted stability boundary, exact halting-time formula, and monotonic reduction in halting time as the bias increases. The idea matched baseline accuracy while using 74.2% fewer parameters and 31.5% fewer average microsteps, but it was slower wall-clock because of recurrent Python/control-flow overhead.
- Agent confidence
- 8/10
- Baseline
- accuracy=0.990234375; parameters=13218; average microsteps=6.0; train time=5.69s
- Idea
- accuracy=0.990234375; parameters=3411; average microsteps=4.1074; train time=14.93s; all three mechanism checks confirmed
Limitations: Only a small 2D synthetic classification task was tested. MNIST, CIFAR-10, arithmetic extrapolation, explicit FLOP measurements, hard/easy error curves, and larger-scale generalization were not tested.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.14019)
Beats the tuned baseline by 46.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.3809
- Idea mean
- 0.2057
- Effect (Δ)
- -0.1752 (−46.0%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the structurally matched sequence track was tested, using 400 training and 200 test examples for 18 epochs. Wall-clock time, FLOPs, parameter counts, and the tabular, vision, and dynamics tracks were not evaluated. The recurrence uses a Python loop, so runtime overhead was not measured.
How to run: python3 stage2_bench.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Worked ✓
Agent confidence: 8/10
The shared adaptive recurrence significantly improved sequence forecasting test MSE on the fixed benchmark. Across 8 paired seeds it won on every seed, with mean MSE 0.2057 versus 0.3809 for the swept transformer baseline, delta -0.1752 and permutation p=0.0081. Trained-model measurements also confirmed early halting, averaging 1.328 microsteps out of Tmax 6.
- Baseline
- MSE=0.3809081800 +/- 0.0454098553; best lr=0.001; epochs=18
- Idea
- MSE=0.2057067603 +/- 0.0273858546; best lr=0.003; epochs=18; delta=-0.1752014197; p=0.0081; 8/8 wins; average microsteps=1.3281/6
Benchmark result
Beats the tuned baseline by 46.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.3809
- Idea mean
- 0.2057
- Effect (Δ)
- -0.1752 (−46.0%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the structurally matched sequence track was tested, using 400 training and 200 test examples for 18 epochs. Wall-clock time, FLOPs, parameter counts, and the tabular, vision, and dynamics tracks were not evaluated. The recurrence uses a Python loop, so runtime overhead was not measured.
How to run: python3 stage2_bench.py
Artifacts
- bench_experiment.py 5.7 KB View Raw
- bench_experiment_corrected.py 3.5 KB View Raw
- bench_report.json 4.1 KB View Raw
- bench_report_corrected.json 3.4 KB View Raw
- experiment.py 4.6 KB View Raw
- report.md 1.2 KB View
- report_bench_2026-09-02T231432.md 3.7 KB View
- results.json 2.9 KB View Raw
- stage2_bench.py 5.0 KB View Raw