Feasible Fusion: Constrained Joint Estimation under Structural Non-Overlap
Yuxi Du, Zhiheng Zhang, Haoxuan Li, Cong Fang, Jixing Xu, Zhen Peng, Jiecheng Guo
Abstract
Causal inference in modern large-scale systems faces growing challenges, including high-dimensional covariates, multi-valued treatments, massive observational (OBS) data, and limited randomized controlled trial (RCT) samples due to cost constraints. We formalize treatment-induced structural non-overlap and show that, under this regime, commonly used weighted fusion methods provably fail to satisfy randomized identifying restrictions.To address this issue,we propose a constrained joint estimation framework that minimizes observational risk while enforcing causal validity through orthogonal experimental moment conditions. We further show that structural non-overlap creates a feasibility obstruction for moment enforcement in the original covariate space.We also derive a penalized primal–dual algorithm that jointly learns representations and predictors, and establish oracle inequalities decomposing error into overlap recovery, moment violation, and statistical terms.Extensive synthetic experiments demonstrate robust performance under varying degrees of non-overlap. A large-scale ride-hailing application shows that our method achieves substantial gains over existing baselines, matching the performance of models trained with significantly more RCT data.
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 7453d1cc-3393-4eab-9d4e-4dd89903bbd4Related papers
- Robust Causal Discovery Under Imperfect Structural ConstraintsZidong Wang, Xi Lin, Chuchao He, Xiaoguang GaoAAAI 2026
- Identification and Estimation of the Bi-Directional MR with Some Invalid InstrumentsFeng Xie, Zhen Yao, Lin Xie, Yan Zeng et al.NeurIPS 2024 · 2 citations
- Falsification before Extrapolation in Causal Effect EstimationZeshan M. Hussain, Michael Oberst, Ming-Chieh Shih, David A. SontagNeurIPS 2022 · 11 citations
- Efficient Online Estimation of Causal Effects by Deciding What to ObserveShantanu Gupta, Zachary C. Lipton, David ChildersNeurIPS 2021 · 19 citations
- Overlap-weighted orthogonal meta-learner for treatment effect estimation over timeKonstantin Hess, Dennis Frauen, Mihaela van der Schaar, Stefan FeuerriegelICLR 2026 · 7 citations
