Balancing Covariates for Continuous Treatments by Exploiting Continuity
Abstract
We consider causal effect estimation in observational studies with continuous treatment. We introduce a general class of causal estimands that, as a special case, includes the average causal effect of assigning treatment according to an arbitrary counterfactual distribution. Balancing estimators are particularly attractive tools for covariate adjustment in this setting because they avoid directly modeling the treatment assignment mechanism, thereby reducing the potential for instability and misspecification bias in subsequent causal effect estimation. In this paper, we develop a general framework of covariate balancing for continuous treatment with the proposed class of causal estimands, leveraging the assumption of continuity of treatment dose. As in the binary treatment setting, analysts specify the basis functions to be balanced, the allowable degree of covariate imbalance, and a criterion for selecting among admissible weights. Under a minimum variance criterion for choosing weights, we establish consistency, characterize the effect of basis misspecification, and prove asymptotic normality. We develop consistent variance estimation, which yields asymptotically valid confidence intervals. Under homoskedasticity, our estimator is provably more efficient than inverse probability weighting based on the \emph{true} propensity score. Simulation studies support our theoretical results, and we illustrate the proposed method by estimating the effect of vocational training hours on expected future earnings using data from the U.S. Job Corps study.
Related
- Weighted Spline-Expanded Networks with Distributional Balancing for Continuous Treatment Effects
- Nonlinear Sufficient Dimension Reduction for Causal Inference with Continuous Treatments
- Separately continuous maps into regular spaces and continuity of operations in separable semitopological groups
- An End-to-End Pipeline for Causal ML with Continuous Treatments: An Application to Financial Decision Making
- Regression adjustment in completely randomized experiments with many covariates
BibTeX
@misc{sengupta2026balancing,
title = {{Balancing Covariates for Continuous Treatments by Exploiting Continuity}},
author = {Souhardya Sengupta and Kosuke Imai},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.41186},
eprint = {2610.41186},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.41186},
}
RIS
TY - UNPB
TI - Balancing Covariates for Continuous Treatments by Exploiting Continuity
AU - Souhardya Sengupta
AU - Kosuke Imai
PY - 2026
DA - 2026/10/30
PB - arXiv
DO - 10.48550/arxiv.2610.41186
UR - https://arxiv.org/abs/2610.41186
AB - We consider causal effect estimation in observational studies with continuous treatment. We introduce a general class of causal estimands that, as a special case, includes the average causal effect of assigning treatment according to an arbitrary counterfactual distribution. Balancing estimators are particularly attractive tools for covariate adjustment in this setting because they avoid directly modeling the treatment assignment mechanism, thereby reducing the potential for instability and misspecification bias in subsequent causal effect estimation. In this paper, we develop a general framework of covariate balancing for continuous treatment with the proposed class of causal estimands, leveraging the assumption of continuity of treatment dose. As in the binary treatment setting, analysts specify the basis functions to be balanced, the allowable degree of covariate imbalance, and a criterion for selecting among admissible weights. Under a minimum variance criterion for choosing weights, we establish consistency, characterize the effect of basis misspecification, and prove asymptotic normality. We develop consistent variance estimation, which yields asymptotically valid confidence intervals. Under homoskedasticity, our estimator is provably more efficient than inverse probability weighting based on the \emph{true} propensity score. Simulation studies support our theoretical results, and we illustrate the proposed method by estimating the effect of vocational training hours on expected future earnings using data from the U.S. Job Corps study.
ER -