Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,43 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- zh
|
| 5 |
+
- en
|
| 6 |
+
datasets:
|
| 7 |
+
- inclusionAI/Ling-Coder-SFT
|
| 8 |
+
- amd/Instella-GSM8K-synthetic
|
| 9 |
+
- tuanha1305/DeepSeek-R1-Distill
|
| 10 |
+
- YeungNLP/school_math_0.25M
|
| 11 |
+
- jingyaogong/minimind_dataset
|
| 12 |
+
- Roman1111111/claude-sonnet-4.6-120000x
|
| 13 |
+
- SnifferCaptain/z1-1m
|
| 14 |
+
pipeline_tag: text-generation
|
| 15 |
+
---
|
| 16 |
+
## 模型描述
|
| 17 |
+
YModel3是SnifferCaptain训练的到目前为止(5/4/2026)最新的大语言模型。模型相比YModel2,支持了如混合思考、可调思考深度的功能,在回答的质量上有一定的进步。
|
| 18 |
+
## 模型细节
|
| 19 |
+
- 模型借鉴了MLA( Deepseek, https://arxiv.org/pdf/2405.04434 )的优化思路,与Gated Attention( Qwen, https://arxiv.org/pdf/2505.06708 )的优化思路,将ymodel2的PEGA2替换为容量更大,scaling更强的MLGA模块。显著增加了模型的参数效率。
|
| 20 |
+
- 模型在FFN部分采用了SwiGLU。
|
| 21 |
+
- 模型在每层RMSNorm后,增加了一层SEBlock。
|
| 22 |
+
## 训练细节
|
| 23 |
+
- 模型全程采用与YModel2相近的SiMuon优化器训练,其中将NS迭代从2步增加到3步。SiMuon部分参数在预训练中,学习率为默认学习率的10倍。在其余阶段,学习率为默认学习率的66倍。AdamW部分学习率为默认参数的1倍。使用0.2*sqrt(max(fan in, fan out))进行学习率缩放。
|
| 24 |
+
- 模型的tokenlizer与词嵌入层使用的是预训练权重,来自MiniMind3-v( https://github.com/jingyaogong/minimind )
|
| 25 |
+
- 在预训练阶段,模型充分使用了5B token,在1e-4带warmup的余弦退火到1e-5的学习率下,完成预训练。
|
| 26 |
+
## 补充细节
|
| 27 |
+
- 由于预训练数据集原因,预训练模型在英文输入后有很大概率生成与各大跑分测试类似的四选一选择题格式
|
| 28 |
+
- 关于调节思考深度,需要遵循特定的思考模板实现:
|
| 29 |
+
```text
|
| 30 |
+
<|im_start|>assistant
|
| 31 |
+
<think>juice = 1.14
|
| 32 |
+
[thinking content]</think>
|
| 33 |
+
[reply content]<|im_end|>
|
| 34 |
+
```
|
| 35 |
+
其中,juice值最好使用两位小数表示,其值过小或者过大可能都会出现一些问题。juice值的大小可以参考以下公式计算:
|
| 36 |
+
$$\max\left(0.0,\ \log_2\left(\frac{\text{token\_count}}{128} + 1\right)\right)$$
|
| 37 |
+
典型值:
|
| 38 |
+
|juice|token count|
|
| 39 |
+
|---|---|
|
| 40 |
+
|0.59|64|
|
| 41 |
+
|1.00|128|
|
| 42 |
+
|2.00|384|
|
| 43 |
+
|3.00|896|
|