← 返回论文检索
ICML 2026PosterAccept (regular)

Partial Identification under High-Dimensional Potential Outcomes and Confounders via Optimal Transport

Yunfeng Wang, Zhiheng Zhang, Zijun Gao

Fudan University · Tenure-track Assistant Professor, Shanghai University of Finance and Economics · University of Southern California

PDF 由论文原始站点提供,PaperCompass 不保存论文文件。

摘要

Partial identification provides informative causal guarantees when point identification is impossible, but existing approaches based on optimal transport (OT) become computationally and statistically intractable in high-dimensional settings. This limitation is particularly severe when both potential outcomes and confounders are high-dimensional, where classical OT-based bounds suffer from the curse of dimensionality and unfavorable convergence rates. To address this challenge, we propose a novel estimator that decomposes the transport problem into a low-dimensional signal subspace and a high-dimensional residual subspace. Unlike existing projection-based methods that discard residual information, we recover the residual transport energy using the Sliced Wasserstein distance, which is computationally efficient and robust to high dimensions. We establish interpretable conditions controlling the approximation gap based on residual structure and provide a data-driven rule for signal dimension selection. Empirical results show that our estimator consistently outperforms projection-only baselines by recovering lost transport energy, yielding more informative causal bounds while remaining computationally tractable in high dimensions.