instanton model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
license: mit
|
| 4 |
+
tags:
|
| 5 |
+
- babylm
|
| 6 |
+
- babylm-2026
|
| 7 |
+
- strict-small
|
| 8 |
+
- gpt2
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# instanton-baseline
|
| 12 |
+
|
| 13 |
+
**instanton** — a small model that trains almost instantaneously (~12 min on one
|
| 14 |
+
A100). BabyLM Challenge 2026, **Strict-Small track** (<=10M words). Part of
|
| 15 |
+
*Reweighting Child-Directed and Conversational Data for Sample-Efficient
|
| 16 |
+
BabyLM Pretraining*.
|
| 17 |
+
|
| 18 |
+
- **Architecture:** GPT-2-style decoder-only transformer, from-scratch init
|
| 19 |
+
(no pretrained weights): 6 layers, 512 hidden, 8 heads, 512-token context,
|
| 20 |
+
27.4M parameters (tied embeddings).
|
| 21 |
+
- **Objective:** causal language modeling.
|
| 22 |
+
- **Tokenizer:** byte-level BPE, 16k vocab, trained **only** on this run's
|
| 23 |
+
<=10M-word training mixture.
|
| 24 |
+
- **Training (v2 recipe):** 10 epochs (~100M words exposure), packed 512-token
|
| 25 |
+
sequences, batch 32 (no gradient accumulation, ~9.9k steps), lr 1e-3 cosine
|
| 26 |
+
(5% warmup), weight decay 0.1, bf16, seed 42, ~12 minutes on one A100-80GB.
|
| 27 |
+
The recipe was selected by a controlled sweep: 4x more optimizer steps at
|
| 28 |
+
higher LR beat both a LLaMA-style architecture swap and a 97M-parameter
|
| 29 |
+
capacity increase at this word budget.
|
| 30 |
+
- **Training data:** official BabyLM 2026 Strict-Small corpus (10,000,000 words), unmodified. Word counts use whitespace tokenization
|
| 31 |
+
matching the official BabyLM 2026 dataset cards.
|
| 32 |
+
|
| 33 |
+
## Intermediate checkpoints
|
| 34 |
+
|
| 35 |
+
As required by BabyLM 2026, checkpoints at every 1M words seen (to 10M) and
|
| 36 |
+
every 10M (to 100M) are available as revisions `chck_1M` ... `chck_100M`:
|
| 37 |
+
|
| 38 |
+
```python
|
| 39 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
+
model = AutoModelForCausalLM.from_pretrained("qyxu1994/instanton-baseline", revision="chck_5M")
|
| 41 |
+
tokenizer = AutoTokenizer.from_pretrained("qyxu1994/instanton-baseline")
|
| 42 |
+
```
|
| 43 |
+
|
| 44 |
+
## Evaluation
|
| 45 |
+
|
| 46 |
+
Official [babylm-eval](https://github.com/babylm-org/babylm-eval) pipeline,
|
| 47 |
+
full zero-shot suite (entity tracking follows the leaderboard's 2026-07-07
|
| 48 |
+
"nothing"-answer-filtered protocol): BLiMP 65.93, BLiMP-supplement 54.33, EWoK 51.91, COMPS 52.00, (Super)GLUE 63.33.
|
| 49 |
+
|
| 50 |
+
## Lineage
|
| 51 |
+
|
| 52 |
+
v2 of our submission family; v1 (same mixtures, shorter schedule) remains at
|
| 53 |
+
`qyxu1994/babylm-2026-strict-small-*` for reproducibility.
|