Abstract:Fine-tuning pretrained models has become a standard approach to adapting pretrained knowledge to improve the accuracy on new sparse, imbalance datasets. However, issues arise when optimization falls into a collapsed state, where the model gets stuck, leading to degraded performance and unstable training. One possible reason for this is the cancellation of gradients across training examples. To address this problem, we propose a novel algorithm, dynamic scaled gradient descent (\mName), that directly modifies the gradients returned by training examples, specifically, scaling down the gradients of correctly classified examples using a dynamic scaler. This strategy offers both theoretical and empirical advantages in improving training stability. Experiments on a variety of benchmark datasets, spanning multiple tasks and large pretrained models, demonstrate that our method consistently reduces performance variance and surpasses the accuracy of existing approaches.
| Subjects: | Machine Learning (cs.LG) |
| Cite as: | arXiv:2604.27987 [cs.LG] |
| (or arXiv:2604.27987v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2604.27987 arXiv-issued DOI via DataCite (pending registration) |
From: Lijing Wang [view email]
[v1]
Thu, 30 Apr 2026 15:16:00 UTC (203 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。