Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data

Abstract

We develop and analyze algorithms for instrumental variable regression by viewing the problem as a conditional stochastic optimization problem. In the context of least-squares instrumental variable regression, our algorithms neither require matrix inversions nor mini-batches thereby providing a fully online approach for performing instrumental variable regression with streaming data. When the true model is linear, we derive rates of convergence in expectation, that are of order $\mathcal{O}(\log T/T)$ and $\mathcal{O}(1/T^{1-\epsilon})$ for any $\epsilon>0$, respectively under the availability of two-sample and one-sample oracles respectively. Importantly, under the availability of the two-sample oracle, the aforementioned rate is actually agnostic to the relationship between confounder and the instrumental variable demonstrating the flexibility of the proposed approach in alleviating the need for explicit model assumptions required in recent works based on reformulating the problem as min-max optimization problems. Experimental validation is provided to demonstrate the advantages of the proposed algorithms over classical approaches like the 2SLS method.

Cite

Text

Chen et al. "Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data." Neural Information Processing Systems, 2024. doi:10.52202/079017-0835

Markdown

[Chen et al. "Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data." Neural Information Processing Systems, 2024.](https://mlanthology.org/neurips/2024/chen2024neurips-stochastic/) doi:10.52202/079017-0835

BibTeX

@inproceedings{chen2024neurips-stochastic,
  title     = {{Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data}},
  author    = {Chen, Xuxing and Roy, Abhishek and Hu, Yifan and Balasubramanian, Krishnakumar},
  booktitle = {Neural Information Processing Systems},
  year      = {2024},
  doi       = {10.52202/079017-0835},
  url       = {https://mlanthology.org/neurips/2024/chen2024neurips-stochastic/}
}