qyxu1994 commited on
Commit
e41bd73
·
verified ·
1 Parent(s): 5ec8436

instanton model card

Browse files
Files changed (1) hide show
  1. README.md +53 -0
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.