qbz506 commited on
Commit
4a194a5
·
verified ·
1 Parent(s): 6010979

Anytime-unrolled EqLM (121M): parity ratio 0.991 vs param-matched explicit baseline

Browse files
Files changed (2) hide show
  1. README.md +46 -21
  2. pytorch_model.pt +3 -0
README.md CHANGED
@@ -1,37 +1,62 @@
1
  ---
2
- license: mit
 
 
 
3
  ---
4
 
5
- # EqLM — kinetic-eqlm-anytime-121m-babylm
6
 
7
- This is an equilibrium language model from the **Kinetic AI** research project.
 
 
 
8
 
9
- ## Architecture
10
 
11
- Config: {'vocab_size': 50257, 'd_model': 1704, 'n_heads': 12, 'd_ff': 6807, 'max_seq_len': 128, 'deq_max_iter': 12, 'deq_tol': 0.001, 'solver': 'anderson', 'jfb': False, 'dropout': 0.1, 'spectral_norm': True, 'residual_damping': 0.2, 'map_form': 'postln', 'aux_residual': False, 'lambda_aux': 0.1}
 
 
12
 
13
- ## Provenance
 
 
 
 
14
 
15
- - **Config Hash:** `33b106ea36f6f2db58eb1748a6169f269559b5d008ea87822816d214e0689b3d`
16
- - **Git Commit:** `unknown`
17
- - **Run Directory:** `/home/sharaths/projects/game-llm/results/exp13_seed43`
 
18
 
19
- All numbers in this model card trace to documented runs; see [research/memory/findings.md](https://github.com/SharathSPhD/game-llm/blob/main/research/memory/findings.md) for validation details.
20
 
21
- ## Metrics
 
 
 
22
 
23
- | Metric | Value |
24
- |--------|-------|
25
- | final_loss | 2.8680 |
26
 
 
 
 
 
 
 
 
 
27
 
28
- ## Disclaimer
29
 
30
- This is research-stage code. Numbers reported here are from `results.json` in the named run directory and trace to `research/memory/findings.md`. For baseline comparisons and metric definitions, consult the paper at https://github.com/SharathSPhD/game-llm/tree/main/paper.
 
 
31
 
32
- ## Links
33
 
34
- - [GitHub Repository](https://github.com/SharathSPhD/game-llm)
35
- - [Paper (arXiv/Site)](https://github.com/SharathSPhD/game-llm/tree/main/paper)
36
- - [Validated Findings](https://github.com/SharathSPhD/game-llm/blob/main/research/memory/findings.md)
37
- - [Kinetic AI Home](https://kinetic.kinetic-ai.workers.dev)
 
 
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