On the Optimal Construction of Unbiased Gradient Estimators for Zeroth-Order Optimization

Shaocong Ma, Heng Huang
10/22/2025
cs.LGcs.AImath.OC

No official code found. Best implementation: mshaocong/unbiased-zoo-optimization (1 star, MIT, last commit 2025-09-27).

Abstract

Zeroth-order optimization (ZOO) is an important framework for stochastic optimization when gradients are unavailable or expensive to compute. A potential limitation of existing ZOO methods is the bias inherent in most gradient estimators unless the perturbation stepsize vanishes. In this paper, we overcome this biasedness issue by proposing a novel family of unbiased gradient estimators based solely on function evaluations. By reformulating directional derivatives as a telescoping series and sampling from carefully designed distributions, we construct estimators that eliminate bias while maintaining favorable variance. We analyze their theoretical properties, derive optimal scaling distributions and perturbation stepsizes of four specific constructions, and prove that SGD using the proposed estimators achieves optimal complexity for smooth non-convex objectives. Experiments on synthetic tasks and language model fine-tuning confirm the superior accuracy and convergence of our approach compared to standard methods.

View on arXivView PDF

Code Implementations(1)

MIT

[NeurIPS 2025 Spotlight] Codes for the paper "On the Optimal Construction of Unbiased Gradient Estimators for Zeroth-Order Optimization"

10Shell, PythonSep 25, 20251 years agoMIT

Similar Papers

Cite this paper

@article{ma2025optimal,
  title  = {On the Optimal Construction of Unbiased Gradient Estimators for Zeroth-Order Optimization},
  author = {Shaocong Ma and Heng Huang},
  year   = {2025},
  eprint = {2510.19953},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.19953v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2510.19953.svg)](https://codeofpaper.com/paper/2510.19953)

Discussion