Accelerated Zeroth-Order and First-Order Momentum Methods from Mini to Minimax Optimization
Abstract
In the paper, we propose a class of accelerated zeroth-order and first-order momentum methods for both nonconvex mini-optimization and minimax-optimization. Specifically, we propose a new accelerated zeroth-order momentum (Acc-ZOM) method for black-box mini-optimization where only function values can be obtained. Moreover, we prove that our Acc-ZOM method achieves a lower query complexity of for finding an -stationary point, which improves the best known result by a factor of where denotes the variable dimension. In particular, our Acc-ZOM does not need large batches required in the existing zeroth-order stochastic algorithms. Meanwhile, we propose an accelerated zeroth-order momentum descent ascent (Acc-ZOMDA) method for black-box minimax optimization, where only function values can be obtained. Our Acc-ZOMDA obtains a low query complexity of without requiring large batches for finding an -stationary point, where and denote variable dimensions and is condition number. Moreover, we propose an accelerated first-order momentum descent ascent (Acc-MDA) method for minimax optimization, whose explicit gradients are accessible. Our Acc-MDA achieves a low gradient complexity of without requiring large batches for finding an -stationary point. In particular, our Acc-MDA can obtain a lower gradient complexity of with a batch size , which improves the best known result by a factor of . Extensive experimental results on black-box adversarial attack to deep neural networks and poisoning attack to logistic regression demonstrate efficiency of our algorithms.
Cite
@article{arxiv.2008.08170,
title = {Accelerated Zeroth-Order and First-Order Momentum Methods from Mini to Minimax Optimization},
author = {Feihu Huang and Shangqian Gao and Jian Pei and Heng Huang},
journal= {arXiv preprint arXiv:2008.08170},
year = {2022}
}
Comments
Published in Journal of Machine Learning Research (JMLR)