arXiv:2410.02559math.OCcs.LG2024-10

用随机梯度估计降低零阶优化查询复杂度,提升效率。

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

  • 提出新型收敛性分析框架,兼容两种误差类型
  • 非凸问题查询复杂度降至近似 (n+d)/ε²,显著优化
  • 适合对计算效率敏感的黑箱优化场景

零阶(ZO)优化是梯度计算困难或不可行时的关键技术。现有方差减少的ZO近端算法多采用坐标型估计器以提升精度,但其计算量为O(d),远高于随机估计器的O(1)。本文首次提出零阶目标下降(ZOOD)性质,可统一处理两类误差,并设计两类通用降维框架,只要内部求解器满足ZOOD性质,即可自动导出凸与非凸问题的收敛结果。将该框架应用于新提出的完全随机估计的ZOR-ProxSVRG和ZOR-ProxSAGA算法,在d > n^{1/2}条件下,非凸问题的函数查询复杂度从O(min{dn^{1/2}/ε², d/ε³})改进至~O((n+d)/ε²),凸问题则从O(d/ε²)改进至~O(n log(1/ε) + d/ε)。

原文摘要 · Abstract (English)

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.

零阶优化随机估计查询复杂度近端算法

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。