English

Quasi-Bayesian Dual Instrumental Variable Regression

Machine Learning 2021-11-04 v2 Machine Learning

Abstract

Recent years have witnessed an upsurge of interest in employing flexible machine learning models for instrumental variable (IV) regression, but the development of uncertainty quantification methodology is still lacking. In this work we present a novel quasi-Bayesian procedure for IV regression, building upon the recently developed kernelized IV models and the dual/minimax formulation of IV regression. We analyze the frequentist behavior of the proposed method, by establishing minimax optimal contraction rates in L2L_2 and Sobolev norms, and discussing the frequentist validity of credible balls. We further derive a scalable inference algorithm which can be extended to work with wide neural network models. Empirical evaluation shows that our method produces informative uncertainty estimates on complex high-dimensional problems.

Keywords

Cite

@article{arxiv.2106.08750,
  title  = {Quasi-Bayesian Dual Instrumental Variable Regression},
  author = {Ziyu Wang and Yuhao Zhou and Tongzheng Ren and Jun Zhu},
  journal= {arXiv preprint arXiv:2106.08750},
  year   = {2021}
}

Comments

Extended version of the NeurIPS'21 paper. ZW and YZ contribute equally

R2 v1 2026-06-24T03:15:53.735Z