arXiv:2603.07122cs.LGstat.ML2026-03

提出DualAdam,融合Adam与逆Adam提升模型泛化能力。

Combining Adam and its Inverse Counterpart to Enhance Generalization of Deep Learning Optimizers

  • 设计逆Adam机制,通过乘法更新避开尖锐极小值。
  • 双优化器融合确保收敛性,图像分类与大模型微调中表现更优。
  • 理论分析支持其逃逸尖锐极小值的能力,适合追求泛化的研究者。

在神经网络训练中,自适应矩估计(Adam)通常收敛快但泛化性能较差,原因在于易陷入尖锐极小值。为此,本文提出逆Adam(InvAdam),其参数更新机制与Adam相反:将一阶与二阶动量做逐元素乘积,而非除法。该设计在二阶动量较大时增大步长,有助于逃离尖锐极小值并停留于平坦区域。然而InvAdam可能存在收敛困难。为此,本文进一步提出双Adam(DualAdam),融合Adam与InvAdam的更新机制,在保证收敛的同时显著提升泛化性能。通过扩散理论对InvAdam的逃逸能力进行数学证明。在图像分类和大语言模型(LLM)微调任务上的大量实验表明,DualAdam优于Adam及其主流变体。代码已开源:https://github.com/LongJin-lab/DualAdam。

原文摘要 · Abstract (English)

In the training of neural networks, adaptive moment estimation (Adam) typically converges fast but exhibits suboptimal generalization performance. A widely accepted explanation for its defect in generalization is that it often tends to converge to sharp minima. To enhance its ability to find flat minima, we propose its new variant named inverse Adam (InvAdam). The key improvement of InvAdam lies in its parameter update mechanism, which is opposite to that of Adam. Specifically, it computes element-wise multiplication of the first-order and second-order moments, while Adam computes the element-wise division of these two moments. This modification aims to increase the step size of the parameter update when the elements in the second-order moments are large and vice versa, which helps the parameter escape sharp minima and stay at flat ones. However, InvAdam's update mechanism may face challenges in convergence. To address this challenge, we propose dual Adam (DualAdam), which integrates the update mechanisms of both Adam and InvAdam, ensuring convergence while enhancing generalization performance. Additionally, we introduce the diffusion theory to mathematically demonstrate InvAdam's ability to escape sharp minima. Extensive experiments are conducted on image classification tasks and large language model (LLM) fine-tuning. The results validate that DualAdam outperforms Adam and its state-of-the-art variants in terms of generalization performance. The code is publicly available at https://github.com/LongJin-lab/DualAdam.

优化器泛化能力深度学习Adam

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。