Triangle104 commited on
Commit
21920db
·
verified ·
1 Parent(s): 4166237

Update README.md

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