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 |
|