Covariate balancing using the integral probability metric for causal inference
Insung Kong, Yuha Park, Joonhyuk Jung, Kwonsang Lee, Yongdai Kim
Abstract
Weighting methods in causal inference have been widely used to achieve a desirable level of covariate balancing. However, the existing weighting methods have desirable theoretical properties only when a certain model, either the propensity score or outcome regression model, is correctly specified. In addition, the corresponding estimators do not behave well for finite samples due to large variance even when the model is correctly specified. In this paper, we consider to use the integral probability metric (IPM), which is a metric between two probability measures, for covariate balancing. Optimal weights are determined so that weighted empirical distributions for the treated and control groups have the smallest IPM value for a given set of discriminators. We prove that the corresponding estimator can be consistent without correctly specifying any model (neither the propensity score nor the outcome regression model). In addition, we empirically show that our proposed method outperforms existing weighting methods with large margins for finite samples.
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 papers8
- Exploiting Geometry for Treatment Effect Estimation via Optimal TransportYuguang Yan, Zeqin Yang, Weilin Chen, Ruichu Cai et al.AAAI 2024 · 4 citations
- A Dual-module Framework for Counterfactual Estimation over TimeXin Wang, Shengfei Lyu, Lishan Yang, Yibing Zhan et al.ICML 2024 · 4 citations
- Reducing Balancing Error for Causal Inference via Optimal TransportYuguang Yan, Hao Zhou, Zeqin Yang, Weilin Chen et al.ICML 2024 · 2 citations
- Testing Distributions against Bounded DistinguishersMark Bun, Rathin Desai, Renato Ferreira Pinto Jr.STOC 2026
- Counterfactual Contrastive Learning with Normalizing Flows for Robust Treatment Effect EstimationJiaxuan Zhang, Emadeldeen Eldele, Fuyuan Cao, Yang Wang et al.ICML 2025
Builds on2
- Generalization Bounds for Estimating Causal Effects of Continuous TreatmentsXin Wang, Shengfei Lyu, Xingyu Wu, Tianhao Wu et al.NeurIPS 2022 · 38 citations
- Learning fair representation with a parametric integral probability metricDongha Kim, Kunwoong Kim, Insung Kong, Ilsang Ohn et al.ICML 2022 · 23 citations
Related papers
- Permutation WeightingDavid Arbour, Drew Dimmery, Arjun SondhiICML 2021 · 24 citations
- Treatment Effects Estimation By Uniform TransformerRuoqi Yu, Shulei WangICLR 2024 · 5 citations
- End-to-End Balancing for Causal Continuous Treatment-Effect EstimationMohammad Taha Bahadori, Eric Tchetgen Tchetgen, David HeckermanICML 2022 · 15 citations
- An effective framework for estimating individualized treatment rulesJoowon Lee, Jared D. Huling, Guanhua ChenNeurIPS 2024 · 4 citations
- Propensity Matters: Measuring and Enhancing Balancing for RecommendationHaoxuan Li, Yanghao Xiao, Chunyuan Zheng, Peng Wu et al.ICML 2023 · 55 citations
