阅读研究雷达投资体系
登录 / 注册
← 返回阅读
资讯专题 · 分层舍入策略1 分钟阅读·2026-10-09DisconfirmAI 编辑部

低精度推理需分层舍入策略

研究发现,在6位精度下,对MLP层使用随机舍入、对输出头使用就近舍入,可将DistilGPT-2的困惑度劣化从2.21倍降至1.10倍。

重要性局部证据E2 未复现写法快讯

原文发布时间 2026-10-02 12:00

在6位虚拟精度下,混合舍入策略将DistilGPT-2的困惑度劣化控制在1.10倍以内。

此前低精度推理常统一采用随机舍入(SR)或就近舍入(RN)。该研究指出,不同网络层对舍入误差的敏感度相反:MLP层的噪声主要导致均匀的对数概率偏移,Softmax对其不敏感,因此SR的方差惩罚较小;而语言模型头的噪声非均匀分布,RN更优。

实验显示,纯RN方案使困惑度升至全精度的2.21倍,纯SR为1.15倍。若对MLP应用SR、对Head应用RN,困惑度仅升至1.10倍,比纯RN降低28%。

结果基于DistilGPT-2小模型的仿真测试,尚未在大规模模型或真实硬件上验证。

来源

  • 原始出处arxiv.org↗
专题

分层舍入策略

【背景】在Transformer模型的低精度推理中,单一舍入规则往往导致性能显著下降。最新研究提出“分层舍入策略”,即根据网络不同部位的特性混合使用随机舍入(SR)和就近舍入(RN)。

2 节点1 篇更新于 2026-10-09 15:20

事情怎么走到这一步

  1. 2026-10-01Chatelain等人提交预印本《Stochastic Rounding in Low-Precision Transformer Inference》,提出变量精度随机舍入(VPSR)算法及分层舍入概念。
  2. 2026-10-08
登录 / 注册
阅读研究雷达投资体系
论文v2版本发布,确认在DistilGPT-2 6位精度下,MLP用SR+输出头用RN的组合可将困惑度劣化从2.21x降至1.10x。

还没定论的部分

  • 该策略在更大规模模型(如7B以上参数)上的泛化能力如何?
  • 实际硬件部署中,动态切换舍入模式的延迟开销是否抵消了精度收益?
  • LAMP等基于重计算的混合精度方案与分层舍入策略能否结合使用?
读完整专题 →
带入研究 →