Optimal Control Variates for Survey Sampling and Causal Inference
arXiv:2608.15333
2026
Optimization
1 ideas extracted · analyzed Sep 1, 2026
What the math gives to ML
The paper's transferable asset is a design-aware control-variate construction: instead of choosing a baseline heuristically, it selects basis directions that optimally explain covariance induced by sampling or exposure mechanisms and outcome uncertainty. In neural-network training, the same principle can reduce minibatch gradient noise by projecting per-example gradients onto a small set of known, cheap feature directions and subtracting the predictable component. The most promising implementation is an optimizer-side control variate with online covariance estimates and spectral truncation to control computational and numerical costs.
Ideas from this paper
Unverified
2026
Replace a raw minibatch gradient with an unbiased control-variate estimator that subtracts predictable components of per-example gradients and adds back their exactly or cheaply known population mean. Select the control-variate directions using leading eigenvectors of an online covariance operator, rather than using arbitrary scalar baselines. This should reduce gradient variance at fixed batch size and permit fewer examples per optimization step.
Useful6/10
Difficulty6/10
Novelty6/10