Incorporating Auxiliary Variables to Improve the Efficiency of Time-Varying Treatment Effect Estimation

Denote \beta_{1,t}^\star = \Sigma(g_t(H_t))^{-1}\mathbb{E}[Y_{t+1}(A_t-p_t)\tilde g_t(H_t)]\in \mathbb{R}^{d}, where \Sigma(g_t(H_t)) = \mathbb{E}\big[ \tilde g_t(H_t) \tilde g_t(H_t)^\top\big] \in \mathbb{R}^{d\times d}. The difference in the asymptotic variance between the WCLS estimator \hat\beta_{0,t}^{\text{WCLS}} and the unadjusted estimator \hat\beta_{0,t}^{\text{U}} is - p_t (1-p_t) \alpha_{1,t}^{\star\top} \Sigma(g_t(H_t)) \big(\alpha^\star_{1,t}+2(1-2p_t) \beta^\star_{1,t}\big).

Suppose Assumptions 2.1 and 3.1 hold, and the randomization probability p_t is known. Given invertibility and regularity conditions, let (\hat\alpha_{0,t},\hat\alpha_{1,t}, \hat\beta_{0,t}^{\text{L}}, \hat \beta_{1,t}) minimize objective function (?):

1. \hat\beta_{0,t}^{\text{L}} is consistent and asymptotically normal such that \sqrt{N}(\hat\beta_{0,t}^{\text{L}} - \beta_{0,t}^\star) \rightarrow \mathcal{N}(0,Q_t^{-1}\Sigma_t^{\text{L}} Q_t^{-1}), where Q_t, \Sigma_t^{\text{L}} are defined in Appendix ?;

2. \hat\beta_{0,t}^{\text{L}} is at least as efficient as \hat\beta_{0,t}^{\text{U}} described in (?), and the asymptotic efficiency gain is

   \frac{\left(\alpha^\star_{1,t}+(1-2p_t) \beta^\star_{1,t}\right)^\top \Sigma(g_t(H_t)) \left(\alpha^\star_{1,t}+(1-2p_t) \beta^\star_{1,t}\right)}{p_t (1-p_t)} + \beta_{1,t}^{\star\top}\Sigma(g_t(H_t))\beta^\star_{1,t};

3. \hat\beta_{0,t}^{\text{L}} is at least as efficient as \hat\beta_{0,t}^{\text{WCLS}} described in (?), and the asymptotic efficiency gain is \frac{1-3p_t + 3p_t^2}{p_t (1-p_t)} \beta_{1,t}^{\star\top}\Sigma(g_t(H_t))\beta^\star_{1,t}.

Suppose Assumptions 2.1 and 3.1 hold, centering function \{\mu_t(S_t)\}_{t=1}^T satisfies the Orthogonality Condition 3.4, and that the randomization probability p_t(A_t=1 | H_t) is known. Let (\hat\alpha, \hat\beta_0^{\text{A2}}, \hat \beta_1) minimize objective function (?). Under regularity conditions, \hat\beta_0^{\text{A2}} is consistent and asymptotically normal such that \sqrt{N}(\hat\beta_0^{\text{A2}}-\beta_0^\star) \rightarrow \mathcal{N}(0,Q^{-1}\Sigma^{\text{A2}} Q^{-1}), where Q, \Sigma^{\text{A2}} are defined in Appendix ?.

Under the conditions of Lemma 3.6 and the reference treatment probability satisfies \pi_{t+u} = p_{t+u} for all 0 < u \leq \Delta - 1. Assuming \beta_{\bf p,\pi}(t+\Delta-1;s) = f_t(s)^\top \beta_0^\star, and let \hat\beta_0^{\text{A2}} minimize objective function (?). Under regularity conditions, \hat\beta_0^{\text{A2}} is consistent and asymptotically normal such that \sqrt{N}(\hat\beta_0^{\text{A2}}-\beta_0^\star) \rightarrow \mathcal{N}(0,Q^{-1}\Sigma_l^{\text{A2}} Q^{-1}), where Q, \Sigma_l^{\text{A2}} are defined in Appendix ?.

For estimating the \Delta-lag causal excursion effect, A2-WCLS offers the following efficiency properties compared to the WCLS estimator [boruvka2018]:

1. Under the conditions of Lemma 3.6 and Assumptions 4.1 and 4.2, incorporating auxiliary variables via Equation (?) with a known centering function \mu_t(S_t) guarantees an asymptotic efficiency improvement over the WCLS estimator in estimating the moderated causal effect for a given S_t = s, by f_t(s)^\top Q^{-1} \Sigma Q^{-1}f_t(s), where Q, \Sigma are defined in Appendix ?.

2. Under the conditions of Lemma 3.7 and Assumptions 4.1, if \{l_{t+u}(H_{t+u}, A_{t+u})\}_{u=0}^{\Delta-1} are correctly specified, incorporating post-treatment auxiliary variables via Equation (?) achieves semiparametric efficiency for estimating the \Delta-lag moderated causal effect. The derivative of Equation (?) with respect to \beta_0 is the efficient influence function for \beta_0.

If the auxiliary variables Z_t is conditionally mean independent of the causal excursion effect given S_t, then incorporating them via A2-WCLS yields an estimator with the same asymptotic variance as the standard WCLS estimator [boruvka2018].

Under the conditions of Lemma 3.6 or Lemma 3.7, if we incorporate auxiliary variables via A2-WCLS with estimated functions \mu_t(S_t,\hat\theta_{\mu}) and l_{t+u}(A_{t+u}, H_{t+u}; \hat\theta_l), where \hat\Theta is obtained from solving U(\Theta) = 0, then \hat\beta^{\text{A2}}_0 is consistent and asymptotically normal such that \sqrt{N}(\hat\beta_0^{\text{A2}}-\beta_0^\star) \rightarrow \mathcal{N}(0,Q^{-1}\tilde\Sigma^{\text{A2}} Q^{-1}), where Q is as defined in Theorem 4.3, and \tilde\Sigma^{\text{A2}} = \mathbb{E}[\tilde U_{\beta_0}(\hat\beta_0;\hat \Theta, \hat\beta_1)^\top\tilde U_{\beta_0}(\hat\beta_0;\hat\Theta, \hat\beta_1)], in which \tilde U_{\beta_0}(\hat\beta_0;\hat \Theta, \hat\beta_1) is obtained as follows: