English

A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimization

Optimization and Control 2022-10-11 v3 Statistics Theory Machine Learning Statistics Theory

Abstract

We propose a projection-free conditional gradient-type algorithm for smooth stochastic multi-level composition optimization, where the objective function is a nested composition of TT functions and the constraint set is a closed convex set. Our algorithm assumes access to noisy evaluations of the functions and their gradients, through a stochastic first-order oracle satisfying certain standard unbiasedness and second moment assumptions. We show that the number of calls to the stochastic first-order oracle and the linear-minimization oracle required by the proposed algorithm, to obtain an ϵ\epsilon-stationary solution, are of order OT(ϵ2)\mathcal{O}_T(\epsilon^{-2}) and OT(ϵ3)\mathcal{O}_T(\epsilon^{-3}) respectively, where OT\mathcal{O}_T hides constants in TT. Notably, the dependence of these complexity bounds on ϵ\epsilon and TT are separate in the sense that changing one does not impact the dependence of the bounds on the other. Moreover, our algorithm is parameter-free and does not require any (increasing) order of mini-batches to converge unlike the common practice in the analysis of stochastic conditional gradient-type algorithms.

Keywords

Cite

@article{arxiv.2202.04296,
  title  = {A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimization},
  author = {Tesi Xiao and Krishnakumar Balasubramanian and Saeed Ghadimi},
  journal= {arXiv preprint arXiv:2202.04296},
  year   = {2022}
}

Comments

To appear in NeurIPS 2022