francescortu commited on
Commit
6afc329
·
verified ·
1 Parent(s): fb37717

Finalize model card with eval results

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