跳到正文
Hugging Face Daily Papers·· 2 天前精选AI 评分58

近零 token 嵌入会静默拖停门控序列模型训练

Near-Zero Token Embeddings Silently Stall Training in Gated Sequence Models

AI 导读

论文发现,在未绑定权重的门控语言模型中,近零输入嵌入会导致梯度极端放大和训练不稳定。作者调查 291 个开放权重模型仓库(280 张不同嵌入表),52.6% 的未绑定门控与混合 checkpoint(20/38)存在范数低于 10^{-6} 的可达输入嵌入,而权重绑定模型为 0%。

推荐理由

论文指出未绑定门控模型中近零输入嵌入会引发梯度爆炸,并给出两种修复方案,对训练稳定性排查有参考价值。

来源:Hugging Face Daily Papers · huggingface.co