arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.13680cs.LG
收藏

非凸随机优化中梯度裁剪的高概率分析

High Probability Analysis for Non-Convex Stochastic Optimization with Clipping

Renmin University of China(中国人民大学);Beijing Key Laboratory of Big Data Management and Analysis Methods(北京市大数据管理与分析方法重点实验室)

Shaojie Li, Yong Liu

AI总结:

本文为非凸随机优化中的梯度裁剪算法(如SGD及其变体)提供高概率分析,在重尾假设下同时推导优化界与泛化界,填补了理论空白。

摘要:

梯度裁剪是一种常用于稳定神经网络训练过程的技术。越来越多的研究表明,梯度裁剪也是处理随机优化中出现重尾行为的一种有前景的技术。尽管梯度裁剪具有重要意义,但其理论保证却十分稀缺。大多数理论保证仅提供期望分析,且仅关注优化性能。在本文中,我们在非凸设置下提供高概率分析,并同时为带有梯度裁剪的流行随机优化算法(包括随机梯度下降及其动量变体和自适应步长变体)推导出优化界和泛化界。借助梯度裁剪,我们研究了一个重尾假设,即梯度仅具有有界的$\alpha$阶矩,其中$\alpha\in(1,2]$,这比标准的有界二阶矩假设弱得多。总体而言,我们的研究为带有裁剪的随机优化算法的理论保证提供了一个相对完整的图景。

英文摘要:

Gradient clipping is a commonly used technique to stabilize the training process of neural networks. A growing body of studies has shown that gradient clipping is a promising technique for dealing with the heavy-tailed behavior that emerged in stochastic optimization as well. While gradient clipping is significant, its theoretical guarantees are scarce. Most theoretical guarantees only provide an in-expectation analysis and only focus on optimization performance. In this paper, we provide high probability analysis in the non-convex setting and derive the optimization bound and the generalization bound simultaneously for popular stochastic optimization algorithms with gradient clipping, including stochastic gradient descent and its variants of momentum and adaptive stepsizes. With the gradient clipping, we study a heavy-tailed assumption that the gradients only have bounded $α$-th moments for some $α\in (1, 2]$, which is much weaker than the standard bounded second-moment assumption. Overall, our study provides a relatively complete picture for the theoretical guarantee of stochastic optimization algorithms with clipping.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑