Accelerated Forward-Backward Optimization Using Deep Learning
提出几种有收敛保证的深度学习加速优化求解器,利用加速前向-后向方案的分析思想,训练深度神经网络在特定集合中选择最优更新,在光滑和非光滑优化问题上优于传统加速求解器。
We propose several deep-learning accelerated optimization solvers with convergence guarantees. We use ideas from the analysis of accelerated forward-backward schemes like FISTA, but instead of the classical approach of proving convergence for a choice of parameters, such as a step-size, we show convergence whenever the update is chosen in a specific set. Rather than picking a point in this set using some predefined method, we train a deep neural network to pick the best update within a given space. Finally, we show that the method is applicable to several cases of smooth and nonsmooth optimization and show superior results to established accelerated solvers.