Safeguarded Learned Convex Optimization

Howard Heaton,Xiaohan Chen,Zhangyang Wang,Wotao Yin
DOI: https://doi.org/10.1609/aaai.v37i6.25950
2023-01-01
Proceedings of the AAAI Conference on Artificial Intelligence
Abstract:Applications abound in which optimization problems must be repeatedly solved,each time with new (but similar) data. Analytic optimization algorithms can behand-designed to provably solve these problems in an iterative fashion. On onehand, data-driven algorithms can "learn to optimize" (L2O) with much feweriterations and similar cost per iteration as general-purpose optimizationalgorithms. On the other hand, unfortunately, many L2O algorithms lack convergeguarantees. To fuse the advantages of these approaches, we present a Safe-L2Oframework. Safe-L2O updates incorporate a safeguard to guarantee convergencefor convex problems with proximal and/or gradient oracles. The safeguard issimple and computationally cheap to implement, and it is activated only whenthe data-driven L2O updates would perform poorly or appear to diverge. Thisyields the numerical benefits of employing machine learning to create rapid L2Oalgorithms while still guaranteeing convergence. Our numerical examples showconvergence of Safe-L2O algorithms, even when the provided data is not from thedistribution of training data.
What problem does this paper attempt to address?