中文

Stochastic Zeroth-Order Optimization Under Heavy-Tailed Noise

最优化与控制 2026-05-19 v1

摘要

We study stochastic zeroth-order (ZO) optimization of smooth nonconvex objectives under heavy-tailed sample-gradient noise. This regime is motivated by empirical evidence that gradient noise in modern machine learning can violate the bounded-variance assumptions used in classical ZO theory. While first-order methods have optimal rates under bounded pp-th moment noise for p(1,2]p\in(1,2], analogous high-probability guarantees for nonconvex ZO methods are much less understood. The ZO setting is not a direct corollary of first-order theory. First-order methods observe stochastic gradients, whereas derivative-free methods only query noisy function values and build finite-difference estimates. Thus, weak-LpL_p control of F(x;ξ)f(x)\nabla F(x;\xi)-\nabla f(x) must first be transferred to scalar directional estimates. We propose the Robust Scalar-Clipped Zeroth-Order method (RSC-ZO), a two-point method that clips each scalar directional derivative before aggregation. Under sample-wise smoothness and a weak-LpL_p tail condition on the sample-gradient noise, RSC-ZO finds an ε\varepsilon-stationary point with high probability using O~ ⁣(dp2(p1)ε3p2p1) \widetilde{O}\!\left( d^{\frac{p}{2(p-1)}}\varepsilon^{-\frac{3p-2}{p-1}} \right) noisy function evaluations. This matches the optimal first-order ε\varepsilon-dependence. At p=2p=2, the bound becomes O~(dε4)\widetilde{O}(d\varepsilon^{-4}), matching the classical stochastic ZO dimension--accuracy dependence, but with a high-probability guarantee and under a weaker weak-L2L_2 condition that can allow infinite variance. We also analyze a momentum variant and quantify its batch-size/stepsize tradeoff.

关键词

引用

@article{arxiv.2605.17394,
  title  = {Stochastic Zeroth-Order Optimization Under Heavy-Tailed Noise},
  author = {Taha El Bakkali and El Mahdi Chayti and Qiuyi Zhang and Imane Rahali and Omar Saadi},
  journal= {arXiv preprint arXiv:2605.17394},
  year   = {2026}
}