Weight-Tied Looped Depth for Data-Constrained Pretraining

Sebastian Miles


Abstract
The BabyLM Strict-Small track fixes the training corpus at 10M words and the budget at 10 epochs. Data cannot be added, so any gain has to come from the architecture. We ask whether weight-tied looped depth, applying the same transformer block stack twice per forward pass, helps under that constraint. On a 50M-parameter model built from nanochat it does. Held-out bits-per-byte (val_bpb, a per-byte cross-entropy that is comparable across tokenizers) falls from 1.3222 to 1.3175 as a three-seed mean, and every looped run lands below every baseline run, at no parameter cost. Downstream the effect points the same way, but we cannot resolve it: averaged over five zero-shot benchmarks under the official harness, the looped mean is 50.00 against the baseline’s 49.90, a gap of 0.10 points against a standard error of 0.17. The second pass is paid for in computation. Training takes 1.71× as long, and a 100M-parameter model trained for a comparable wall-clock time still scores better on both metrics. What looping buys here is parameter efficiency, not compute efficiency.

The baseline these experiments run on was tuned by a human-directed agentic loop. The researcher proposed each hypothesis; an agent implemented it, trained for the fixed budget around the clock on one GPU, and kept or reverted the change under a fixed rule on val_bpb. We treat the loop as methodology and report its rejected hypotheses alongside the accepted ones.
Anthology ID:
2026.babylm-main.41
Volume:
Proceedings of the Second BabyLM Workshop
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Leshem Choshen, Ryan Cotterell, Mustafa Omer Gul, Jaap Jumelet, Tal Linzen, Aaron Mueller, Suchir Salhan, Raj Sanjay Shah, Alex Warstadt, Ethan Gotlieb Wilcox
Venues:
BabyLM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
525–538
Language:
URL:
https://aclanthology.org/2026.babylm-main.41/
DOI:
Bibkey:
Cite (ACL):
Sebastian Miles. 2026. Weight-Tied Looped Depth for Data-Constrained Pretraining. In Proceedings of the Second BabyLM Workshop, pages 525–538, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Weight-Tied Looped Depth for Data-Constrained Pretraining (Miles, BabyLM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.babylm-main.41.pdf