只微调小权重参数,节省资源且效果更好。
Pay Attention to Small Weights
- 根据权重大小动态选择微调参数,无需计算梯度。
- 保留大权重避免灾难性遗忘,支持更大学习率。
- 适用于NLP和视觉任务,提升泛化性能。
微调大型预训练神经网络资源消耗大。通过分析微调过程中梯度与权重的关系,发现大梯度常对应小权重,这一现象在微调中比从头训练更显著。基于此,提出NANOADAM方法:仅动态更新小幅度权重。该方法无需梯度计算,能保留大权重以维持预训练关键特征,降低灾难性遗忘风险;同时允许使用更大学习率,在实验中一致取得更好泛化性能。在NLP和视觉任务上均验证了有效性。
原文摘要 · Abstract (English)
Finetuning large pretrained neural networks is known to be resource-intensive, both in terms of memory and computational cost. To mitigate this, a common approach is to restrict training to a subset of the model parameters. By analyzing the relationship between gradients and weights during finetuning, we observe a notable pattern: large gradients are often associated with small-magnitude weights. This correlation is more pronounced in finetuning settings than in training from scratch. Motivated by this observation, we propose NANOADAM, which dynamically updates only the small-magnitude weights during finetuning and offers several practical advantages: first, this criterion is gradient-free -- the parameter subset can be determined without gradient computation; second, it preserves large-magnitude weights, which are likely to encode critical features learned during pretraining, thereby reducing the risk of catastrophic forgetting; thirdly, it permits the use of larger learning rates and consistently leads to better generalization performance in experiments. We demonstrate this for both NLP and vision tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。