Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data
Xuxing Chen, Abhishek Roy, Yifan Hu, Krishnakumar Balasubramanian
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 and provides 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 and for any , respectively under the availability of two-sample and one-sample oracles, respectively, where is the number of iterations. Importantly, under the availability of the two-sample oracle, our procedure avoids explicitly modeling and estimating the relationship between confounder and the instrumental variables, demonstrating the benefit of the proposed approach over recent works based on reformulating the problem as minimax optimization problems. Numerical experiments are provided to corroborate the theoretical results.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 85f76b6d-6bac-4cce-b66a-ef7b2f959caaCited by top-tier papers3
- Differentially Private Two-Stage Gradient Descent for Instrumental Variable RegressionHaodong Liang, Yanhao Jin, Krishna Balasubramanian, Lifeng LaiICLR 2026
- Transformers Handle Endogeneity in In-Context Linear RegressionHaodong Liang, Krishna Balasubramanian, Lifeng LaiICLR 2025
- Transformers with Endogenous In-Context Learning: Bias Characterization and MitigationHaotian Wang, Hao Zou, Haoxuan Li, Haoang Chi et al.ICLR 2026
Builds on15
- Minimax Estimation of Conditional Moment ModelsNishanth Dikkala, Greg Lewis, Lester Mackey, Vasilis SyrgkanisNeurIPS 2020 · 125 citations
- Dual Instrumental Variable RegressionKrikamol Muandet, Arash Mehrjou, Si Kai Lee, Anant RajNeurIPS 2020 · 87 citations
- Learning Deep Features in Instrumental Variable RegressionLiyuan Xu, Yutian Chen, Siddarth Srinivasan, Nando de Freitas et al.ICLR 2021 · 85 citations
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 83 citations
- Proximal Causal Learning with Kernels: Two-Stage Estimation and Moment RestrictionAfsaneh Mastouri, Yuchen Zhu, Limor Gultchin, Anna Korba et al.ICML 2021 · 78 citations
Related papers
- Stochastic Online Instrumental Variable Regression: Regrets for Endogeneity and Bandit FeedbackRiccardo Della Vecchia, Debabrota BasuAAAI 2025 · 7 citations
- Fast Instrument Learning with Faster RatesZiyu Wang, Yuhao Zhou, Jun ZhuNeurIPS 2022 · 6 citations
- Instrumental Variable Regression with Confounder BalancingAnpeng Wu, Kun Kuang, Bo Li, Fei WuICML 2022 · 31 citations
- Hybrid Variance-Reduced SGD Algorithms For Minimax Problems with Nonconvex-Linear FunctionQuoc Tran-Dinh, Deyi Liu, Lam M. NguyenNeurIPS 2020 · 28 citations
- Nonparametric Instrumental Variable Regression through Stochastic Approximate GradientsYuri R. Fonseca, Caio Peixoto, Yuri F. SaporitoNeurIPS 2024 · 8 citations
