Anytime-unrolled EqLM (121M): parity ratio 0.991 vs param-matched explicit baseline
Browse files- README.md +46 -21
- pytorch_model.pt +3 -0
README.md
CHANGED
|
@@ -1,37 +1,62 @@
|
|
| 1 |
---
|
| 2 |
-
license:
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
-
#
|
| 6 |
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
-
##
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
| 12 |
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
|
|
|
| 18 |
|
| 19 |
-
|
| 20 |
|
| 21 |
-
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
-
|
| 24 |
-
|--------|-------|
|
| 25 |
-
| final_loss | 2.8680 |
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
-
##
|
| 29 |
|
| 30 |
-
|
|
|
|
|
|
|
| 31 |
|
| 32 |
-
##
|
| 33 |
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language: [en]
|
| 4 |
+
tags: [deep-equilibrium, weight-tying, babylm, equilibrium-model, kinetic-ai]
|
| 5 |
+
datasets: [BabyLM-community/BabyLM-2026-Strict]
|
| 6 |
---
|
| 7 |
|
| 8 |
+
# kinetic-eqlm-anytime-121m-babylm
|
| 9 |
|
| 10 |
+
A weight-tied transformer block solved to a fixed point, trained with anytime
|
| 11 |
+
unrolled supervision. Its depth is a stopping criterion rather than an
|
| 12 |
+
architecture: the same 120.7M parameters are applied repeatedly until the
|
| 13 |
+
representation stops moving, instead of being spread across distinct layers.
|
| 14 |
|
| 15 |
+
## What it is for
|
| 16 |
|
| 17 |
+
This checkpoint exists to answer one question at a like-for-like setting: can a
|
| 18 |
+
model whose depth is an equilibrium computation match a conventional stack of
|
| 19 |
+
the same size trained on the same data for the same number of steps?
|
| 20 |
|
| 21 |
+
| | this model | explicit baseline |
|
| 22 |
+
|---|---|---|
|
| 23 |
+
| parameters | 120.7M | 123.8M |
|
| 24 |
+
| depth | one tied block, iterated | 12 distinct layers |
|
| 25 |
+
| BLiMP ratio vs baseline | **0.991** (3 seeds: 0.971 / 1.033 / 0.970) | 1.000 |
|
| 26 |
|
| 27 |
+
The mean ratio of 0.991 is statistical parity, and one seed exceeds its baseline.
|
| 28 |
+
Parity is the honest description; this model is not better than a conventional
|
| 29 |
+
transformer of the same size, it is equal to one while storing its depth as
|
| 30 |
+
iteration rather than as parameters.
|
| 31 |
|
| 32 |
+
## Training
|
| 33 |
|
| 34 |
+
Anytime-unrolled supervision: cross-entropy is applied to intermediate iterates
|
| 35 |
+
at depths 4, 8 and 12 rather than only at the final one, which makes truncated
|
| 36 |
+
solves usable and closes the width gap that earlier fixed-point training left
|
| 37 |
+
open. BabyLM strict-small, matched data, steps and batch against the baseline.
|
| 38 |
|
| 39 |
+
## Honest limitations
|
|
|
|
|
|
|
| 40 |
|
| 41 |
+
Training supervises plain unrolled iterates while evaluation budgets cap an
|
| 42 |
+
Anderson-accelerated solver, so the two use different algorithms at the same
|
| 43 |
+
nominal depth; the parity figure is a claim about the end-to-end configuration
|
| 44 |
+
rather than about the solver in isolation. Converting a *pretrained* large model
|
| 45 |
+
to this topology is a different matter and does not work at budgets we could
|
| 46 |
+
afford: at 1.7B, conversion retained 0.354 of above-chance benchmark capability
|
| 47 |
+
and grade-school arithmetic collapsed from 0.457 to 0.017 while perplexity moved
|
| 48 |
+
only to 1.19x baseline. Train in this topology rather than converting into it.
|
| 49 |
|
| 50 |
+
## Provenance
|
| 51 |
|
| 52 |
+
Every number traces to a results file with a resolved-config hash and commit in
|
| 53 |
+
https://github.com/SharathSPhD/game-llm — findings F16 through F24 for this
|
| 54 |
+
model, F25/F26/F35 for the conversion results.
|
| 55 |
|
| 56 |
+
## Loading
|
| 57 |
|
| 58 |
+
```python
|
| 59 |
+
import torch
|
| 60 |
+
blob = torch.load("pytorch_model.pt", map_location="cpu", weights_only=True)
|
| 61 |
+
# blob["config_dict"] -> EqLMConfig fields; blob["model_class"] == "EqLM"
|
| 62 |
+
```
|
pytorch_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:24704232f3263a10494df06b70370b6f7ee87e5a2e2e25c58fb567c6dbec3010
|
| 3 |
+
size 482926309
|