Stochastic Zeroth-Order Optimization Under Heavy-Tailed Noise
摘要
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 -th moment noise for , 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- control of 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- tail condition on the sample-gradient noise, RSC-ZO finds an -stationary point with high probability using noisy function evaluations. This matches the optimal first-order -dependence. At , the bound becomes , matching the classical stochastic ZO dimension--accuracy dependence, but with a high-probability guarantee and under a weaker weak- 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}
}