Smooth Sailing: Lipschitz-Driven Uncertainty Quantification for Spatial Association
David R. Burt, Renato Berlinghieri, Stephen Bates, Tamara Broderick
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/davidrburt/lipschitz-driven-inferenceOfficialIn papernone★ 2
Abstract
Estimating associations between spatial covariates and responses - rather than merely predicting responses - is central to environmental science, epidemiology, and economics. For instance, public health officials might be interested in whether air pollution has a strictly positive association with a health outcome, and the magnitude of any effect. Standard machine learning methods often provide accurate predictions but offer limited insight into covariate-response relationships. And we show that existing methods for constructing confidence (or credible) intervals for associations fail to provide nominal coverage in the face of model misspecification and distribution shift - despite both being essentially always present in spatial problems. We introduce a method that constructs valid frequentist confidence intervals for associations in spatial settings. Our method requires minimal assumptions beyond a form of spatial smoothness. In particular, we do not require model correctness or covariate overlap between training and target locations. Our approach is the first to guarantee nominal coverage in this setting and outperforms existing techniques in both real and simulated experiments.