Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# grpo-llama3.1-8b-0623-final
|
| 2 |
+
|
| 3 |
+
**Base model:** `meta-llama/Llama-3.1-8B-Instruct`
|
| 4 |
+
**Method:** GRPO (Group Relative Policy Optimization, TRL) with dual-RM reward signal.
|
| 5 |
+
**Final checkpoint** = `load_best_model_at_end=True`로 골라진 best step 5,185 가중치.
|
| 6 |
+
|
| 7 |
+
## Reward models used during training (frozen)
|
| 8 |
+
|
| 9 |
+
| Slot | Model |
|
| 10 |
+
|---|---|
|
| 11 |
+
| RM 1 | [`WooYoungSeok/rm-qwen2.5-math-7b-0622`](https://huggingface.co/WooYoungSeok/rm-qwen2.5-math-7b-0622) — verifier-set acc 0.8354 |
|
| 12 |
+
| RM 2 | [`WooYoungSeok/rm-deepseek-r1-qwen3-8b-0622`](https://huggingface.co/WooYoungSeok/rm-deepseek-r1-qwen3-8b-0622) — verifier-set acc 0.8497 |
|
| 13 |
+
|
| 14 |
+
## Training
|
| 15 |
+
|
| 16 |
+
| Hyperparameter | Value |
|
| 17 |
+
|---|---|
|
| 18 |
+
| Algorithm | GRPO (TRL) |
|
| 19 |
+
| Step at best ckpt (saved as final) | **5,185** |
|
| 20 |
+
| Epoch at best ckpt | 1.6989 / 2 |
|
| 21 |
+
| Total training step | 6,100 (≈ 2 epochs completed) |
|
| 22 |
+
| Group size (G) | 8 |
|
| 23 |
+
| Per-device batch | 8 |
|
| 24 |
+
| Gradient accumulation | 2 |
|
| 25 |
+
| Learning rate | 1e-6 |
|
| 26 |
+
| Warmup ratio | 0.1 |
|
| 27 |
+
| KL coefficient (β) | 0.04 |
|
| 28 |
+
| Clip ε | 0.2 |
|
| 29 |
+
| λ (verifiable penalty) | 0.75 |
|
| 30 |
+
| Format penalty | 0.5 |
|
| 31 |
+
| Answer-tag c | 0.5 |
|
| 32 |
+
| Max completion length | 1024 |
|
| 33 |
+
| Optimizer | DeepSpeed ZeRO-2 + CPU offload |
|
| 34 |
+
| Generation backend | vLLM (server mode, port 8001) |
|
| 35 |
+
| Seed | 42 |
|
| 36 |
+
|
| 37 |
+
## Data (in-distribution 90/10 split of `train_subset.json`)
|
| 38 |
+
|
| 39 |
+
| Split | Source | Rows |
|
| 40 |
+
|---|---|---|
|
| 41 |
+
| Train | `train_subset_90pct.json` (90% stratified by error_type, 5-cluster taxonomy from MathEDU + Stepwise Verification + MathClean + EIC) | 6,105 |
|
| 42 |
+
| Eval (during training) | `train_subset_10pct.json` (in-distribution 10% held-out, stratified, seed=42) | 678 |
|
| 43 |
+
|
| 44 |
+
## Performance
|
| 45 |
+
|
| 46 |
+
| Metric | Value | Step |
|
| 47 |
+
|---|---|---|
|
| 48 |
+
| **eval_reward (best, saved as final)** | **1.2560** | 5,185 |
|
| 49 |
+
| eval_loss (at best) | 0.0187 | 5,185 |
|
| 50 |
+
| eval_reward (last) | 1.2236 | 6,100 |
|
| 51 |
+
| eval_loss (last) | 0.0308 | 6,100 |
|
| 52 |
+
| train reward (last 100-step window) | ≈ 1.20 | — |
|
| 53 |
+
|
| 54 |
+
## License
|
| 55 |
+
|
| 56 |
+
Apache-2.0. Base model is governed by the Llama 3.1 Community License — verify compliance for your use case.
|