English

Spatially Robust Inference with Predicted and Missing at Random Labels

Machine Learning 2026-03-13 v1 Machine Learning Econometrics Applications Methodology

Abstract

When outcome data are expensive or onerous to collect, scientists increasingly substitute predictions from machine learning and AI models for unlabeled cases, a process which has consequences for downstream statistical inference. While recent methods provide valid uncertainty quantification under independent sampling, real-world applications involve missing at random (MAR) labeling and spatial dependence. For inference in this setting, we propose a doubly robust estimator with cross-fit nuisances. We show that cross-fitting induces fold-level correlation that distorts spatial variance estimators, producing unstable or overly conservative confidence intervals. To address this, we propose a jackknife spatial heteroscedasticity and autocorrelation consistent (HAC) variance correction that separates spatial dependence from fold-induced noise. Under standard identification and dependence conditions, the resulting intervals are asymptotically valid. Simulations and benchmark datasets show substantial improvement in finite-sample calibration, particularly under MAR labeling and clustered sampling.

Keywords

Cite

@article{arxiv.2603.11368,
  title  = {Spatially Robust Inference with Predicted and Missing at Random Labels},
  author = {Stephen Salerno and Zhenke Wu and Tyler McCormick},
  journal= {arXiv preprint arXiv:2603.11368},
  year   = {2026}
}
R2 v1 2026-07-01T11:15:40.356Z