Convergence of ease-controlled random reshuffling gradient algorithms under Lipschitz smoothness
研究了在Lipschitz光滑条件下,通过轻量计算开销的监控规则和导数自由线搜索来控制增量梯度和随机重排算法的步长,保证收敛到平稳点,并在深度神经网络上验证了效果。
Abstract In this work, we consider minimizing the average of a very large number of smooth and possibly non-convex functions, and we focus on two widely used minibatch frameworks to tackle this optimization problem: incremental gradient (IG) and random reshuffling (RR). We define ease-controlled modifications of the IG/RR schemes, which require a light additional computational effort but can be proved to converge to a stationary point under weak and standard assumptions. In particular, we define two algorithmic schemes in which the IG/RR iteration is controlled by using a watchdog rule and a derivative-free linesearch that activates only sporadically to adjust the stepsize so to guarantee convergence. The two schemes differ in the watchdog rule and the linesearch, which are performed using either a monotonic or a non-monotonic rule. The two schemes allow controlling the updating of the stepsize used in the main IG/RR iteration, avoiding the use of pre-set rules that may drive the stepsize to zero too fast, reducing the effort in designing effective updating rules of the stepsize. We perform computational analysis using different deep neural architectures and a benchmark of varying-size datasets. We compare our implementation with both a full batch gradient method (i.e. L-BFGS) and a fair implementation of IG/RR methods, proving that our algorithms require a similar computational effort compared to the other online algorithms and that the control on the learning rate may allow a faster decrease of the objective function.