Finalize model card with eval results
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ reproduction and is **not** affiliated with the authors.
|
|
| 21 |
|
| 22 |
- **Base (student) model:** [Qwen/Qwen2.5-3B](https://huggingface.co/Qwen/Qwen2.5-3B)
|
| 23 |
- **Teacher:** openai/gpt-oss-120b
|
| 24 |
-
- **Training data:** s1 (1K prompts) — 1000 teacher-generated responses, shipped verbatim in the authors' repo
|
| 25 |
- **Prompt template:** plain `Problem:\n{question}\n\nSolution:\n`
|
| 26 |
|
| 27 |
## Training
|
|
@@ -29,11 +29,33 @@ reproduction and is **not** affiliated with the authors.
|
|
| 29 |
SFT with the authors' released `training/` scripts (paper Appendix A recipe):
|
| 30 |
3 epochs, LR 1e-5, cosine schedule, 5% warmup, per-device batch 4 x grad-accum 4
|
| 31 |
(effective batch 16), block size 4096, bf16, gradient checkpointing, loss on
|
| 32 |
-
response tokens only (prompt masked -100).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
|
| 34 |
-
## Evaluation
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
-
⏳ **GSM8K / MATH500 results are still being computed and will be added here.**
|
| 37 |
|
| 38 |
## License
|
| 39 |
|
|
|
|
| 21 |
|
| 22 |
- **Base (student) model:** [Qwen/Qwen2.5-3B](https://huggingface.co/Qwen/Qwen2.5-3B)
|
| 23 |
- **Teacher:** openai/gpt-oss-120b
|
| 24 |
+
- **Training data:** s1 (1K prompts) — 1000 teacher-generated responses, shipped verbatim in the authors' repo (`data/training/Teacher=GPT-OSS-120B_Data=S1_Template=Chat.jsonl`)
|
| 25 |
- **Prompt template:** plain `Problem:\n{question}\n\nSolution:\n`
|
| 26 |
|
| 27 |
## Training
|
|
|
|
| 29 |
SFT with the authors' released `training/` scripts (paper Appendix A recipe):
|
| 30 |
3 epochs, LR 1e-5, cosine schedule, 5% warmup, per-device batch 4 x grad-accum 4
|
| 31 |
(effective batch 16), block size 4096, bf16, gradient checkpointing, loss on
|
| 32 |
+
response tokens only (prompt masked -100). Teacher responses were pre-truncated
|
| 33 |
+
to 2,048 tokens in the released data. Trained on 1x H100
|
| 34 |
+
(paper used 2x H200; hyperparameters identical), transformers 4.55.4 / trl 0.19.1,
|
| 35 |
+
seed 42 (HF default; paper seed unknown). One compatibility patch: trl renamed
|
| 36 |
+
`max_seq_length` to `max_length`, so the block size is passed explicitly
|
| 37 |
+
(no behavioral change).
|
| 38 |
|
| 39 |
+
## Evaluation (ours vs. paper Table 9)
|
| 40 |
+
|
| 41 |
+
Greedy decoding, template matched to training, scored with `math_verify`.
|
| 42 |
+
GSM8K 4-shot; MATH500 zero-shot. Few-shot counts were calibrated so the
|
| 43 |
+
*base* models reproduce their Table 9 baselines. The paper does not document its
|
| 44 |
+
eval protocol, so treat cross-paper comparisons as approximate.
|
| 45 |
+
|
| 46 |
+
| Benchmark | This reproduction | Paper Table 9 | Gen. budget | Hit budget cap |
|
| 47 |
+
|---|---|---|---|---|
|
| 48 |
+
| GSM8K | 82.49 | 79.90 | 4096 tok | 1.0% |
|
| 49 |
+
| MATH500 | 44.00 | 53.40 | 16384 tok | 49.8% |
|
| 50 |
+
|
| 51 |
+
A high "hit budget cap" fraction means the model was still generating when the
|
| 52 |
+
token budget ran out, so that accuracy is a **lower bound** — these students are
|
| 53 |
+
SFT'd on teacher traces that were themselves truncated at 2,048 tokens, which
|
| 54 |
+
makes some of them generate very long self-checking traces.
|
| 55 |
+
|
| 56 |
+
Base-model reference (our protocol / paper): GSM8K 67.63 / 75.82,
|
| 57 |
+
MATH500 40.20 / 39.80.
|
| 58 |
|
|
|
|
| 59 |
|
| 60 |
## License
|
| 61 |
|