Permutation Weighting
David Arbour, Drew Dimmery, Arjun Sondhi
Abstract
This work introduces permutation weighting: a weighting estimator for observational causal inference under general treatment regimes which preserves arbitrary measures of covariate balance. We show that estimating weights which obey balance constraints is equivalent to a simple binary classification problem between the observed data and a permuted dataset (no matter the cardinality of treatment). Arbitrary probabilistic classifiers may be used in this method; the hypothesis space of the classifier corresponds to the nature of the balance constraints imposed through the resulting weights.We show equivalence between existing covariate balancing weight estimators and permutation weighting and demonstrate estimation with improved efficiency through this regime. We provide theoretical results on the consistency of estimation of causal effects, and the necessity of balance infinite samples. Empirical evaluations indicate that the proposed method outperforms existing state of the art weighting methods for causal effect estimation, even when the data generating process corresponds to the assumptions imposed by prior work.
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.
Cited by top-tier papers11
- Diagnosing failures of fairness transfer across distribution shift in real-world medical settingsJessica Schrouff, Natalie Harris, Sanmi Koyejo, Ibrahim M. Alabdulmohsin et al.NeurIPS 2022 · 84 citations
- Causally motivated multi-shortcut identification and removalJiayun Zheng, Maggie MakarNeurIPS 2022 · 27 citations
- End-to-End Balancing for Causal Continuous Treatment-Effect EstimationMohammad Taha Bahadori, Eric Tchetgen Tchetgen, David HeckermanICML 2022 · 15 citations
- Contrastive Balancing Representation Learning for Heterogeneous Dose-Response Curves EstimationMinqin Zhu, Anpeng Wu, Haoxuan Li, Ruoxuan Xiong et al.AAAI 2024 · 12 citations
- Mining Stable Preferences: Adaptive Modality Decorrelation for Multimedia RecommendationJinghao Zhang, Qiang Liu, Shu Wu, Liang WangSIGIR 2023 · 8 citations
Related papers
- Treatment Effects Estimation By Uniform TransformerRuoqi Yu, Shulei WangICLR 2024 · 5 citations
- Covariate balancing using the integral probability metric for causal inferenceInsung Kong, Yuha Park, Joonhyuk Jung, Kwonsang Lee et al.ICML 2023 · 9 citations
- Reducing Confounding Bias without Data Splitting for Causal Inference via Optimal TransportYuguang Yan, Zongyu Li, Haolin Yang, Zeqin Yang et al.ICML 2025
- Instrumental Variable Regression with Confounder BalancingAnpeng Wu, Kun Kuang, Bo Li, Fei WuICML 2022 · 31 citations
- DeepMatch: Balancing Deep Covariate Representations for Causal Inference Using Adversarial TrainingNathan KallusICML 2020 · 84 citations
