跳到正文
arXiv · Machine Learning· Pierre-Olivier Bonin·· 10 小时前AI 评分46

误差扩散取代潜在权重:用少量位的抖动状态训练二进制权重语言模型

Error Diffusion Replaces Latent Weights: Training Binary-Weight Language Models on a Few Bits of Dithered State

AI 导读

研究提出误差扩散(ED)优化器,以2–8位残差缓冲区替代二进制权重语言模型训练中的全精度潜在权重及Adam状态,将每权重训练状态从97位降至3–9位,最多减少19倍。在97%参数为二进制的字符级Transformer上,ED收敛时与参考差1.7%;融合CPU内核可在275 MB训练内存预算内继续适配已部署模型。

来源:arXiv · Machine Learning · arxiv.org