English

Compander-Aligned Query Geometry for Quantized Zeroth-Order Optimization

Machine Learning 2026-05-12 v1

Abstract

Low-bit forward evaluation is an attractive route to memory-efficient zeroth-order (ZO) adaptation: the optimizer needs only scalar losses, and the model can be queried near deployment precision. The obstacle is that a quantized ZO query is not a continuous finite difference followed by harmless storage rounding. The query chooses endpoints, the low-precision engine rounds them, and the loss difference is measured along the rounded chord. For nonuniform companding quantizers, this makes the codebook insufficient to predict ZO behavior: a fixed weight-space radius can collapse in dense cells, over-span sparse cells, or assign a rounded chord to an unrounded update direction. We identify the missing object as query geometry and model scalar nonuniform quantization as Q=ϕ1UϕQ = \phi^{-1} \circ U \circ \phi. CAQ-ZO (Compander-Aligned Queries for Zeroth-Order Optimization) forms one-grid-step Rademacher stencils z±Δrz \pm \Delta r in z=ϕ(x)z = \phi(x), maps endpoints back through ϕ1\phi^{-1}, and updates in zz. Our theory proves the grid-span mismatch, decomposes endpoint-rounding estimator residuals, and gives stationarity bounds in which generic off-grid queries retain a Δ2/μ2\Delta^2/\mu^2 residual channel while CAQ-ZO makes the query-time residual exactly zero. Synthetic experiments isolate this channel, and matched NF4 Qwen/Llama fine-tuning shows that CAQ-ZO improves the trained NF4 baseline under the same quantizer and evaluation budget.

Keywords

Cite

@article{arxiv.2605.10673,
  title  = {Compander-Aligned Query Geometry for Quantized Zeroth-Order Optimization},
  author = {Yao Shu and Zilin Zhu},
  journal= {arXiv preprint arXiv:2605.10673},
  year   = {2026}
}