Add zero-shot benchmark results (ARC, HellaSwag, SciQ, PIQA) and full-split val ppl

#1
Files changed (1) hide show
  1. README.md +20 -2
README.md CHANGED
@@ -16,6 +16,7 @@ datasets:
16
  - roneneldan/TinyStories
17
  metrics:
18
  - perplexity
 
19
  ---
20
 
21
  # TinyStories-24m
@@ -37,13 +38,30 @@ narrative flow.
37
 
38
  ## Quality
39
 
40
- - **Val perplexity:** 8.76 (2.1618 nats/token on the 2M held-out val set).
 
 
 
41
  - **Generation:** coherent. Sampled 9/9 seeded generations (3 seeds × 3 prompts)
42
  produce proper dialogue, character names (Ben, Lily, Mom, Tom, Sarah, Max),
43
  punctuation and narrative flow. This model is a story generator for its
44
  training domain — it is **not** a general-purpose assistant and will not
45
  answer questions it was not trained on.
46
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
47
  ## Honest caveats
48
 
49
  - **Divergence:** the full 13,600-step run diverged to NaN at step 9,350 (LR 6e-4
@@ -85,4 +103,4 @@ print(tok.decode(ids, skip_special_tokens=True))
85
  | `model.safetensors` | 98,349,056 | 75 tensors, float32 |
86
  | `config.json` | — | architecture + training metadata |
87
  | `modeling.py` | — | the `TinyStoriesGPT` class (load with `from_pretrained`) |
88
- | `tokenizer.json` | 560,804 | BPE-8k tokenizer (HF `tokenizers` format) |
 
16
  - roneneldan/TinyStories
17
  metrics:
18
  - perplexity
19
+ - accuracy
20
  ---
21
 
22
  # TinyStories-24m
 
38
 
39
  ## Quality
40
 
41
+ - **Val perplexity:** 8.76 (2.1618 nats/token, training-time fixed-window score).
42
+ Full-split repro (100 random 512-token windows over the 4.5M-token val split):
43
+ **12.39** (2.5172 nats/token). The gap is methodology (fixed window vs random
44
+ windows), not a card error.
45
  - **Generation:** coherent. Sampled 9/9 seeded generations (3 seeds × 3 prompts)
46
  produce proper dialogue, character names (Ben, Lily, Mom, Tom, Sarah, Max),
47
  punctuation and narrative flow. This model is a story generator for its
48
  training domain — it is **not** a general-purpose assistant and will not
49
  answer questions it was not trained on.
50
 
51
+ ### Benchmark results (zero-shot loglikelihood, 400 examples per task)
52
+
53
+ | Task | Accuracy | Chance | Notes |
54
+ |------|----------|--------|-------|
55
+ | ARC-Easy | 13.3% | 25% | below chance |
56
+ | ARC-Challenge | 12.5% | 25% | below chance |
57
+ | HellaSwag | 25.0% | 25% | at chance |
58
+ | SciQ | 25.0% | 25% | at chance |
59
+ | PIQA | 50.0% | 50% | at chance |
60
+
61
+ All results are at or below chance — expected for a 24M model trained exclusively
62
+ on simple children's stories. The model has learned the distribution of story
63
+ text but has no general reasoning, commonsense, or science knowledge.
64
+
65
  ## Honest caveats
66
 
67
  - **Divergence:** the full 13,600-step run diverged to NaN at step 9,350 (LR 6e-4
 
103
  | `model.safetensors` | 98,349,056 | 75 tensors, float32 |
104
  | `config.json` | — | architecture + training metadata |
105
  | `modeling.py` | — | the `TinyStoriesGPT` class (load with `from_pretrained`) |
106
+ | `tokenizer.json` | 560,804 | BPE-8k tokenizer (HF `tokenizers` format) |