WillHeld commited on
Commit
23c3dee
·
verified ·
1 Parent(s): 4d9b1c1

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +15 -3
README.md CHANGED
@@ -9,6 +9,20 @@ GRPO RL fine-tune of the Marin **Delphi-25B** base model (`marin-community/delph
9
 
10
  This is the **step-160** checkpoint (20 epochs). Training reward on the hard-math distribution rose from ~4% to ~29%.
11
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  ## Recipe
13
  - GRPO, rule-based boxed-answer reward (`math_verify`), `kl_loss_coef=1e-3`, `rollout.n=8`, lr 5e-7, top_p 1.0.
14
  - **Rollout temperature 0.5** (not the recipe default of 1.0 — a sweep showed the reward signal collapses above ~0.5 for this weak base on hard math).
@@ -44,8 +58,6 @@ Please reason step by step, and put your final answer within \boxed{}.<|im_end|>
44
  ```
45
 
46
  ## Example problems (illustrative of the reliable type; not guaranteed)
47
- Append each to the 3-shot prefix above. Sampling: greedy, stop=`<|im_end|>`.
48
-
49
  | Problem | Answer |
50
  |---|---|
51
  | What is the remainder when $2^{10}$ is divided by 7? | 2 |
@@ -55,4 +67,4 @@ Append each to the 3-shot prefix above. Sampling: greedy, stop=`<|im_end|>`.
55
  | A rectangle has length 8 and width 5. What is its area? | 40 |
56
 
57
  ## Capability notes (honest)
58
- RL gave a weak base **genuine but narrow** competence, and reasoning in the correct shape (steps -> single \boxed{} answer -> clean stop). It is reliable on problems solvable by **a single operation or one known formula with small/obvious inputs** (e.g. a Pythagorean step, a short modular cycle, a percentage). It is **unreliable** whenever it must (a) **track/combine several quantities** (multi-term word problems) or (b) **discover** something, such as a non-trivial factorization — in these cases it tends to confabulate a relation or factorization and not check it. ~29% on hard (level 3-5) MATH reflects this. It is a demonstration that the SimpleRL recipe works mechanically on a weak base; it is **not** a strong general math model.
 
9
 
10
  This is the **step-160** checkpoint (20 epochs). Training reward on the hard-math distribution rose from ~4% to ~29%.
11
 
12
+ ## Training dynamics & generalization
13
+
14
+ ![RL scaling](rl_scaling.png)
15
+
16
+ Held-out generalization (greedy, few-shot prompt) tracked against training step:
17
+
18
+ | step | MATH-500 (few-shot) | MATH-500 (plain) | AIME-24 |
19
+ |---|---|---|---|
20
+ | 0 (base) | 5.2% | 1.4% | 0% |
21
+ | 70 | **17.4%** (peak) | 0.6% | 0% |
22
+ | 160 (this ckpt) | 13.6% | 2.4% | 0% |
23
+
24
+ Read: RL produced **real held-out gains** (MATH-500 5.2% -> ~17%), but held-out **peaks around step 70 and then overfits** while train reward keeps climbing. The ability is **few-shot-format-locked** (plain-prompt held-out stays ~1-2.5%) and AIME-24 stays at **0%** throughout. The best-generalizing checkpoint is ~step 70, not the final step.
25
+
26
  ## Recipe
27
  - GRPO, rule-based boxed-answer reward (`math_verify`), `kl_loss_coef=1e-3`, `rollout.n=8`, lr 5e-7, top_p 1.0.
28
  - **Rollout temperature 0.5** (not the recipe default of 1.0 — a sweep showed the reward signal collapses above ~0.5 for this weak base on hard math).
 
58
  ```
59
 
60
  ## Example problems (illustrative of the reliable type; not guaranteed)
 
 
61
  | Problem | Answer |
62
  |---|---|
63
  | What is the remainder when $2^{10}$ is divided by 7? | 2 |
 
67
  | A rectangle has length 8 and width 5. What is its area? | 40 |
68
 
69
  ## Capability notes (honest)
70
+ RL gave a weak base **genuine but narrow** competence, and reasoning in the correct shape (steps -> single \boxed{} answer -> clean stop). It is reliable on problems solvable by **a single operation or one known formula with small/obvious inputs**. It is **unreliable** whenever it must (a) **track/combine several quantities** or (b) **discover** something like a non-trivial factorization. It is a demonstration that the SimpleRL recipe works mechanically on a weak base; it is **not** a strong general math model.