EXAdam: The Power of Adaptive Cross-Moments

ML Adam等 POA
2024年12月29日
本文介绍了一种新的优化算法EXAdam(扩展Adam),它在广泛使用的Adam优化器基础上进行了改进。EXAdam引入了三项关键增强:(1)改进的去偏项,以提高动量估计的准确性;(2)基于梯度的加速机制,以增强对当前损失景观的响应速度;(3)动态步长公式,允许在整个训练过程中持续增加学习率。这些创新协同作用,旨在解决原始Adam算法的局限性,可能提供更好的收敛特性、更强的鞍点逃离能力以及对超参数选择更高的鲁棒性。 我提供了EXAdam各组件及其相互作用的理论分析,突显了该算法在应对复杂优化问题中的潜在优势。实证评估表明,EXAdam在多个方面优于Adam,具体表现为:在应用于CIFAR-10数据集上的卷积神经网络时,EXAdam实现了48.07%更快的收敛速度,并分别在训练、验证和测试准确率上提升了4.6%、4.13%和2.39%。尽管这些结果令人鼓舞,但仍需在更多不同任务中进行进一步的实证验证,以全面评估EXAdam的有效性。 尽管如此,EXAdam代表了自适应优化技术的一个重要进展,对广泛的机器学习应用具有积极的影响。本研究旨在推动更高效、更具适应性和普遍适用性的优化方法在机器学习和人工智能领域的持续发展。
This paper introduces EXAdam ($\textbf{EX}$tended $\textbf{Adam}$), a novel optimization algorithm that builds upon the widely-used Adam optimizer. EXAdam incorporates three key enhancements: (1) new debiasing terms for improved moment estimation, (2) a gradient-based acceleration mechanism for increased responsiveness to the current loss landscape, and (3) a dynamic step size formula that allows for continuous growth of the learning rate throughout training. These innovations work synergistically to address limitations of the original Adam algorithm, potentially offering improved convergence properties, enhanced ability to escape saddle points, and greater robustness to hyperparameter choices. I provide a theoretical analysis of EXAdam's components and their interactions, highlighting the algorithm's potential advantages in navigating complex optimization landscapes. Empirical evaluations demonstrate EXAdam's superiority over Adam, achieving 48.07% faster convergence and yielding improvements of 4.6%, 4.13%, and 2.39% in training, validation, and testing accuracies, respectively, when applied to a CNN trained on the CIFAR-10 dataset. While these results are promising, further empirical validation across diverse tasks is essential to fully gauge EXAdam's efficacy. Nevertheless, EXAdam represents a significant advancement in adaptive optimization techniques, with promising implications for a wide range of machine learning applications. This work aims to contribute to the ongoing development of more efficient, adaptive, and universally applicable optimization methods in the field of machine learning and artificial intelligence.
许愿