arXiv:2604.14587cs.LGmath.OC2026-04

提出改进版Lion优化器CLion,显著提升模型泛化能力

CLion: Efficient Cautious Lion Optimizer with Enhanced Generalization

  • 通过谨慎使用符号函数设计新优化器CLion
  • 理论证明其泛化误差降至O(1/N),优于原Lion的O(1/(Nτ^T))
  • 适合追求高泛化性能的深度学习模型训练者

Lion优化器在深度学习训练中表现优异,但其泛化性能尚未被充分研究。本文基于算法稳定性与数学归纳法,首次证明Lion的泛化误差为O(1/(Nτ^T)),其中N为样本数,τ>0为梯度估计中非零元素的最小绝对值,T为迭代次数。有趣的是,该误差与SignSGD相同。为增强泛化性,本文提出高效且谨慎的Cautious Lion(CLion)优化器,通过谨慎使用符号函数实现。理论分析表明,CLion的泛化误差降低至O(1/N),优于原Lion。同时,证明其在非凸随机优化下具有O(√d/T^{1/4})的快速收敛率,其中d为模型维度。大量实验验证了CLion的有效性。

原文摘要 · Abstract (English)

Lion optimizer is a popular learning-based optimization algorithm in machine learning, which shows impressive performance in training many deep learning models. Although convergence property of the Lion optimizer has been studied, its generalization analysis is still missing. To fill this gap, we study generalization property of the Lion via algorithmic stability based on the mathematical induction. Specifically, we prove that the Lion has a generalization error of $O(\frac{1}{Nτ^T})$, where $N$ is training sample size, and $τ>0$ denotes the smallest absolute value of non-zero element in gradient estimator, and $T$ is the total iteration number. In addition, we obtain an interesting byproduct that the SignSGD algorithm has the same generalization error as the Lion. To enhance generalization of the Lion, we design a novel efficient Cautious Lion (i.e., CLion) optimizer by cautiously using sign function. Moreover, we prove that our CLion has a lower generalization error of $O(\frac{1}{N})$ than $O(\frac{1}{Nτ^T})$ of the Lion, since the parameter $τ$ generally is very small. Meanwhile, we study convergence property of our CLion optimizer, and prove that our CLion has a fast convergence rate of $O(\frac{\sqrt{d}}{T^{1/4}})$ under $\ell_1$-norm of gradient for nonconvex stochastic optimization, where $d$ denotes the model dimension. Extensive numerical experiments demonstrate effectiveness of our CLion optimizer.

优化器泛化性深度学习

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