提出可调适应性的新优化器,兼顾SGD与Adam优点
HVAdam: A Full-Dimension Adaptive Optimizer
- 设计连续可调适应性的优化器,灵活切换SGD与Adam行为
- 在图像分类、扩散模型和语言建模任务上均超越现有最优方法
- 理论保证收敛性,适合需要稳定训练的复杂模型场景
自适应优化器如Adam在大规模模型训练中表现优异,但在经典架构(如CNN)上泛化能力常弱于非自适应方法(如SGD)。我们发现性能差距的关键在于预条件器中的自适应性限制了对多样化优化景观的适应能力。为此,提出Anon(Adaptivity Non-restricted Optimizer with Novel convergence technique),一种具有连续可调自适应性的新型优化器,可在SGD-like与Adam-like行为间插值,甚至超越二者。为确保全自适应范围内的收敛性,引入增量延迟更新(IDU)机制,比AMSGrad的硬最大跟踪策略更灵活,且对梯度噪声更具鲁棒性。理论上建立了凸与非凸设置下的收敛保证。实验表明,Anon在代表性图像分类、扩散模型与语言建模任务中持续优于当前最先进优化器。结果证明自适应性可作为可调节的设计原则,Anon首次提供了统一可靠的框架,弥合传统与现代优化器的差距并超越其优势。
原文摘要 · Abstract (English)
Adaptive optimizers such as Adam have achieved great success in training large-scale models like large language models and diffusion models. However, they often generalize worse than non-adaptive methods, such as SGD on classical architectures like CNNs. We identify a key cause of this performance gap: adaptivity in pre-conditioners, which limits the optimizer's ability to adapt to diverse optimization landscapes. To address this, we propose Anon (Adaptivity Non-restricted Optimizer with Novel convergence technique), a novel optimizer with continuously tunable adaptivity , allowing it to interpolate between SGD-like and Adam-like behaviors and even extrapolate beyond both. To ensure convergence across the entire adaptivity spectrum, we introduce incremental delay update (IDU), a novel mechanism that is more flexible than AMSGrad's hard max-tracking strategy and enhances robustness to gradient noise. We theoretically establish convergence guarantees under both convex and non-convex settings. Empirically, Anon consistently outperforms state-of-the-art optimizers on representative image classification, diffusion, and language modeling tasks. These results demonstrate that adaptivity can serve as a valuable tunable design principle, and Anon provides the first unified and reliable framework capable of bridging the gap between classical and modern optimizers and surpassing their advantageous properties.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。