改进的自适应PID优化器,提升深度学习收敛与稳定性。
An Improved Adaptive PID Optimizer with Enhanced Convergence and Stability for Deep Learning
- 融合非递增学习率与梯度差调制因子,改进自适应PID优化器。
- 在多个数据集上优于现有优化器,尤其在真实世界数据上表现突出。
- 适合追求高收敛性与稳定性的深度学习模型训练者使用。
优化是深度学习的核心。主流优化器基于动量随机梯度下降,但存在梯度噪声大和超调问题。为应对噪声,提出了Adam;为解决超调,提出基于控制论的PID优化器。近期出现多种自适应PID(AdaPID)变体,但仍继承Adam的收敛与稳定性缺陷。本文提出IAdaPID-ADG优化器,首次将AMSGrad中的非递增有效学习率机制与DiffGrad中的梯度差调制因子引入AdaPID。在MNIST、CIFAR10、IARC和AnnoCerv等多个数据集上测试,结果显著优于所有对比优化器。消融实验验证了各组件贡献。在相同训练条件下,该方法在图像分类任务中达到更高准确率且更稳定。
原文摘要 · Abstract (English)
Optimization is essential in deep learning. The foundational method upon which most optimizers are built is momentum-based stochastic gradient descent. However, it suffers from two key drawbacks. First, it has noisy and varying gradients, and second, it has an overshoot phenomenon. To address noisy gradients, Adam was proposed, which remains the most widely used adaptive optimizer. To address the overshoot phenomenon, a control-theory-based PID optimizer was proposed. To tackle both the limitations within a single framework, several variants of Adaptive PID (AdaPID) have recently been proposed. Although AdaPID performs well, it still inherits two critical drawbacks from Adam, namely convergence and stability issues. In this work, we address both these limitations. To fix the convergence issue, we uniquely integrate the idea of using a non-increasing effective learning rate into AdaPID (originally proposed in AMSGrad, an extension of Adam). To fix the stability issue, we innovatively integrate a gradient difference based modulation factor into AdaPID (originally proposed in DiffGrad, another extension of Adam). Combining both these ideas in AdaPID, results in our novel IAdaPID-ADG optimizer. We evaluate our proposed optimizer on multiple datasets, including benchmark datasets (MNIST and CIFAR10) and real-world datasets (IARC and AnnoCerv). The IAdaPID-ADG substantially outperforms all competing optimizers. Additionally, we perform an ablation study on the MNIST dataset to demonstrate the contribution of each added component.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。