Obtaining Lower Query Complexities through Lightweight Zeroth-Order Proximal Gradient Algorithms

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gu, Bin, Wei, Xiyuan, Zhang, Hualin, Chang, Yi, Huang, Heng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909334554804224
author Gu, Bin
Wei, Xiyuan
Zhang, Hualin
Chang, Yi
Huang, Heng
author_facet Gu, Bin
Wei, Xiyuan
Zhang, Hualin
Chang, Yi
Huang, Heng
contents Zeroth-order (ZO) optimization is one key technique for machine learning problems where gradient calculation is expensive or impossible. Several variance reduced ZO proximal algorithms have been proposed to speed up ZO optimization for non-smooth problems, and all of them opted for the coordinated ZO estimator against the random ZO estimator when approximating the true gradient, since the former is more accurate. While the random ZO estimator introduces bigger error and makes convergence analysis more challenging compared to coordinated ZO estimator, it requires only $\mathcal{O}(1)$ computation, which is significantly less than $\mathcal{O}(d)$ computation of the coordinated ZO estimator, with $d$ being dimension of the problem space. To take advantage of the computationally efficient nature of the random ZO estimator, we first propose a ZO objective decrease (ZOOD) property which can incorporate two different types of errors in the upper bound of convergence rate. Next, we propose two generic reduction frameworks for ZO optimization which can automatically derive the convergence results for convex and non-convex problems respectively, as long as the convergence rate for the inner solver satisfies the ZOOD property. With the application of two reduction frameworks on our proposed ZOR-ProxSVRG and ZOR-ProxSAGA, two variance reduced ZO proximal algorithms with fully random ZO estimators, we improve the state-of-the-art function query complexities from $\mathcal{O}\left(\min\{\frac{dn^{1/2}}{ε^2}, \frac{d}{ε^3}\}\right)$ to $\tilde{\mathcal{O}}\left(\frac{n+d}{ε^2}\right)$ under $d > n^{\frac{1}{2}}$ for non-convex problems, and from $\mathcal{O}\left(\frac{d}{ε^2}\right)$ to $\tilde{\mathcal{O}}\left(n\log\frac{1}ε+\frac{d}ε\right)$ for convex problems.
format Preprint
id arxiv_https___arxiv_org_abs_2410_02559
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Obtaining Lower Query Complexities through Lightweight Zeroth-Order Proximal Gradient Algorithms
Gu, Bin
Wei, Xiyuan
Zhang, Hualin
Chang, Yi
Huang, Heng
Optimization and Control
Machine Learning
Zeroth-order (ZO) optimization is one key technique for machine learning problems where gradient calculation is expensive or impossible. Several variance reduced ZO proximal algorithms have been proposed to speed up ZO optimization for non-smooth problems, and all of them opted for the coordinated ZO estimator against the random ZO estimator when approximating the true gradient, since the former is more accurate. While the random ZO estimator introduces bigger error and makes convergence analysis more challenging compared to coordinated ZO estimator, it requires only $\mathcal{O}(1)$ computation, which is significantly less than $\mathcal{O}(d)$ computation of the coordinated ZO estimator, with $d$ being dimension of the problem space. To take advantage of the computationally efficient nature of the random ZO estimator, we first propose a ZO objective decrease (ZOOD) property which can incorporate two different types of errors in the upper bound of convergence rate. Next, we propose two generic reduction frameworks for ZO optimization which can automatically derive the convergence results for convex and non-convex problems respectively, as long as the convergence rate for the inner solver satisfies the ZOOD property. With the application of two reduction frameworks on our proposed ZOR-ProxSVRG and ZOR-ProxSAGA, two variance reduced ZO proximal algorithms with fully random ZO estimators, we improve the state-of-the-art function query complexities from $\mathcal{O}\left(\min\{\frac{dn^{1/2}}{ε^2}, \frac{d}{ε^3}\}\right)$ to $\tilde{\mathcal{O}}\left(\frac{n+d}{ε^2}\right)$ under $d > n^{\frac{1}{2}}$ for non-convex problems, and from $\mathcal{O}\left(\frac{d}{ε^2}\right)$ to $\tilde{\mathcal{O}}\left(n\log\frac{1}ε+\frac{d}ε\right)$ for convex problems.
title Obtaining Lower Query Complexities through Lightweight Zeroth-Order Proximal Gradient Algorithms
topic Optimization and Control
Machine Learning
url https://arxiv.org/abs/2410.02559