File size: 4,188 Bytes
97de591
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# openPangu-R-72B-2512-Int8
中文 | [English](README_EN.md)

## 1. 简介
openPangu-R-72B-2512 是基于昇腾集群训练的MoE模型。模型总参数量74B,激活参数量15B,采用80选8的专家选择机制,支持128k长序列处理。训练数据总量约24T tokens。同一个模型支持快慢思考两种模式切换,慢思考模式下,支持思维链分档(“低”和“高”两种推理深度)。

openPangu-R-72B-2512-Int8 是 [openPangu-R-72B-2512](https://gitcode.com/ascend-tribe/openPangu-R-72B-2512) 的量化版本,使用动态 per-token 量化方法,能够减少约一半的显存占用,提升超20%吞吐,综合精度损失小于1%。

## 2. 模型架构
openPangu-R-72B-2512-Int8 在模型稳定收敛和效果提升方向进行了以下优化:
- 在注意力机制中引入参数式Sink Token技术: 有效缓解极大激活值问题,训练中最大激活值从$10^3$降至$10^2$量级, 提升训练稳定性并对后量化亲和。

- K-Norm与Depth-Scaled Sandwich-Norm:为保证attention logits的稳定性,我们采用了K-Norm结构。K-Norm与QK-Norm类似,但只对attention的key施加RMS Norm。K-Norm可起到与QK-Norm类似的稳定性效果,但引入的计算开销更小,同时K-Norm不影响Query的scale,带来更灵活的表达能力。为了保证残差连接的稳定性,我们采用了Depth-Scaled Sandwich-Norm方法。

- 注意力架构优化:增加Query头数和注意力头维度,使模型能够从更多角度捕获细粒度语义关系。引入Partial RoPE机制,仅对Query和Key中1/3维度应用位置编码。尽管Key头维度有所增加,但通过将KV组数量减半,KV cache仍可减少37.5%,在保持推理阶段显存和速度优化的同时,实现了更低的训练损失和更优的推理性能。

- Adaptive Aux Free负载优化技术:能够自适应调整专家bias更新幅度,减少均衡震荡现象,优化专家负载分布均衡性。


详细架构参数如下:

| | |
|:---:|:---:|
| **Architecture** | Mixture-of-Experts (MoE) |
| **Total Parameters** | 74B |
| **Activated Parameters** | 15B |
| **Number of Layers** (Dense layer included) | 50 |
| **Number of Dense Layers** | 4 |
| **Number of MTP Modules** | 1 |
| **Hidden Dimension** | 4608 |
| **MoE Hidden Dimension** (per Expert) | 1280 |
| **Attention Mechanism** | GQA |
| **Number of Attention Heads** | 64 |
| **Number of Query Groups** | 4 |
| **Number of Experts** | 80 |
| **Selected Experts per Token** | 8 |
| **Number of Shared Experts** | 2 |
| **Vocabulary Size** | 153K |
| **Context Length** | 128K |


## 3. 部署和使用
- 使用omni-infer推理框架,部署方式与非量化版本一致,参考[[omniinfer_for_openpangu_r_72b_2512_Int8](doc/omniinfer_for_openpangu_r_72b_2512.md)]

## 4. 模型许可证
除文件中对开源许可证另有约定外,openPangu-R-72B-2512-Int8 模型根据 OPENPANGU MODEL LICENSE AGREEMENT VERSION 1.0 授权,旨在允许使用并促进人工智能技术的进一步发展。有关详细信息,请参阅模型存储库根目录中的 [LICENSE](LICENSE) 文件。

## 5. 免责声明
由于 openPangu-R-72B-2512-Int8 (“模型”)所依赖的技术固有的限制,以及人工智能生成的内容是由盘古自动生成的,华为无法对以下事项做出任何保证:
- 该模型的输出通过AI算法自动生成,不能排除某些信息可能存在缺陷、不合理或引起不适的可能性,生成的内容不代表华为的态度或立场; 
- 无法保证该模型100%准确、可靠、功能齐全、及时、安全、无错误、不间断、持续稳定或无任何故障; 
- 该模型的输出内容不构成任何建议或决策,也不保证生成的内容的真实性、完整性、准确性、及时性、合法性、功能性或实用性。生成的内容不能替代医疗、法律等领域的专业人士回答您的问题。生成的内容仅供参考,不代表华为的任何态度、立场或观点。您需要根据实际情况做出独立判断,华为不承担任何责任。

## 7. 反馈
如果有任何意见和建议,请提交issue或联系[openPangu@huawei.com](url)。