跳到正文
arXiv · cs.AI · Zaiqian Chen, Aditya Somasundaram, Sophie Li, Sean McLeish, John Kirchenbauer, Tom Goldstein, Micah Goldblum · 10 小时前模型评分45

量化误差如何引发大语言模型预训练中的“定时炸弹”式损失尖峰

The Role of Quantization in Loss Spikes

导读

研究发现,大语言模型预训练中延迟出现的“定时炸弹”式损失尖峰,由语言建模头输出 logits 增长引发的量化误差快速上升所致。降低数值精度会加速这类尖峰,提高精度则改善稳定性;将语言建模头的 AdamW 换成带动量的 SGD 可抑制 logits 增长。现有缓解方法与作者提出的方法都能降低量化误差,指向共同的数值机制。

生成六维分析
分类置信度

中 · 待人工复核

  • 只有单一媒体来源,需要人工复核。

来源级别:媒体与个人

事件类型:研究发布

证据摘录

  • “We find that these instabilities are preceded by rapidly growing quantization error in the LM head, which eventually culminates in a loss spike.”

来源:arXiv · cs.AI · arxiv.org