NeurIPS2023
First Order Stochastic Optimization with Oblivious Noise
Ilias Diakonikolas, Sushrut Karmalkar, Jongho Park, Christos Tzamos
1 citation
Abstract
We initiate the study of stochastic optimization with oblivious noise, broadly generalizing the standard heavy-tailed noise setup. In our setting, in addition to random observation noise, the stochastic gradient may be subject to independent oblivious noise, which may not have bounded moments and is not necessarily centered. Specifically, we assume access to a noisy oracle for the stochastic gradient of at , which returns a vector , where is the bounded variance observation noise and is the oblivious noise that is independent of and . The only assumption we make on the oblivious noise is that for some . In this setting, it is not information-theoretically possible to recover a single solution close to the target when the fraction of inliers is less than . Our main result is an efficient list-decodable learner that recovers a small list of candidates, at least one of which is close to the true solution. On the other hand, if , where is sufficiently small constant, the algorithm recovers a single solution. Along the way, we develop a rejection-sampling-based algorithm to perform noisy location estimation, which may be of independent interest.