开发者
资源
[object Object]
[object Object]
[object Object]

离群值抑制(Anti-Outlier)主要用于解决模型量化中因激活值分布存在异常值(Outlier)而导致的精度损失问题。在大模型量化过程中,如果激活值中存在数值极大的离群点,会拉大量化区间(Scale),导致大部分正常数值的量化分辨率降低,从而严重影响模型精度。该技术通过对离群值进行平滑或抑制处理,有效改善数据分布,确保量化后的模型仍能保持较高的推理精度。

[object Object]

以下展示了W8A8 + Anti-Outlier + PDMIX量化后权重描述文件 [object Object] 中的部分内容:

[object Object]

当前主流开源大模型(如LLaMA、Qwen等)通常使用RmsNorm作为[object Object][object Object]。当启用非对称离群值抑制算法时,会引入额外的偏置项(以下称为[object Object])。

为了保证计算等价性,该算法的执行逻辑如下:

  1. Norm层引入:在执行Norm操作时,将[object Object]加到权重中。
  2. Linear层抵消:在随后的Linear层计算前,将对应的[object Object]减去。

在实际的模型权重中,[object Object]在Linear层的抵消张量,会根据量化场景的不同,以不同方式融合:

  • Per-tensor场景:直接融合进Linear层的量化偏置[object Object]中。
    • 例如上文示例中的[object Object]
  • Per-token场景:体现为Linear层的普通偏置[object Object]
    • 例如上文示例中的[object Object]
[object Object]

性能优化建议: 在PDMIX场景(P阶段使用Per-token量化)下,离群值抑制的bias对量化精度的影响通常较小。为提升性能,可考虑在保证等价性的前提下,在Norm层和Linear层同时移除Bias,从而节省一次Add计算开销。

图 1 量化权重推理时流程

表 1 权重量化后部分层的dtype及shape信息(假设原始权重的shape为 [object Object]

[object Object]undefined
[object Object]

您可以使用msModelSlim工具生成量化权重:

以Qwen3-14B为例,安装msModelSlim工具后,可以使用如下命令快速生成一份带有离群值抑制的W8A8PDMIX量化权重:

[object Object]

执行上述命令会默认使用msModelSlim工具的最佳实践方式执行量化,如需了解更多量化参数配置,请参考msModelSlim工具文档。

[object Object]

以Qwen3-14B-W8A8PDMIX权重为例,您可以使用以下指令执行对话测试,推理内容为"What's deep learning?",最长输出20个token。

[object Object]