arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.10409stat.MLcs.LGmath.PRmath.STstat.TH

高维推断中非凸损失上的在线随机梯度下降

Online stochastic gradient descent on non-convex losses from high-dimensional inference

University of California, Berkeley(加州大学伯克利分校);University of Waterloo(滑铁卢大学)

Gerard Ben Arous, Reza Gheissari, Aukosh Jagannath

AI总结:

本文研究高维推断中非凸损失上的在线随机梯度下降,提出基于信息指数的样本复杂度阈值,揭示初始搜索阶段消耗大部分数据,随后快速收敛。

摘要:

随机梯度下降(SGD)是解决高维推断任务中出现的优化问题的流行算法。在此类任务中,通过迭代优化一个损失函数,从独立数据样本中产生未知参数的估计量。该损失函数是随机的,且通常是非凸的。我们研究了最简单的SGD版本,即在线SGD,在参数空间为高维且从随机起点出发时的性能。我们为随着维度变化时一致估计所需的样本数量建立了近乎尖锐的阈值。我们的阈值仅依赖于总体损失的一个内在性质,我们称之为信息指数。特别是,我们的结果不假设对损失本身的均匀控制,如凸性或均匀导数界。我们获得的阈值在维度上是多项式的,精确的指数显式依赖于信息指数。作为我们结果的一个推论,我们发现除最简单的任务外,几乎所有的数据仅用于初始搜索阶段,以获得与真实参数的非平凡相关性。在获得非平凡相关性后,下降是快速的,并表现出大数定律类型的行为。我们通过将我们的方法应用于广泛的推断任务来说明我们的方法,如相位恢复、广义线性模型的参数估计、在线PCA、尖峰张量模型,以及具有一般激活函数的单层网络的监督学习。

英文摘要:

Stochastic gradient descent (SGD) is a popular algorithm for optimization problems arising in high-dimensional inference tasks. Here one produces an estimator of an unknown parameter from independent samples of data by iteratively optimizing a loss function. This loss function is random and often non-convex. We study the performance of the simplest version of SGD, namely online SGD, from a random start in the setting where the parameter space is high-dimensional. We develop nearly sharp thresholds for the number of samples needed for consistent estimation as one varies the dimension. Our thresholds depend only on an intrinsic property of the population loss which we call the information exponent. In particular, our results do not assume uniform control on the loss itself, such as convexity or uniform derivative bounds. The thresholds we obtain are polynomial in the dimension and the precise exponent depends explicitly on the information exponent. As a consequence of our results, we find that except for the simplest tasks, almost all of the data is used simply in the initial search phase to obtain non-trivial correlation with the ground truth. Upon attaining non-trivial correlation, the descent is rapid and exhibits law of large numbers type behavior. We illustrate our approach by applying it to a wide set of inference tasks such as phase retrieval, and parameter estimation for generalized linear models, online PCA, and spiked tensor models, as well as to supervised learning for single-layer networks with general activation functions.

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

↑