Three Operator Splitting with Subgradients, Stochastic Gradients, and Adaptive Learning Rates
2021 (Engelska) Ingår i: Advances in Neural Information Processing Systems 34 (NeurIPS 2021) / [ed] M. Ranzato, A. Beygelzimer, P.S. Liang, J.W. Vaughan, Y. Dauphin, San Diego: Neural Information Processing Systems , 2021, s. 19743-19756Konferensbidrag, Publicerat paper (Refereegranskat)
Abstract [en]
Three Operator Splitting (TOS) (Davis & Yin, 2017) can minimize the sum of multiple convex functions effectively when an efficient gradient oracle or proximal operator is available for each term. This requirement often fails in machine learning applications: (i) instead of full gradients only stochastic gradients may be available; and (ii) instead of proximal operators, using subgradients to handle complex penalty functions may be more efficient and realistic. Motivated by these concerns, we analyze three potentially valuable extensions of TOS. The first two permit using subgradients and stochastic gradients, and are shown to ensure a O(1/√t) convergence rate. The third extension AdapTOS endows TOS with adaptive step-sizes. For the important setting of optimizing a convex loss over the intersection of convex sets AdapTOS attains universal convergence rates, i.e., the rate adapts to the unknown smoothness degree of the objective. We compare our proposed methods with competing methods on various applications.
Ort, förlag, år, upplaga, sidor San Diego: Neural Information Processing Systems , 2021. s. 19743-19756
Nyckelord [en]
Three operator splitting, Davis-Yin splitting, adaptive optimization, nonsmooth optimization, subgradient method, stochastic optimization, proximal methods, proximal gradient method
Nationell ämneskategori
Beräkningsmatematik
Identifikatorer URN: urn:nbn:se:umu:diva-190486 Scopus ID: 2-s2.0-85131876010 OAI: oai:DiVA.org:umu-190486 DiVA, id: diva2:1620755
Konferens Thirty-fifth Conference on Neural Information Processing Systems (NeruIPS 2021), Online, 6-14 December, 2021.
Forskningsfinansiär Wallenberg AI, Autonomous Systems and Software Program (WASP)
Anmärkning ISBN: 978-171384539-3
2021-12-162021-12-162024-07-02 Bibliografiskt granskad