Causal Matrix Completion under Multiple Treatments via Mixed Synthetic Nearest Neighbors
Minrui Luo, Zhiheng Zhang
Abstract
Synthetic Nearest Neighbors (SNN) provides a principled solution to causal matrix completion under missing-not-at-random (MNAR) by exploiting local low-rank structure through fully observed anchor submatrices. However, its effectiveness critically relies on sufficient data availability within each treatment level, a condition that often fails in settings with multiple or complex treatments. In this work, we propose Mixed Synthetic Nearest Neighbors (MSNN), a new entry-wise causal identification estimator that integrates information across treatment levels. We show that MSNN retains the finite-sample error bounds and asymptotic normality guarantees of SNN, while enlarging the effective sample size available for estimation. Empirical results on synthetic and real-world datasets illustrate the efficacy of the proposed approach, especially under data-scarce treatment levels.
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.
Builds on1
Related papers
- Identifiable Generative models for Missing Not at Random Data ImputationChao Ma, Cheng ZhangNeurIPS 2021 · 56 citations
- Estimation and Imputation in Probabilistic Principal Component Analysis with Missing Not At Random DataAude Sportisse, Claire Boyer, Julie JosseNeurIPS 2020 · 38 citations
- A Pairwise Pseudo-likelihood Approach for Matrix Completion with Informative MissingnessJiangyuan Li, Jiayi Wang, Raymond K. W. Wong, Kwun Chuen Gary ChanNeurIPS 2024 · 8 citations
- Identification and Estimation for Nonignorable Missing Data: A Data Fusion ApproachZixiao Wang, AmirEmad Ghassami, Ilya ShpitserICML 2024 · 1 citation
- Learning Treatment Effects in Panels with General Intervention PatternsVivek F. Farias, Andrew A. Li, Tianyi PengNeurIPS 2021 · 11 citations
