中文

具有平滑截断的类 Adam 算法达到全局最小值:基于泛函随机微分方程遍历性的分析

机器学习 2023-12-06 v1 最优化与控制 概率论

摘要

在本文中,我们证明了具有平滑截断的 Adam 型算法能够逼近正则化非凸损失函数的全局最小化器。通过添加平滑截断并将状态空间取为所有轨迹的集合,我们可以对该算法应用马尔可夫半群的遍历理论,并研究其渐近行为。我们在本文中建立的遍历理论,将该算法的收敛性、泛化误差和离散化误差的评估问题,简化为评估两个具有不同漂移系数的泛函随机微分方程(SDE)之间差异的问题。作为我们分析的结果,我们证明了该算法最小化正则化非凸损失函数的误差形式为 n1/2n^{-1/2}η1/4\eta^{1/4}β1log(β+1)\beta^{-1} \log (\beta + 1)ecte^{- c t}。这里,cc 是常数,nnη\etaβ\betatt 分别表示训练数据集大小、学习率、逆温度和时间。

关键词

引用

@article{arxiv.2312.02182,
  title  = {Adam-like Algorithm with Smooth Clipping Attains Global Minima: Analysis Based on Ergodicity of Functional SDEs},
  author = {Keisuke Suzuki},
  journal= {arXiv preprint arXiv:2312.02182},
  year   = {2023}
}