Covariate Balancing Propensity Score by Tailored Loss Functions
Not on the map. Why?
Abstract (arXiv)
In observational studies, propensity scores are commonly estimated by maxi- mum likelihood but may fail to balance high-dimensional pre-treatment covariates even after specification search. We introduce a general framework that unifies and generalizes several recent proposals to improve covariate balance when designing an observational study. In- stead of the likelihood function, we propose to optimize special loss functions---covariate balancing scoring rules (CBSR)---to estimate the propensity score. A CBSR is uniquely determined by the link function in the GLM and the estimand (a weighted average treatment effect). We show CBSR does not lose asymptotic efficiency to the Bernoulli likelihood in estimating the weighted average treatment effect compared, but CBSR is much more robust in finite sample. Borrowing tools developed in statistical learning, we propose practical strategies to balance covariate functions in rich function classes. This is useful to estimate the maximum bias of the inverse probability weighting (IPW) estimators and construct honest confidence interval in finite sample. Lastly, we provide several numerical examples to demonstrate the trade-off of bias and variance in the IPW-type estimators and the trade-off in balancing different function classes of the covariates.
Key facts. Cited by 34 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 Jan 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 6; 2022: 4; 2023: 7; 2024: 3; 2025: 8; 2026: 6.
34 papers.
Sort:
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2023Augmented balancing weights as linear regressioncited by 21
- 2021End-to-End Balancing for Causal Continuous Treatment-Effect Estimationcited by 9
- 2024Stabilized Inverse Probability Weighting via Isotonic Calibrationcited by 8
- 2025Direct Bias-Correction Term Estimation for Average Treatment Effect Estimationcited by 8
- 2023Fair and Robust Estimation of Heterogeneous Treatment Effects for Policy Learningcited by 6
- 2021Kernel Ridge Riesz Representers: Generalization, Mis-specification, and the Counterfactual Effective Dimensioncited by 5
- 2021Sequential Kernel Embedding for Mediated and Time-Varying Dose Response Curvescited by 5
- 2022Outcome Assumptions and Duality Theory for Balancing Weightscited by 5
- 2024Differentially Private Covariate Balancing Causal Inferencecited by 5
- 2025Direct Debiased Machine Learning via Bregman Divergence Minimizationcited by 5
- 2021Estimating Average Treatment Effects with Support Vector Machinescited by 3
22 more
- 2021Statistical inference using Regularized M-estimation in the reproducing kernel Hilbert space for handling missing datacited by 3
- 2023Calibrated and Conformal Propensity Scores for Causal Effect Estimationcited by 3
- 2026Optimally taming biases in black-box models for efficient semiparametric estimationcited by 3
- 2023CATE Lasso: Conditional Average Treatment Effect Estimation with High-Dimensional Linear Regressioncited by 2
- 2025A Unified Theory for Causal Inference: Direct Debiased Machine Learning via Bregman-Riesz Regressioncited by 2
- 2025Riesz Regression As Direct Density Ratio Estimationcited by 2
- 2022Benign-Overfitting in Conditional Average Treatment Effect Prediction with Linear Regressioncited by 1
- 2022Estimating causal effects with optimization-based methods: A review and empirical comparisoncited by 1
- 2022Synthetic Principal Component Design: Fast Covariate Balancing with Synthetic Controlscited by 1
- 2023Scalable kernel balancing weights in a nationwide observational study of hospital profit status and heart attack outcomescited by 1
- 2026De-paradox Tree: Breaking Down Simpson's Paradox via A Kernel-Based Partition Algorithmcited by 1
- 2026Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inferencecited by 1
- 2026Optimal use of a black-box learner in semiparametric estimationcited by 1
- 2021Rejoinder: Learning Optimal Distributionally Robust Individualized Treatment Rulescited by 0
- 2023Stable Probability Weighting: Large-Sample and Finite-Sample Estimation and Inference Methods for Heterogeneous Causal Effects of Multivalued Treatments Under Limited Overlapcited by 0
- 2024Triple/Debiased Lasso for Statistical Inference of Conditional Average Treatment Effectscited by 0
- 2025Penalized Empirical Likelihood for Doubly Robust Causal Inference under Contamination in High Dimensionscited by 0
- 2025Data Fusion for High-Resolution Estimationcited by 0
- 2025Learning density ratios in causal inference using Bregman-Riesz regressioncited by 0
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inferencecited by 0
- 2026genriesz: A Python Package for Automatic Debiased Machine Learning with Generalized Riesz Regressioncited by 0
- 2026Covariate Balancing and Riesz Regression Should Be Guided by the Neyman Orthogonal Score in Debiased Machine Learningcited by 0
- 2026Optimal use of a black-box learner in semiparametric estimation23 Jul 2026
- 2026Optimally taming biases in black-box models for efficient semiparametric estimation4 Jun 2026
- 2026Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inference2 Jun 2026
- 2026Covariate Balancing and Riesz Regression Should Be Guided by the Neyman Orthogonal Score in Debiased Machine Learning7 May 2026
- 2026De-paradox Tree: Breaking Down Simpson's Paradox via A Kernel-Based Partition Algorithm2 Mar 2026
- 2026genriesz: A Python Package for Automatic Debiased Machine Learning with Generalized Riesz Regression19 Feb 2026
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inference11 Nov 2025
- 2025Riesz Regression As Direct Density Ratio Estimation6 Nov 2025
- 2025A Unified Theory for Causal Inference: Direct Debiased Machine Learning via Bregman-Riesz Regression30 Oct 2025
- 2025Direct Debiased Machine Learning via Bregman Divergence Minimization27 Oct 2025
- 2025Learning density ratios in causal inference using Bregman-Riesz regression17 Oct 2025
- 2025Direct Bias-Correction Term Estimation for Average Treatment Effect Estimation26 Sep 2025
22 more
- 2025Data Fusion for High-Resolution Estimation20 Aug 2025
- 2025Penalized Empirical Likelihood for Doubly Robust Causal Inference under Contamination in High Dimensions23 Jul 2025
- 2024Stabilized Inverse Probability Weighting via Isotonic Calibration10 Nov 2024
- 2024Differentially Private Covariate Balancing Causal Inference18 Oct 2024
- 2024Triple/Debiased Lasso for Statistical Inference of Conditional Average Treatment Effects5 Mar 2024
- 2023Scalable kernel balancing weights in a nationwide observational study of hospital profit status and heart attack outcomes1 Nov 2023
- 2023CATE Lasso: Conditional Average Treatment Effect Estimation with High-Dimensional Linear Regression25 Oct 2023
- 2023Fair and Robust Estimation of Heterogeneous Treatment Effects for Policy Learning6 Jun 2023
- 2023Calibrated and Conformal Propensity Scores for Causal Effect Estimation1 Jun 2023
- 2023Augmented balancing weights as linear regression27 Apr 2023
- 2023Diagnosing Model Performance Under Distribution Shift3 Mar 2023
- 2023Stable Probability Weighting: Large-Sample and Finite-Sample Estimation and Inference Methods for Heterogeneous Causal Effects of Multivalued Treatments Under Limited Overlap13 Jan 2023
- 2022Synthetic Principal Component Design: Fast Covariate Balancing with Synthetic Controls28 Nov 2022
- 2022Outcome Assumptions and Duality Theory for Balancing Weights17 Mar 2022
- 2022Estimating causal effects with optimization-based methods: A review and empirical comparison28 Feb 2022
- 2022Benign-Overfitting in Conditional Average Treatment Effect Prediction with Linear Regression10 Feb 2022
- 2021Sequential Kernel Embedding for Mediated and Time-Varying Dose Response Curves6 Nov 2021
- 2021Rejoinder: Learning Optimal Distributionally Robust Individualized Treatment Rules17 Oct 2021
- 2021End-to-End Balancing for Causal Continuous Treatment-Effect Estimation27 Jul 2021
- 2021Statistical inference using Regularized M-estimation in the reproducing kernel Hilbert space for handling missing data15 Jul 2021
- 2021Estimating Average Treatment Effects with Support Vector Machines23 Feb 2021
- 2021Kernel Ridge Riesz Representers: Generalization, Mis-specification, and the Counterfactual Effective Dimension22 Feb 2021
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2023Augmented balancing weights as linear regressioncited by 21
- 2021End-to-End Balancing for Causal Continuous Treatment-Effect Estimationcited by 9
- 2024Stabilized Inverse Probability Weighting via Isotonic Calibrationcited by 8
- 2025Direct Bias-Correction Term Estimation for Average Treatment Effect Estimationcited by 8
- 2023Fair and Robust Estimation of Heterogeneous Treatment Effects for Policy Learningcited by 6
- 2021Kernel Ridge Riesz Representers: Generalization, Mis-specification, and the Counterfactual Effective Dimensioncited by 5
- 2021Sequential Kernel Embedding for Mediated and Time-Varying Dose Response Curvescited by 5
- 2022Outcome Assumptions and Duality Theory for Balancing Weightscited by 5
- 2024Differentially Private Covariate Balancing Causal Inferencecited by 5
- 2025Direct Debiased Machine Learning via Bregman Divergence Minimizationcited by 5
- 2021Estimating Average Treatment Effects with Support Vector Machinescited by 3
12 most cited of 34 All 34 (JSON)
- 2026Optimal use of a black-box learner in semiparametric estimation23 Jul 2026
- 2026Optimally taming biases in black-box models for efficient semiparametric estimation4 Jun 2026
- 2026Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inference2 Jun 2026
- 2026Covariate Balancing and Riesz Regression Should Be Guided by the Neyman Orthogonal Score in Debiased Machine Learning7 May 2026
- 2026De-paradox Tree: Breaking Down Simpson's Paradox via A Kernel-Based Partition Algorithm2 Mar 2026
- 2026genriesz: A Python Package for Automatic Debiased Machine Learning with Generalized Riesz Regression19 Feb 2026
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inference11 Nov 2025
- 2025Riesz Regression As Direct Density Ratio Estimation6 Nov 2025
- 2025A Unified Theory for Causal Inference: Direct Debiased Machine Learning via Bregman-Riesz Regression30 Oct 2025
- 2025Direct Debiased Machine Learning via Bregman Divergence Minimization27 Oct 2025
- 2025Learning density ratios in causal inference using Bregman-Riesz regression17 Oct 2025
- 2025Direct Bias-Correction Term Estimation for Average Treatment Effect Estimation26 Sep 2025
12 newest of 34 All 34 (JSON)