Add zero-shot benchmark results (ARC, HellaSwag, SciQ, PIQA) and full-split val ppl
#1
by Compactbot - opened
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
|
|
|
|
|
|
|
|
|
|
| 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) |
|