We propose a new randomized method for solving systems of nonlinear equations, which can find sparse solutions or solutions under certain simple constraints. The scheme only takes gradients of component functions and uses Bregman projections onto the solution space of a Newton equation. In the special case of euclidean projections, the method is known as nonlinear Kaczmarz method. Furthermore, if the component functions are nonnegative, we are in the setting of optimization under the interpolation assumption and the method reduces to SGD with the recently proposed stochastic Polyak step size. For general Bregman projections, our method is a stochastic mirror descent with a novel adaptive step size. We prove that in the convex setting each iteration of our method results in a smaller Bregman distance to exact solutions as compared to the standard Polyak step. Our generalization to Bregman projections comes with the price that a convex one-dimensional optimization problem needs to be solved in each iteration. This can typically be done with globalized Newton iterations. Convergence is proved in two classical settings of nonlinearity: for convex nonnegative functions and locally for functions which fulfill the tangential cone condition. Finally, we show examples in which the proposed method outperforms similar methods with the same memory requirements.
翻译:我们提出一种用于求解非线性方程组的新型随机化方法,该方法能够获得稀疏解或满足特定简单约束的解。该方案仅需利用分量函数的梯度,并在牛顿方程解空间上进行Bregman投影。在欧几里得投影的特殊情形下,该方法即为非线性Kaczmarz法。此外,当分量函数非负时,我们处于插值假设下的优化场景,该方法可简化为采用近期提出的随机Polyak步长的随机梯度下降法。对于一般性Bregman投影,我们的方法是一种具有新型自适应步长的随机镜像下降法。我们证明在凸优化框架下,相较于标准Polyak步长,该方法每次迭代能获得更小的精确解Bregman距离。向Bregman投影的推广需要每次迭代求解一维凸优化问题作为代价,该问题通常可通过全局化牛顿迭代求解。我们在两种经典非线性场景中证明收敛性:凸非负函数情形以及满足切向锥条件的局部函数情形。最后通过算例表明,在同等内存需求下,所提方法性能优于同类方法。