arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.11860math.OCcs.LG

Stochastic Frank-Wolfe for Constrained Finite-Sum Minimization

University of California, Berkeley(加州大学伯克利分校);ETH Zurich(苏黎世联邦理工学院);SumUp Analytics(SumUp分析公司);Max-Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所);MIT Sloan School of Management(麻省理工学院斯隆管理学院);Google Research(谷歌研究院)

Geoffrey Négiar, Gideon Dresdner, Alicia Tsai, Laurent El Ghaoui, Francesco Locatello, Robert M. Freund, Fabian Pedregosa

英文摘要:

We propose a novel Stochastic Frank-Wolfe (a.k.a. conditional gradient) algorithm for constrained smooth finite-sum minimization with a generalized linear prediction/structure. This class of problems includes empirical risk minimization with sparse, low-rank, or other structured constraints. The proposed method is simple to implement, does not require step-size tuning, and has a constant per-iteration cost that is independent of the dataset size. Furthermore, as a byproduct of the method we obtain a stochastic estimator of the Frank-Wolfe gap that can be used as a stopping criterion. Depending on the setting, the proposed method matches or improves on the best computational guarantees for Stochastic Frank-Wolfe algorithms. Benchmarks on several datasets highlight different regimes in which the proposed method exhibits a faster empirical convergence than related methods. Finally, we provide an implementation of all considered methods in an open-source package.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑