Abstract
Summary
The current statistical literature on causal inference is mostly concerned with binary or categorical exposures, even though exposures of a quantitative nature are frequently encountered in epidemiologic research. In this article, we review the available methods for estimating the dose–response curve for a quantitative exposure, which include ordinary regression based on an outcome regression model, inverse propensity weighting and stratification based on a propensity function model, and an augmented inverse propensity weighting method that is doubly robust with respect to the two models. We note that an outcome regression model often imposes an implicit constraint on the dose–response curve, and propose a flexible modeling strategy that avoids constraining the dose–response curve. We also propose two new methods: a weighted regression method that combines ordinary regression with inverse propensity weighting and a stratified regression method that combines ordinary regression with stratification. The proposed methods are similar to the augmented inverse propensity weighting method in the sense of double robustness, but easier to implement and more generally applicable. The methods are illustrated with an obstetric example and compared in simulation studies.
Keywords
1 Introduction
Understanding the causal effect of a treatment or exposure is an important objective in many epidemiologic studies. A common framework for understanding causality is Rubin's 1 causal model, and a standard assumption is strongly ignorable treatment assignment. 2 Under this assumption, valid causal inference can be based on an outcome regression model that relates the outcome of interest to the treatment or exposure under investigation as well as observed confounders. This ordinary regression (OR) approach is straightforward to implement and the resulting inference is efficient (under a correctly specified parametric model). Many alternative methods require a model for the propensity score, that is, the conditional probability of being treated or exposed given covariate values. 2 Estimated propensity scores can be used for matching, stratification, or inverse probability weighting.3–6 Much of the recent research has been directed toward doubly robust estimators that involve both an outcome regression model and a propensity score model and that remain consistent if one of the models (but not both) is misspecified.6–18
Most of the existing literature on causal inference is concerned with binary or categorical exposures. In practice, exposures of a quantitative nature are frequently encountered. For example, although smoking is often treated as a binary exposure, information is often available on the amount of smoking (e.g. frequency times duration) for a smoker. As another example, obesity is usually defined by dichotomizing the body mass index (BMI), a continuous variable that is arguably more informative than a dichotomous indicator of obesity. This article gives a overview of the existing methods for estimating a does–response curve that summarizes the effect of a quantitative exposure. The OR approach mentioned earlier can accommodate any type of exposures. We note, however, that an outcome regression model can impose an implicit constraint on the dose–response curve, and propose a flexible modeling strategy that does not constrain the dose–response curve. The propensity score, originally defined for a binary exposure, can be generalized into a propensity function for a quantitative exposure.19,20 Robins et al. 19 show that a marginal structural model (MSM) for a quantitative exposure can be estimated consistently using an inverse propensity weighting (IPW) method, weighting each subject by the inverse of the estimated propensity function. Following Joffe and Rosenbaum, 21 Imai and Van Dyk 20 assume that the propensity function has a certain structure that allows dimension reduction and propose a stratification method that is approximately consistent if the propensity function model is correct and the strata are approximately homogeneous. Each of the aforementioned methods for a quantitative exposure is based on either an outcome regression model or a propensity function model. These models are typically parametric or at least semiparametric due to the curse of dimensionality, raising concerns about potential bias due to model misspecification.
For a parametric MSM, Van Der Laan and Robins 10 develop an augmented IPW (AIPW) method that is doubly robust and locally efficient. The double robustness property of this method appears limited to special models by a compatibility requirement for the MSM and the outcome regression model. Furthermore, it can be challenging to implement the AIPW method, which involves an integral in the estimating function. In this article, we propose two methods that are similar to the AIPW method in the sense of double robustness, but are easier to implement and more generally applicable. Specifically, we propose a weighted regression (WR) method that combines OR with IPW and a stratified regression method that combines OR with stratification. For estimating a suitably parameterized or completely non-parametric MSM, the WR method is doubly robust and the stratified regression method is approximately so (consistent under the outcome regression model and approximately consistent under the propensity function model). Although the stratified regression method is theoretically somewhat inferior to the AIPW and WR methods, it is more protected from very large weights, a common problem with IPW.
The rest of the article is organized as follows. In the next section, we set up the notation and state the key assumptions. We then give an overview of the existing methods in Section 3, and present the proposed methods in Section 4. In Section 5, the methods are illustrated with an obstetric example and compared in simulation experiments. The article ends with a discussion in Section 6. Technical details are provided in Appendix.
2 Notation and assumptions
Let
denote the set of all
possible values of the treatment or exposure of interest. For a binary exposure, it is
customary to take
= {0, 1}.
For a qualitative exposure with more than two levels, one may take
= {0, 1 , … , K} for
some K > 1.21,22 Of
particular interest to us is a quantitative exposure such as dose, for which
is usually an interval. For each
t ∈
, let
Y(t) denote the potential outcome that would realize if
a subject is exposed at level t. Suppose we are primarily interested in
marginal means of the potential outcomes, then the causal effect of the exposure can be
described by μ(t) = E{Y(t)} as a function
of t, often referred to as the dose–response curve.
Of course, the potential outcomes are not observed completely. In reality, we only observe
the actual treatment T and the corresponding outcome
Y = Y(T). In randomized experiments,
T is independent of the Y(t), written
The observed data consist of (X i , T i , Y i ), i = 1 , … , n, which we conceptualize as independent copies of (X, T, Y). Our objective is to estimate μ(t) using these data under assumptions (1) and (2) and appropriate modeling assumptions. We consider both parametric and non-parametric models for μ(t), recognizing that a non-parametric μ(t) cannot be estimated at the parametric rate.
3 Existing methods
3.1 Ordinary regression
The standard approach is to adjust for confounding using an outcome regression model for
E(Y|T, X), say
m(T, X; β), where m
is a known function and β an unknown parameter which is usually finite dimensional but can
be infinite dimensional. Now, equation (3) can be rewritten as
Consider first the generalized linear model (GLM)
Case A. For the identity link, expression (7) holds with α1 = β1 +β X ′E X and α T = β T + β TX ′E X. For convenience, we assume in this case that X is centered (by subtracting E X), so that α = (β1, β T )′.
Case B. For the log link without the interaction term, expression (7) holds with α1 = β1 + log{Eexp(β X ′X)} and α T = β T .
In general, however, expression (6) does not correspond to an MSM in a transparent form. For example, if a logit link is used in model (5), then equation (6) is generally not a logistic regression model. The issue is also known as non-collapsibility. 24
To avoid imposing implicit constraints on μ(t), we also consider the
following generalized additive model (GAM)
Model (5) can
be fitted using standard techniques such as (iteratively reweighted) least squares, and
model (8) using
the local scoring algorithm.
25
The local scoring algorithm consists of two nested loops, the inner
loop being a weighted backfitting procedure to fit an additive model for an adjusted
dependent variable, and the outer loop updating the adjusted dependent variable and the
weight. The algorithm can be implemented with any smoother for estimating
β
T
(e.g. smoothing spline, local regression) as long as it
is consistent in the usual setting. Suppose the appropriate algorithm is chosen, and
denote the resulting estimate by
In the case of model (5), standard asymptotic arguments
26
can be used to show that
3.2 (Augmented) IPW
The IPW approach involves two models: an MSM μ(t; α), which may be parametric or non-parametric, and a propensity function model r(t|x; γ). One possible choice for the MSM is given by expression (7). The propensity function model is just a parametric model for the conditional distribution of T given X. If T is continuous, it is natural to consider a normal linear model or a Box–Cox-transformed linear model 27 and, if necessary, suitable transformations for some or all elements of X. If T follows a mixed distribution, with both discrete and continuous components, it will be necessary to model each component separately.28,29
For a given propensity function model, the parameter γ can be estimated by maximizing the
likelihood
Further insights into the IPW approach can be gained from the following heuristic
argument. Recall that direct (unweighted) fitting of the MSM would be invalid due to
confounding. Weighting each subject by the inverse propensity function has the effect of
creating a pseudo-population that can be shown to be free of confounding. To avoid
confusion, we will use the superscript * to denote random variables in the
pseudo-population. Thus, the joint density of (X*,
T*) is proportional to the joint density of
(X, T) multiplied by the (asymptotic) weight and can
be written as
For a parametric MSM (i.e. finite dimensional α), the IPW method can be improved under an
AIPW approach, which projects an IPW estimating function into the orthogonal complement of
the nuisance tangent spact.10,34 Let
e(Y, T; α) denote an estimating
function for α that would be used in the absence of confounding; this is typically given
by e(Y, T;
α) = {Y − μ(T;
α)}h(T) for a vector-valued function
h of the same dimension as α. The AIPW approach is based on the
following estimating function
3.3 Stratification
Joffe and Rosenbaum
21
and Imai and Van Dyk
20
assume that the propensity function
r(t|x) depends on x
only through u(x) for some function u,
that is
This strategy can be implemented via stratification if the dimension of
u(x) is very low, say one or two. This dimensional
requirement is met for some choices of the propensity function model
r(t|x; γ), under which
u(x) = u(x; γ) may
depend on all or part of γ. For example, if we assume that
(T|X = x) is distributed as
N(x′λ, σ2), then
u(x; λ) = x′λ is one dimensional.
Further examples for categorical and ordinal treatments are discussed by Imai and Van
Dyk.
20
Suppose
the dimension of u(x) is sufficiently low, and let the
range of u(x) be stratified as
k
,
k = 1 , … , K. Experience with propensity score
stratification for a binary treatment suggests that, if
u(x) is one dimensional, it may be reasonable to
define the strata using quintiles of U. If each stratum is reasonably
homogeneous, it makes sense to define
μ
k
(t) = E{Y(t)|U ∈
k
}, the
dose–response relationship in the kth stratum. Write
k
based on
the
3.4 Summary
Other than the AIPW method described in Section 3.2, each of the aforementioned methods is based on either an m-model or an r-model, and its consistency depends on the relevant model being correctly specified. The AIPW method, which involves both models, is doubly robust and locally efficient. However, the double robustness of the AIPW method appears limited to Cases A and B, as mentioned earlier, and is not available for a binary outcome (with a range-approriate link function). Also, it can be challenging to implement the AIPW method, which involves an integral in the estimating function.
4 Proposed methods
We now propose two methods that are similar to the AIPW method in the sense of double robustness, but easier to implement and more generally applicable (beyond Cases A and B). As mentioned earlier, double robustness requires that the m-model be compatible with the MSM in the sense of (4). This requirement is met in Cases A and B with equation (7) as the MSM and the GLM (5) as the m-model. Beyond those cases, one might be tempted to simply assume, as an MSM, that condition (4) holds for some β, even when the m-model is misspecified. However, with a GLM (5) that does not satisfy equation (7), equation (4) as an assumption is not straightforward to interpret. While it may be acceptable to make assumptions about μ(t), it is always important for the investigator to understand what is assumed. To address that interpretability problem, we relax the m-model and work with the GAM (8), which does not constrain μ(t) through equation (4). Thus, beyond Cases A and B, we work with a non-parametric MSM in conjunction with an m-model given by equation (8), which covers a broad range of practical situations such as logistic regression for a binary outcome.
4.1 Weighted regression
Our first proposal is a WR method which is a simple combination of IPW and OR. The idea
is to fit an outcome regression model m(t,
x; β), which may be a GLM given by equation (5) or a GAM given by
equation (8),
using the same IPW as in Section 3.2. Denote by
In the special cases corresponding to equation (7), we obtain
For the GAM (8), the local scoring algorithm will be used with the same IPW as in Section 3.2.
The consistency and convergence properties of the local scoring algorithm have been
studied by Buja et al.
38
for the identity link and seem less well understood for GAMs. Here, we
assume that the local scoring algorithm is consistent when model (8) is correctly
specified and convergent otherwise. Denote by β* the probability limit of
4.2 Stratified regression
We also propose a stratified regression approach that combines stratification with OR.
Under this approach, we specify and estimate a propensity function model
r(t|x; γ) which satisfies equation
(11) for a
low-dimensional u(x), and use the estimates
We recognize that Imai and Van Dyk
20
also consider fitting an
m-model in each stratum. However, their proposal is to summarize the
estimates
The stratified regression method we propose is approximately doubly robust in the sense
that
k
}. It
follows that
Consider first the special cases corresponding to equation (7). For each k, we obtain
k
). It is
easy to see that, in this case,
Beyond these special cases, we propose to fit a GAM like (8) in each stratum using the local scoring
algorithm. With a canonical link, the resulting
Theoretically, this stratified regression method is not as appealing as the AIPW and WR methods. However, the latter methods, like the IPW method, can be numerically unstable when the estimated propensity function is very small for some subjects. The problem is well documented for doubly robust methods involving inverse probability weighting in the literature for binary treatments.6,15 Like the stratification method, the stratified regression method can be expected to be more stable numerically.
5 Numerical results
5.1 An obstetric example
We now use the methods described in Sections 3 and 4 to assess the effect of a mother's pre-pregnancy BMI on the infant's birth weight. The BMI is a heuristic proxy for human body fat based on an individual's weight and height, and is often used to define obesity, a major public health concern in the United States and worldwide. As a risk factor, the BMI is often dichotomized into an indicator of obesity in epidemiologic research. While convenient, such dichotomization results in a loss of information and possibly an oversimplification of the causal effect of interest. The methods described and proposed in this article allow us to make use of all relevant information in the original BMI and better characterize its effect on the birth weight.
Our research question arose from a large obstetric study known as the Consortium on Safe Labor (CSL). 39 The CSL is a retrospective observational study conducted by the National Institutes of Health, Eunice Kennedy Shriver National Institute of Child Health and Human Development, in collaboration with 12 institutions across the United States. The goal was to collect comprehensive information on contemporary labor and delivery practice in the United States population. Participating institutions extracted detailed information from their electronic medical records on maternal demographic characteristics, medical history, reproductive and prenatal history, labor and delivery summary, and postpartum and newborn information.
Our analysis is based on 5194 Caucasian women in the CSL with complete information on the outcome (birth weight in grams), the exposure (BMI in kg/cm2), and all relevant covariates (identified prospectively as maternal age in years, parity, smoking, and diabetes). The outcome regression model we use is a linear model with four linear terms (BMI, maternal age, parity, and smoking) and no interactions. Our propensity function model is also a linear model (assuming normality) which includes maternal age, parity, smoking, and diabetes, as well as interactions of diabetes with maternal age and parity. Both models were constructed using a systematic variable selection procedure, starting with univariate analyses and then considering interactions among the selected covariates. Both models appear to fit the data well.
Analysis of the CSL data: point estimates (PEs), analytic standard errors (ASEs), and bootstrap standard errors (BSEs) based on 1000 bootstrap samples, for α = (α1, α T )′ in a linear MSM: μ(t; α) = α1 + α T t.
CSL: Consortium on safe labor; MSM: Marginal structural model; NV: Naive; OR: Ordinary regression; IPW: Inverse propensity weighting; AIPW: Augmented inverse propensity weighting; S: Stratification; WR: Weighted regression; and SR: Stratified regression. The results are based on the naive method, OR, IPW, AIPW, stratification, WR, and stratified regression.
The results of analyzing a single sample may be arbitrary and inadequate for method comparison; so, we compare the same methods in simulation experiments mimicking the CSL study. Specifically, we focus on the same set of 5194 Caucasian women in the CSL with their original covariate values, and generate values of (T, Y) using the r- and m-models described earlier with parameter values estimated from the previous analyses. The working m- and r-models may or may not be the same as the models for data generation. For both models, misspecification results from omitting an important covariate (smoking). This is motivated by the fact that investigators are primarily concerned about omitting important confounders in analyzing the CSL data. If all such variables are identified and measured, plausible models could be constructed using conventional model-building techniques and serious misspecifications could be detected given the large sample size. In contrast, it is much more difficult, if not impossible, to insure that all important covariates have been considered, due to the finiteness of human knowledge at any point in time.
Simulations based on the CSL example: bias, standard deviation (SD), median (analytic) standard error (SE), and coverage probability (CP) at level 0.95, for estimating α T in a linear MSM: μ(t; α) = α1 + α T t, using the same methods as in Table 1.
CSL: Consortium on safe labor; MSM: Marginal structural model; NV: Naive; OR: Ordinary regression; IPW: Inverse propensity weighting; AIPW: Augmented inverse propensity weighting; S: Stratification; WR: Weighted regression; and SR: Stratified regression. Each entry is based on 1000 replicates.
5.2 Additional simulations
Additional simulations for a continuous outcome: bias, SD, median standard error (ASE for analytic, BSE for bootstrap with 1000 samples) and CP at level 0.95 based on ASE, BSE or BP, for estimating α T in a linear MSM: μ(t; α) = α1 + α T t, using the same methods as in Table 1.
SD: Standard Deviation; ASE: Analytic standard error; BSE: Bootstrap standard error; CP: Coverage probability; BP: Bootstrap percentile; MSM: Marginal structural model; NV: Naive; OR: Ordinary regression; IPW: Inverse propensity weighting; AIPW: Augmented inverse propensity weighting; S: Stratification; WR: Weighted regression; and SR: Stratified regression. Each entry is based on 1000 replicates.
Additional simulations for a binary outcome: bias and SD for estimating μ(t0.25) (response probability at the first quartile of T) and the inter-quartile log-odds ratio ρ (defined in Section 5.2) under a non-parametric MSM, using the same methods as in Table 1 (except AIPW).
SD: Standard Deviation; MSM: Marginal structural model; AIPW: Augmented inverse propensity weighting; NV: Naive; OR: Ordinary regression; IPW: Inverse propensity weighting; S: Stratification; WR: Weighted regression; and SR: Stratified regression. Each entry is based on 1000 replicates.
6 Discussion
We have reviewed the existing methods and proposed two new ones (WR and stratified regression) for causal inference with a quantitative exposure. The WR method is similar to AIPW in the sense of double robustness, but easier to implement (in Case B) and more available (for binary outcomes). The stratified regression method is only approximately doubly robust, but it does have some extra robustness in the sense that valid stratification is possible under an incorrect r-model (Section 5.2). In our simulation experiments (in Case A), the WR method behaves in a manner similar to the AIPW method, while the stratified regression method appears more efficient, especially when the IPW is highly variable. In the latter case, the stratified regression method seems more appealing than the other methods. The stratified regression method does require condition (11) to hold for a low-dimensional function u(x), which is the case in our simulation studies but not in general. For example, if the normal linear regression of T on X is heteroskedastic, it will be necessary to model the heteroskedasticity and include additional terms in u(x) to account for the heteroskedasticity. The situation can be even more difficult if (T|X) follows a non-normal or mixed distribution. In such cases, the WR method may be preferable if the IPW is not too variable.
A practical approach, suggested by a referee, is to perform a sensitivity analysis using all available methods and comparing the results, as we did for the CSL. The inference will be more credible if the results from different methods are similar than if they are dissimilar. In the latter case, one might want to reexamine the working models for their goodness of fit. Even with double robustness, one cannot be too careful in specifying the working m- and r- models. Among other possibilities, loss-based cross-validation 40 , 41 appears to be a promising approach to model selection in this situation.
It should be noted that the sandwich and bootstrap variance estimates do not work well for the methods that involve weighting when the weight is highly variable (Table 3). We recommend the use of BP confidence intervals in this situation.
With focus on prospectively specified models, we have not discussed data-adaptive models such as the loss-based cross-validation approach40,41 developed recently by Van Der Laan and colleagues. The latter approach attempts to choose parametric models based on the data, in such a way that, asymptotically, the model selection procedure works as well as if the true distribution is known. This approach may have profound implications on statistical modeling in general. Targeted maximum likelihood learning 14 is another promising approach to causal inference with a quantitative exposure.
Footnotes
Acknowledgements
The authors thank the two anonymous referees for insightful comments that have greatly improved the manuscript. The views expressed in this article are not necessarily those of the US Food and Drug Administration.
Funding
Zhiwei Zhang and Jun Zhang were supported in part by the Intramural Research Program of the National Institutes of Health, Eunice Kennedy Shriver National Institute of Child Health and Human Development.
