计算机科学 > 机器学习
[提交于 2025年6月26日
]
标题: 注意小权重
标题: Pay Attention to Small Weights
摘要: 微调大型预训练神经网络被证明在内存和计算成本方面都是资源密集型的。 为了缓解这个问题,一种常见的方法是将训练限制在模型参数的一个子集上。 通过分析微调过程中梯度与权重之间的关系,我们观察到一个显著的模式:大的梯度通常与小幅度的权重相关联。 这种相关性在微调设置中比从头开始训练更为明显。 受这一观察的启发,我们提出了NANOADAM,它在微调过程中仅动态更新小幅度的权重,并提供了几个实际优势:首先,这种标准是无梯度的——参数子集可以在不进行梯度计算的情况下确定;其次,它保留了大尺度的权重,这些权重可能编码了预训练期间学到的关键特征,从而减少了灾难性遗忘的风险;第三,它允许使用更大的学习率,并在实验中始终导致更好的泛化性能。 我们在自然语言处理和视觉任务中都验证了这一点。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.