Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,86 @@ tags:
|
|
| 18 |
This model was converted to GGUF format from [`agentica-org/DeepScaleR-1.5B-Preview`](https://huggingface.co/agentica-org/DeepScaleR-1.5B-Preview) using llama.cpp via the ggml.ai's [GGUF-my-repo](https://huggingface.co/spaces/ggml-org/gguf-my-repo) space.
|
| 19 |
Refer to the [original model card](https://huggingface.co/agentica-org/DeepScaleR-1.5B-Preview) for more details on the model.
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
## Use with llama.cpp
|
| 22 |
Install llama.cpp through brew (works on Mac and Linux)
|
| 23 |
|
|
|
|
| 18 |
This model was converted to GGUF format from [`agentica-org/DeepScaleR-1.5B-Preview`](https://huggingface.co/agentica-org/DeepScaleR-1.5B-Preview) using llama.cpp via the ggml.ai's [GGUF-my-repo](https://huggingface.co/spaces/ggml-org/gguf-my-repo) space.
|
| 19 |
Refer to the [original model card](https://huggingface.co/agentica-org/DeepScaleR-1.5B-Preview) for more details on the model.
|
| 20 |
|
| 21 |
+
---
|
| 22 |
+
DeepScaleR-1.5B-Preview is a language model fine-tuned from
|
| 23 |
+
DeepSeek-R1-Distilled-Qwen-1.5B using distributed reinforcement learning
|
| 24 |
+
(RL) to scale up to long context lengths. The model achieves 43.1%
|
| 25 |
+
Pass@1 accuracy on AIME 2024, representing a 15% improvement over the
|
| 26 |
+
base model (28.8%) and surpassing OpenAI's O1-Preview performance with
|
| 27 |
+
just 1.5B parameters.
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
|
| 35 |
+
Data
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
Our training dataset consists of approximately 40,000 unique problem-answer pairs compiled from:
|
| 41 |
+
|
| 42 |
+
|
| 43 |
+
AIME problems (1984-2023)
|
| 44 |
+
AMC problems (prior to 2023)
|
| 45 |
+
Omni-MATH dataset
|
| 46 |
+
Still dataset
|
| 47 |
+
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
|
| 54 |
+
Training Recipe
|
| 55 |
+
|
| 56 |
+
|
| 57 |
+
|
| 58 |
+
|
| 59 |
+
We employ Deepseek's Group Relative Policy Optimization (GRPO), a simplified RL algorithm that extends PPO by:
|
| 60 |
+
|
| 61 |
+
|
| 62 |
+
Normalizing advantage function over all samples generated from the same prompt.
|
| 63 |
+
Applying KL divergence regularization on top of PPO's surrogate loss to prevent significant policy drift.
|
| 64 |
+
|
| 65 |
+
|
| 66 |
+
Reward Function: Our reward function is simple but effective:
|
| 67 |
+
|
| 68 |
+
|
| 69 |
+
1 for correct answers passing LaTeX/Sympy checks
|
| 70 |
+
0 for incorrect or improperly formatted answers
|
| 71 |
+
Note: No partial rewards (such as PRMs) or intermediate feedback.
|
| 72 |
+
|
| 73 |
+
|
| 74 |
+
Iterative Context Lengthening: A key challenge in
|
| 75 |
+
scaling RL for reasoning is compute cost. Our approach trains models
|
| 76 |
+
with progressively longer contexts as the model improves, thus saving
|
| 77 |
+
monetary costs and end2end training time:
|
| 78 |
+
|
| 79 |
+
|
| 80 |
+
Initial 8K Context (0-1040 steps):
|
| 81 |
+
22.9% -> 33% Pass@1 on AIME 2024
|
| 82 |
+
Trained on 8 A100-80GB GPUs, BS= (Prompts) * (Samples/Prompt) = 128 * 8 = 1024
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
Extended to 16K (steps 1040-1520):
|
| 86 |
+
33% -> 43% Pass@1 on AIME 2024
|
| 87 |
+
Trained on 32 A100-80GB GPUs, BS= (Prompts) * (Samples/Prompt) = 128 * 16 = 2048
|
| 88 |
+
|
| 89 |
+
|
| 90 |
+
Further extended to 24K (step 1520+):
|
| 91 |
+
38% -> 43% Pass@1 on AIME 2024
|
| 92 |
+
Trained on 32 A100-80GB GPUs, BS= (Prompts) * (Samples/Prompt) = 128 * 16 = 2048
|
| 93 |
+
Significant improvements within <200 steps
|
| 94 |
+
|
| 95 |
+
|
| 96 |
+
|
| 97 |
+
|
| 98 |
+
A more detailed description of the training recipe can be found in our blog post.
|
| 99 |
+
|
| 100 |
+
---
|
| 101 |
## Use with llama.cpp
|
| 102 |
Install llama.cpp through brew (works on Mac and Linux)
|
| 103 |
|