arXiv reader

Kernel Ridge Riesz Representers: Generalization, Mis-specification, and the Counterfactual Effective Dimension

Rahul Singh

arXiv
2102.11076v4 · PDF · 4 July 2024
Categories
stat.ML, cs.LG, econ.EM, math.ST, stat.TH
Licence
CC BY 4.0 The site shows the text.

Abstract

Kernel balancing weights provide confidence intervals for average treatment effects, based on the idea of balancing covariates for the treated group and untreated group in feature space, often with ridge regularization. Previous works on the classical kernel ridge balancing weights have certain limitations: (i) not articulating generalization error for the balancing weights, (ii) typically requiring correct specification of features, and (iii) justifying Gaussian approximation for only average effects. I interpret kernel balancing weights as kernel ridge Riesz representers (KRRR) and address these limitations via a new characterization of the counterfactual effective dimension. KRRR is an exact generalization of kernel ridge regression and kernel ridge balancing weights. I prove strong properties similar to kernel ridge regression: population L_2 rates controlling generalization error, and a standalone closed form solution that can interpolate. The framework relaxes the stringent assumption that the underlying regression model is correctly specified by the features. It extends Gaussian approximation beyond average effects to heterogeneous effects, justifying confidence sets for causal functions. I use KRRR to quantify uncertainty for heterogeneous treatment effects, by age, of 401(k) eligibility on assets.

Items 75

Assumption

Assumption 1 (Mean square continuity)

3BJP

The functional f\mapsto \mathbb{E}\{m(W,f)\} is linear. There exists some \bar{M}<\infty such that \mathbb{E}\{m(W,f)^2\}\leq \bar{M} \|f\|_2^2 for all f \in L_2.

Example

Example 1 (Heterogeneous policy effects)

2N5T

Here W=X are regressors and \check{m}(w,f)=f\{t(x)\}-f(x), where t is a known mapping that serves as the counterfactual policy of interest. The weighting \ell_h(w)=\textsc{local}\{(x_1-x_1^*)/h\} is a localization around the location X_1=x_1^* (see Appendix ? for details). Suppose that the density ratio is bounded above, i.e. \omega(W)=\textsc{density}\{t(W)\}/\textsc{density}(W)\leq\bar{\omega} almost surely. Suppose that the weighting function at a fixed h is bounded, i.e. |\ell_h(W)|\leq \bar{\ell}_h almost surely. Then \bar{M}_h=2\bar{\ell}_h^2(\bar{\omega}+1).

Example

Example 2 (Heterogeneous treatment effects)

MNKY

Here W=(D,X) concatenates the treatment D\in\{0,1\} and covariates X, and \check{m}(w,f)=f(1,x)-f(0,x). The weighting \ell_h(w)=\textsc{local}\{(x_1-x_1^*)/h\} is a localization around the location X_1=x_1^* (see Appendix ? for details). Suppose that the propensity score is bounded away from zero and one, i.e. \pi_0(X)\in(1-\bar{\pi},\bar{\pi}) almost surely. Suppose that the weighting function at a fixed h is bounded, i.e. |\ell_h(W)|\leq \bar{\ell}_h almost surely. Then \bar{M}_h=2\bar{\ell}_h^2\{\bar{\pi}^{-1}+(1-\bar{\pi})^{-1}\}.

Lemma

Lemma 1 (Riesz representation; Lemma S3.1)

Y2YJ

Suppose Assumption 1 holds and \gamma_0\in \mathcal{G} \subset L_2. Then there exists a Riesz representer \alpha_0\in L_2 such that \mathbb{E}\{m(W,f)\}=\mathbb{E}\{\alpha_0(W)f(W)\} for all f\in \mathcal{G}. Moreover, there exists a unique minimal Riesz representer \alpha_0^{\min}\in closure\{span(\mathcal{G})\} that satisfies this equation.

Uses Assumption 1 · Used by Proof
Assumption

Assumption 2 (Smoothness)

08N4

Assume \alpha_H\in H^c for some c\in[1,2].

Assumption

Assumption 3 (Spectral decay)

05IW

(i) If H is infinite dimensional, the eigenvalues (\eta_j) decay at least polynomially: j^b \cdot \eta_j\leq \bar{B} for j\geq1, where b\in(1,\infty) and \bar{B}>0 is a constant. If H is finite dimensional, write its dimension J as J\leq\bar{B}<\infty and write b=\infty. (ii) If H is infinite dimensional, we may further impose that the eigenvalues decay exactly polynomially: \underline{B} \leq j^b \cdot \eta_j \leq \bar{B} where \underline{B},\bar{B}>0 are constants.

Used by Lemma 6
Lemma

Lemma 2 (Main lemma)

GDPT

Under Assumption 1 and weak regularity conditions (Assumptions 4 and 5 below), the counterfactual effective dimension of \phi^{(m)}(W) is upper bounded by \bar{M} times the effective dimension of \phi(W). See Section ? for details.

Theorem

Theorem 1 (Main theoretical result)

ZF1K

Under Assumptions 1, 2 and 3(i) as well as weak regularity conditions (Assumptions 4 and 5 below), \|\tilde{\alpha}-\alpha_H\|_2^2=O_p\{n^{-bc/(bc+1)}\} using regularization \lambda=n^{-b/(bc+1)}, when b\in(1,\infty), c\in(1,2], and \bar{M} is fixed. This rate is optimal for some choices of m when Assumption 3(ii) also holds. See Section ? for details, including results when b=\infty, c=1, or \bar{M}_h\uparrow \infty.

Corollary

Corollary 1 (Main corollary)

1JF5

Under weak regularity conditions and correct specification of \hat{\gamma} and \hat{\alpha}, perhaps outside of the RKHS, if \bar{M} is fixed then \hat{\theta} converges to \theta_0 at the rate n^{-1/2} and is asymptotically normal. Under mis-specification of \hat{\gamma} or \hat{\alpha}, \hat{\theta} converges to \theta_0, albeit at a slower rate. For pointwise approximations of nonparametric causal functions with \bar{M}_h\uparrow\infty, \hat{\theta} converges to \theta_0 at the rate (nh)^{-1/2} and is asymptotically normal when the bandwidth is h=o(1) and n^{-1/2}h^{-3/2}=o(1). See Section ? and Appendix ?.

Proposition

Proposition 1 (A practical result)

5HR2

The estimator \tilde{\alpha} has a standalone closed form solution that can be computed from k and m, without directly evaluating the feature maps, and that can interpolate. Kernel ridge regression and kernel ridge balancing weights are special cases, and the latter have an equivalent minimax formulation. See Section ? for details.

Definition

Definition 1 (Kernel approximation to Riesz representer)

RG4D

Let \alpha_H\in\operatorname*{\arg\!\min}_{\alpha\in H} \|\alpha-\alpha_0\|_2.

Used by Proof
Assumption

Assumption 5 (RKHS regularity)

PDHZ

The kernels k and k^{(m)} are bounded, i.e. \sup_{w\in\mathcal{W}}\|\phi(w)\|_H\leq \sqrt{\kappa} and \sup_{w\in\mathcal{W}}\|\phi^{(m)}(w)\|_H\leq \sqrt{\kappa^{(m)}}. The feature maps \phi(w) and \phi^{(m)}(w) are measurable. The kernel k^{(m)} is characteristic [sriperumbudur2010relation] in its components that vary.

Lemma

Lemma 3 (Towards a loss for KRRR)

K3AU

Suppose Assumptions 1, 4, and 5 hold. Then we have \alpha_H\in\operatorname*{\arg\!\min}_{\alpha\in H} \mathcal{L}(\alpha) where \mathcal{L}(\alpha)=-2\langle \alpha, \mu^{(m)} \rangle_H+\langle \alpha, T\alpha\rangle_H.

Proof of Lemma 3

To begin, write

\|\alpha-\alpha_0\|^2_2=\|\alpha_0\|_2^2-2\mathbb{E}\{\alpha_0(W)\alpha(W)\}+\|\alpha\|_2^2.

The first term does not depend on \alpha. Kernel boundedness in Assumption 5 implies Bochner integrability [steinwart2008support, Definition A.5.20], which allows us to exchange expectation and RKHS inner product in the second and third terms. By Lemma 1, the definition of counterfactual features, and boundedness of the counterfactual kernel, the second term has

\mathbb{E}\{\alpha_0(W)\alpha(W)\}=\mathbb{E}\{m(W,\alpha)\} =\mathbb{E}\{\langle \alpha,\phi^{(m)}(w) \rangle_H\} =\langle \alpha, \mu^{(m)} \rangle_H.

By the reproducing property and boundedness of the actual kernel, the third term is

\|\alpha\|_2^2=\mathbb{E}\{\alpha(W)^2\}]=\mathbb{E}\{\langle \alpha,\phi(W)\rangle^2_H\} =\mathbb{E}[\langle \alpha,\{\phi(W) \otimes \phi(W)^*\} \alpha \rangle_H] =\langle \alpha, T\alpha\rangle_H.
Definition

Definition 2 (Loss for KRRR)

Y6WW

Let \alpha_{\lambda}=\operatorname*{\arg\!\min}_{\alpha\in H} \mathcal{L}_{\lambda}(\alpha) where \mathcal{L}_{\lambda}(\alpha)=\mathcal{L}(\alpha)+\lambda \|\alpha\|^2_H. The estimator is its empirical analogue: \tilde{\alpha}=\operatorname*{\arg\!\min}_{\alpha\in H}\mathcal{L}^{n}_{\lambda}(\alpha) where \mathcal{L}^{n}_{\lambda}(\alpha)=-2\langle \alpha, \hat{\mu}^{(m)} \rangle_H+\langle \alpha, \hat{T}\alpha\rangle_H+\lambda \|\alpha\|^2_H.

Lemma

Lemma 4 (First order condition)

SA33

Suppose the conditions of Lemma 3 hold. The first order conditions give T\alpha_H=\mu^{(m)}, \alpha_{\lambda}=(T+\lambda)^{-1}\mu^{(m)}, and \tilde{\alpha}=(\hat{T}+\lambda)^{-1}\hat{\mu}^{(m)}.

Proof of Lemma 4

Take derivatives of the various losses in Definitions 1 and 2. See [de2005risk, Proof of Proposition 2] for technical details of differentiation in the RKHS.

Corollary

Corollary 2 (Balance-variance trade-off)

16NN

Suppose the conditions of Lemma 3 hold. Then we have \|\mathbb{E}_n\{\tilde{\alpha}(W)\phi(W)-\phi^{(m)}(W)\}\|_{H}/\|\tilde{\alpha}\|_{H}=\lambda.

Proof of Corollary 2

Rearranging Lemma 4 gives \hat{T}\tilde{\alpha}-\hat{\mu}^{(m)}=-\lambda \tilde{\alpha}. Taking the RKHS norm of both sides, \|\hat{T}\tilde{\alpha}-\hat{\mu}^{(m)}\|_H=\lambda \|\tilde{\alpha}\|_H. Finally, write

\hat{T}\tilde{\alpha} =\mathbb{E}_n[\{\phi(W)\otimes \phi(W)^*\}\tilde{\alpha}] =\mathbb{E}_n\{\phi(W)\langle \phi(W),\tilde{\alpha}\rangle_{H}\} =\mathbb{E}_n\{\phi(W)\tilde{\alpha}(W)\}

and \hat{\mu}^{(m)}=\mathbb{E}_n\{\phi^{(m)}(W)\}.

Uses Lemma 4
Corollary

Corollary 3 (Special case: Kernel ridge regression)

4VLP

\tilde{\alpha}=\tilde{\gamma} when m(w,f)=yf(w).

Used by Proof
Proof of Corollary 3

The standard formula for ridge regression is \tilde{\gamma}=(\hat{T}+\lambda)^{-1}\mathbb{E}_n\{Y\phi(W)\}. Comparing to the formula \tilde{\alpha}=(\hat{T}+\lambda)^{-1}\hat{\mu}^{(m)} in Lemma 4, \mathbb{E}_n\{Y\phi(W)\} equals \hat{\mu}^{(m)}=\mathbb{E}\{\phi^{(m)}(W)\} when m(w,f)=yf(w).

Uses Lemma 4
Definition

Definition 3 (Loss for kernel ridge balancing weights)

G0J6

Let \tilde{\beta}=\operatorname*{\arg\!\min}_{\beta\in\mathbb{R}^n} \tilde{\mathcal{L}}^{n}_{\lambda}(\beta) where \tilde{\mathcal{L}}^{n}_{\lambda}(\beta)=[\sup_{f\in H:\|f\|_H\leq 1}\frac{1}{n}\sum_{i=1}^n \{\beta_i f(W_i)-m(W_i,f)\}]^2+n^{-1}\lambda \beta^\top\beta.

Corollary

Corollary 4 (Special case: Kernel ridge balancing weights; eq. 8)

SYXK

On the training data, \tilde{\alpha}(W_i)=\tilde{\beta}_i. Appendix D.3 of [hirshberg2019kernel] studies \frac{1}{n}\sum_{i=1}^n \{\tilde{\beta}_i-\alpha_0(D_i,X_i)\}^2, i.e. training error.

Proof of Corollary 4

I restate the proof of [bruns2023augmented, p. 5] for completeness. Define

\textsc{imbalance}(\beta)=\sup_{f\in H:\|f\|_H\leq 1}\frac{1}{n}\sum_{i=1}^n \beta_i f(W_i)-m(W_i,f)

so that \tilde{\mathcal{L}}_{\lambda}^n(\beta)=\{\textsc{imbalance}(\beta)\}^2+\lambda n^{-1}\beta^{\top}\beta.

To begin, use the reproducing property to write

\beta_i f(W_i)-m(W_i,f) =\beta_i \langle f,\phi(W_i)\rangle_H-\langle f,\phi^{(m)}(W_i)\rangle_H =\langle f,\beta_i\phi(W_i)-\phi^{(m)}(W_i) \rangle_H.

Since H is self-dual,

\begin{aligned} \textsc{imbalance}(\beta) &=\sup_{f\in H:\|f\|_H\leq 1}\frac{1}{n}\sum_{i=1}^n \langle f,\beta_i\phi(W_i)-\phi^{(m)}(W_i) \rangle_H \\ &= \sup_{f\in H:\|f\|_H\leq 1}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\langle f,\frac{1}{n}\sum_{i=1}^n \{\beta_i\phi(W_i)-\phi^{(m)}(W_i)\} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\rangle_H \\ &=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|\frac{1}{n}\sum_{i=1}^n \{\beta_i\phi(W_i)-\phi^{(m)}(W_i)\}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|_H =\|\Phi^*\beta/n-\hat{\mu}^{(m)}\|_H \end{aligned}

where \Phi^* is the adjoint of the operator \Phi:H\rightarrow \mathbb{R}^n whose ith component is \langle \phi(W_i),\cdot\rangle_H.

Therefore the loss \tilde{\mathcal{L}}_{\lambda}^n(\beta) becomes

\begin{aligned} \|\Phi^*\beta/n-\hat{\mu}^{(m)}\|^2_H+\lambda n^{-1}\beta^{\top}\beta =\frac{1}{n^2}\beta^{\top} \Phi \Phi^*\beta -\frac{2}{n}\beta^{\top}\Phi \hat{\mu}^{(m)} +\|\hat{\mu}^{(m)}\|_H^2+\frac{\lambda}{n}\beta^{\top}\beta. \end{aligned}

Its first order condition yields \tilde{\beta}=(\Phi\Phi^*/n+\lambda)^{-1}\Phi\hat{\mu}^{(m)} whose ith component is \beta_i.

By Lemma 4, \Phi\tilde{\alpha}=\Phi(\Phi^*\Phi/n+\lambda)^{-1}\hat{\mu}^{(m)} whose ith component is \langle \phi(W_i),\tilde{\alpha}\rangle_H=\tilde{\alpha}(W_i). Thus what remains is to show (\Phi\Phi^*/n+\lambda)^{-1}\Phi=\Phi(\Phi^*\Phi/n+\lambda)^{-1}, which follows by expressing \Phi via its singular value decomposition U\Sigma V^*. In particular,

\begin{aligned} (\Phi\Phi^*/n+\lambda)^{-1}\Phi & =U(\Sigma^2n^{-1}+\lambda)^{-1} U^*U\Sigma V^* =U(\Sigma^2n^{-1}+\lambda)^{-1}\Sigma V^* \\ \Phi(\Phi^*\Phi/n+\lambda)^{-1} & =U\Sigma V^*V(\Sigma^2 n^{-1}+\lambda)^{-1} V^* =U\Sigma (\Sigma^2 n^{-1}+\lambda)^{-1} V^* \end{aligned}

and (\Sigma^2n^{-1}+\lambda)^{-1}\Sigma=\Sigma (\Sigma^2 n^{-1}+\lambda)^{-1} by diagonality of each factor.

Uses Lemma 4
Corollary

Corollary 5 (Equivalence in a special case; Proposition 27)

I2OA

When \tilde{\gamma} is kernel ridge regression and observations are reused for \tilde{\gamma}, \tilde{\alpha}, and the causal parameter, \mathbb{E}_n\{m(W,\tilde{\gamma})\}=\mathbb{E}_n\{Y\tilde{\alpha}(W)\}.

Proof of Corollary 5

By the reproducing property, boundedness of the counterfactual kernel, and the well known expression for kernel ridge regression, write

\mathbb{E}_n\{m(W,\tilde{\gamma})\} =\mathbb{E}_n\{\langle \tilde{\gamma},\phi^{(m)}(W)\rangle_H\} =\langle \tilde{\gamma},\hat{\mu}^{(m)}\rangle_H =\langle (\hat{T}+\lambda)^{-1}\mathbb{E}_n\{Y\phi(W)\},\hat{\mu}^{(m)}\rangle_H.

By the reproducing property, boundedness of the actual kernel, and Lemma 4, write

\begin{aligned} \mathbb{E}_n\{Y\tilde{\alpha}(W)\} &=\mathbb{E}_n\{Y\langle \tilde{\alpha},\phi(W)\rangle_H\} =\langle \tilde{\alpha},\mathbb{E}_n\{Y\phi(W)\}\rangle_H\\ &=\langle (\hat{T}+\lambda)^{-1}\hat{\mu}^{(m)},\mathbb{E}_n\{Y\phi(W)\}\rangle_H. \end{aligned}

Finally observe that (\hat{T}+\lambda)^{-1} is self adjoint.

Uses Lemma 4
Corollary

Corollary 6 (Non-equivalence in general)

RSSR

For a generic regression estimator \hat{\gamma}\neq\tilde{\gamma}, we have \mathbb{E}_n\{m(W,\hat{\gamma})\}\neq \mathbb{E}_n\{Y\tilde{\alpha}(W)\}. For any regression estimator \hat{\gamma} (including \tilde{\gamma}), if \hat{\gamma} or \tilde{\alpha} is estimated on a held out sample, then the \mathbb{E}_n\{m(W,\hat{\gamma})\}\neq \mathbb{E}_n\{Y\tilde{\alpha}(W)\}.

Proof of Corollary 6

See the proof of Corollary 5.

Lemma

Lemma 5 (Standalone closed form exists; c.f. Lemma 2)

Y4TL

Suppose the conditions of Lemma 3 hold. There exists some \rho\in\mathbb{R}^{2n} such that \tilde{\alpha}=\Psi^*\rho.

Uses Lemma 3 · Used by Proof
Proof of Lemma 5

The argument is based on [chernozhukov2020adversarial, Lemma 2], which instead finds a solution in \mathbb{R}^{2n} for an adversary that tries to violate Riesz representation.

By Definition 2, the definitions of \hat{\mu}^{(m)} and \hat{T}, and boundedness of the kernels,

\begin{aligned} \mathcal{L}^n_{\lambda}(\alpha) &=-2\langle \alpha, \hat{\mu}^{(m)} \rangle_H+\langle \alpha, \hat{T}\alpha\rangle_H+\lambda \|\alpha\|^2_H\\ &=-2\langle \alpha, \mathbb{E}_n\{\phi^{(m)}(W)\} \rangle_H+\langle \alpha, \mathbb{E}_n\{\phi(W)\otimes \phi(W)^*\}\alpha\rangle_H+\lambda \|\alpha\|^2_H \\ &=\mathbb{E}_n[-2\langle \alpha, \phi^{(m)}(W) \rangle_H+\langle \alpha, \{\phi(W)\otimes \phi(W)^*\}\alpha\rangle_H]+\lambda \|\alpha\|^2_H \\ &=\mathbb{E}_n\{-2\langle \alpha, \phi^{(m)}(W) \rangle_H+\langle \alpha, \phi(W)\rangle_H \langle \alpha, \phi(W)\rangle_H \}+\lambda \|\alpha\|^2_H. \end{aligned}

Due to the ridge penalty, \mathcal{L}^n_{\lambda} is coercive and strongly convex. Hence it has a unique minimizer \tilde{\alpha}. Write \tilde{\alpha}=\tilde{\alpha}_n+\tilde{\alpha}^{\perp}_n where \tilde{\alpha}_n\in row(\Psi) and \tilde{\alpha}_n^{\perp}\in null(\Psi). In words, \tilde{\alpha}^{\perp}_n is orthogonal to the features and counterfactual features evaluated at data. I use this orthogonality and the final expression in the display to conclude that \mathcal{L}^n_{\lambda}(\tilde{\alpha})=\mathcal{L}^n_{\lambda}(\tilde{\alpha}_n)+\lambda \|\tilde{\alpha}_n^{\perp}\|^2_H. Therefore \mathcal{L}^n_{\lambda}(\tilde{\alpha})\geq \mathcal{L}^n_{\lambda}(\tilde{\alpha}_n). Since \tilde{\alpha} is the unique minimizer, \tilde{\alpha}=\tilde{\alpha}_n.

Proposition

Proposition 2 (Standalone closed form for KRRR)

CEYM

Suppose the conditions of Lemma 3 hold. Given observations (W_i) and evaluation location w, as well as the formula m, kernel k, and regularization \lambda, the KRRR closed form is as follows.

  1. Calculate K^{(j)}\in\mathbb{R}^{n\times n} as defined above, for j\in\{1,2,3,4\}.

  2. Calculate \Omega^{2n\times 2n} and u(w),v\in\mathbb{R}^{2n} by

    \Omega=\begin{bmatrix} K^{(1)}K^{(1)} & K^{(1)}K^{(2)}\\ K^{(3)}K^{(1)} & K^{(3)}K^{(2)} \end{bmatrix},\; v=\begin{bmatrix} K^{(2)} \\ K^{(4)} \end{bmatrix} \mathbbm{1}_{n},\; u_i(w)=\begin{cases} k(W_i,w) & \text{ if }i\in \{1,...,n\}\\ k_m(W_i,w)& \text{ if }i\in \{n+1,...,2n\}\end{cases}

    where \mathbbm{1}_{n}\in\mathbb{R}^{n} is a vector of ones, k(w,w')=\langle \phi(w),\phi(w')\rangle_H, and k_m(w,w')=\langle \phi^{(m)}(w),\phi(w')\rangle_H.

  3. Set \tilde{\alpha}(w)=v^{\top}(\Omega+n\lambda K)^{-1} u(w).

Uses Lemma 3
Proof of Proposition 2

Using Definition 2 and Lemma 5, we have

\mathcal{L}^n_{\lambda}(\tilde{\alpha})=-2\langle \tilde{\alpha}, \hat{\mu}^{(m)} \rangle_H+\langle \tilde{\alpha}, \hat{T}\tilde{\alpha}\rangle_H+\lambda \|\alpha\|^2_H=-2\rho^\top \Psi \hat{\mu}^{(m)}+\rho^{\top}\Psi\hat{T}\Psi^*\rho+\lambda \rho^\top \Psi \Psi^* \rho.

The first order condition yields

\hat{\rho}=\{\Psi\hat{T}\Psi^*+\lambda\Psi\Psi^*\}^{-1}\Psi\hat{\mu}^{(m)}=\{\Psi\Phi^*\Phi \Psi^*+n\lambda\Psi\Psi^*\}^{-1}\{n \Psi \hat{\mu}^{(m)}\}.

To evaluate the estimator at a test location w, compute \tilde{\alpha}(w)=\langle \tilde{\alpha}, \phi(w)\rangle_H =\rho^{\top}\Psi\phi(w). Finally, note that \Omega=\Psi \Phi^*\Phi \Psi^*, K=\Psi \Psi^*, v=n\Psi\hat{\mu}^{(m)}, and u(w)=\Psi\phi(w).

Theorem

Theorem 2 (Upper bound for KRRR)

NUBJ

If Assumptions 1, 2, 3(i), 4, and 5 hold with \bar{M} bounded above, then \lim_{\tau\rightarrow \infty} \lim\sup_{n\rightarrow\infty} \sup_{p\in\mathcal{P}(b,c)} \mathbb{P}_{(W_i)\sim p^{n}}\{\mathcal{R}(\tilde{\alpha})>\tau \cdot r_n\}=0, where

r_n=\begin{cases} n^{-1} & \text{ if }\quad b=\infty \text{ and } \lambda=n^{-\frac{1}{2}}; \\ n^{-\frac{bc}{bc+1}} & \text{ if }\quad b\in(1,\infty),\; c\in(1,2]\text{ and } \lambda=n^{-\frac{b}{bc+1}}; \\ \ln^{\frac{b}{b+1}}(n)\cdot n^{-\frac{b}{b+1}} & \text{ if }\quad b\in(1,\infty),\; c=1\text{ and } \lambda=\ln^{\frac{b}{b+1}}(n)\cdot n^{-\frac{b}{b+1}}. \end{cases}
Proof of Theorem 2

I combine the nonasymptotic bound in Proposition 5 with the learning theory bounds of Lemma 11, following [sutherland2017fixing, Section 2]. The absolute constant depends only on the quantities (R,\kappa,\kappa^{(m)},\Upsilon,\Sigma,\underline{B}, \bar{B},b,c). Note that (\Upsilon,\Sigma) introduce dependence on (\|\alpha_H\|_H,\bar{M}). Finally appeal to Lemma 8.

Remark

Remark 1 (Causal functions)

3TAX

When the causal parameter has a formula of the form m_h(w,f)=\ell_h(w)\check{m}(w,f), as in Example 2, \bar{M}_h\uparrow \infty. In such case, I use the estimator \hat{\alpha}_h(W)=\ell_h(W)\tilde{\alpha}(W), where \tilde{\alpha} is KRRR for the formula \check{m}(w,f). When \tilde{\alpha} has the rate r_n, \hat{\alpha}_h has the rate h^{-2}r_n under weak regularity conditions. See Appendix ? for details.

Corollary

Corollary 7 (Special case: Kernel ridge regression; Theorem 1)

7YAI

Suppose the conditions of Theorem 2 hold, replacing Assumption 1 with |Y|\leq \bar{Y} almost surely. Then \lim_{\tau\rightarrow \infty} \lim\sup_{n\rightarrow\infty} \sup_{p\in\mathcal{P}(b,c)} \mathbb{P}_{(W_i)\sim p^{n}}\{\mathcal{R}(\tilde{\gamma})>\tau \cdot r_n\}=0.

Uses Theorem 2, Assumption 1 · Used by Lemma 6
Proof of Corollary 7

By Corollary 3, KRRR nests kernel ridge regression. If |Y|\leq \bar{Y}, then Assumption 1 holds with \bar{M}=\bar{Y}^2. Finally appeal to Theorem 2.

Remark

Remark 2 (Excess risk)

HS94

Some works on kernel ridge regression state results in terms of the excess risk \mathcal{E}(\tilde{\gamma})-\inf_{\gamma\in H}\mathcal{E}(\gamma), where \mathcal{E}(f)=\mathbb{E}[\{Y-f(W)\}^2]. This is equivalent to \mathcal{R}(\tilde{\gamma})=\mathbb{E}[\{\tilde{\gamma}(W)-\gamma_H(W)\}^2|(W_i)_{i\in[n]}] when \gamma_H\in \operatorname*{\arg\!\min}_{\gamma \in H} \mathcal{E}(\gamma) exists, which is assumed in those works and in this one. See e.g. [caponnetto2007optimal, eq. 24] and [fischer2017sobolev, Lemma 12].

Lemma

Lemma 6 (Lower bound; Theorem 2)

37E6

Suppose the conditions of Corollary 7 hold as well as Assumption 3(ii), b\in (1,\infty), and c\in [1,2]. Then \lim_{\tau\rightarrow 0} \lim\inf_{n\rightarrow\infty} \inf_{\hat{\gamma}} \sup_{p\in \mathcal{P}(b,c)}\mathbb{P}_{(W_i)\sim p^{n}}\{\mathcal{R}(\hat{\gamma})>\tau \cdot n^{-\frac{bc}{bc+1}}\}=1.

Corollary

Corollary 8 (Optimality)

F7X0

When b\in(1,\infty) and c\in(1,2], Theorem 2 matches Lemma 6. When b\in(1,\infty) and c=1, they match up to a log factor.

Proof of Corollary 8

The conclusion is immediate.

Definition

Definition 4 (Debiased machine learning)

N9A8

Given a sample (Y_i,W_i)_{i\in[n]}, partition the sample into folds (I_{\ell})_{\ell\in [L]}. Denote by I_{\ell}^c the complement of I_{\ell}.

  1. For each fold \ell, estimate \hat{\gamma}_{\ell} and \hat{\alpha}_{\ell} from observations in I_{\ell}^c.

  2. Estimate \theta_0 as \hat{\theta}=n^{-1}\sum_{\ell=1}^L\sum_{i\in I_{\ell}} [m(W_i,\hat{\gamma}_{\ell})+\hat{\alpha}_{\ell}(W_i)\{Y_i-\hat{\gamma}_{\ell}(W_i)\}].

  3. Estimate \hat{\sigma}^2=n^{-1}\sum_{\ell=1}^L\sum_{i\in I_{\ell}} [m(W_i,\hat{\gamma}_{\ell})+\hat{\alpha}_{\ell}(W_i)\{Y_i-\hat{\gamma}_{\ell}(W_i)\}-\hat{\theta}]^2.

  4. Construct the (1-a) 100% confidence interval as \hat{\theta}\pm c_{a}\hat{\sigma} n^{-1/2}, where c_{a} is the 1-a/2 quantile of the standard Gaussian.

Lemma

Lemma 7 (Inference; Corollary 1)

M51C

Suppose Assumption 1 holds. Suppose the following regularity conditions: \mathbb{E}[\{Y-\gamma_0(W)\}^2 \mid W]\leq \bar{\sigma}^2, \|\alpha_0\|_{\infty}\leq\bar{\alpha}, \|\hat{\alpha}\|_{\infty}\leq\bar{\alpha}', and \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\zeta/\sigma\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)^3+\chi^2\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}n^{-1/2}\rightarrow0. Suppose the following learning rate conditions:

  1. \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\bar{M}^{1/2}+\bar{\alpha}/\sigma+\bar{\alpha}'\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\{\mathcal{R}(\hat{\gamma})+\|\gamma_{\mathcal{G}}-\gamma_0\|_2^2\}^{1/2}=o_p(1); \bar{\sigma}\{\mathcal{R}(\hat{\alpha})+\|\alpha_{\mathcal{A}}-\alpha_0\|_2^2\}^{1/2}=o_p(1);

  2. [n \{\mathcal{R}(\hat{\gamma})+\|\gamma_{\mathcal{G}}-\gamma_0\|_2^2\} \{\mathcal{R}(\hat{\alpha})+\|\alpha_{\mathcal{A}}-\alpha_0\|_2^2\}]^{1/2}/\sigma =o_p(1).

Then \sigma^{-1}n^{1/2}(\hat{\theta}-\theta_0)\leadsto\mathcal{N}(0,1) and \mathbb{P}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\theta_0 \in \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\hat{\theta}\pm c_a\hat{\sigma} n^{-1/2} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}\rightarrow 1-a.

Remark

Remark 3 (Mis-specification of features)

JUXC

Crucially, \gamma_0 does not need to be well specified in the RKHS H. Instead, \|\gamma_{\mathcal{G}}-\gamma_0\|_2 must vanish for some \mathcal{G} that may not be H. Previous works on classical kernel ridge balancing weights, which provide rates similar to n^{-1/2}\sigma for the causal parameter, typically require \gamma_0\in H; see e.g. [hazlett2020kernel, Assumption 2], [wong2018kernel, Assumption 3], [zhao2019covariate, Assumption 2], [kallus2020generalized, Theorem 18(a)], [nie2021quasi, Assumption 3], [hirshberg2019kernel, Assumption 3], [wang2021estimation, Assumption 4] and [mou2023kernel, Theorem 2].

[hirshberg2019augmented, eq. 10] requires \|\hat{\gamma}-\gamma_0\|_H=O_p(1) and \mathbb{E}_n[\{\hat{\gamma}(W)-\gamma_0(W)\}^2]=o_p(1), which is satisfied by correctly specified kernel ridge regression, i.e. \hat{\gamma}=\tilde{\gamma} and \gamma_0\in H [hirshberg2019augmented, Remark 2].

To use Lemma 7 with classical kernel ridge balancing weights, allowing \gamma_0\not \in H, we require a rate \mathcal{R}(\hat{\alpha}), which is the main result of this paper (Theorem 2).

Uses Lemma 7
Remark

Remark 4 (Rate-optimal ridge regularization)

ZCIR

Lemma 7 does not require undersmoothing of the regularization in \hat{\gamma} or \hat{\alpha} in order to prove inference for \hat{\theta}, departing from e.g. [hirshberg2019kernel, mou2023kernel]. Instead, it requires L_2 rates for \hat{\gamma} and \hat{\alpha}, the latter of which Theorem 2 provides. These rates are obtained with rate-optimal ridge regularization when optimality is known, as formalized in Corollary 8. When b<\infty and c=1, optimal ridge regularization (up to log factors) is \lambda=n^{-\frac{b}{b+1}}.

For comparison, when b<\infty, [hirshberg2019kernel, Assumption 3] imposes \lambda \ll n^{-1}, which undersmooths.

The regularization in [hirshberg2019augmented, Theorem 1] is \lambda\asymp n^{-1}, and the authors “generally recommend” \lambda=\bar{\sigma}^2 n^{-1}, which undersmooths. The discussion following [hirshberg2019augmented, Theorem 2] considers alternative choices, e.g. \lambda\asymp n^{-\frac{b}{2}} when b<\infty [hirshberg2019kernel, Lemma 6] which seems to undersmooth for b>1.(Future work may characterize whether rate-optimal ridge regularization of KRRR is compatible with [hirshberg2019augmented, eq. 19]. [hirshberg2019augmented, Appendix B.2] demonstrates rate-optimal regularization for Hölder spaces rather than RKHSs.)

See [bruns2023augmented, Section 7] for a comprehensive and insightful discussion when \hat{\gamma}=\tilde{\gamma} and \hat{\alpha}=\tilde{\alpha}.

Proposition

Proposition 3 (Consistency under mis-specification)

B7RJ

Suppose Assumption 1 holds. Suppose the following regularity conditions: \mathbb{E}[\{Y-\gamma_{\mathcal{G}}(W)\}^2 \mid W]\leq \bar{\sigma}^2, \|\hat{\alpha}\|_{\infty}\leq\bar{\alpha}', and \sigma_{\textsc{mis}} n^{-1/2}\rightarrow0. Suppose the following learning rate conditions: \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\bar{M}^{1/2}+\bar{\alpha}'\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\mathcal{R}(\hat{\gamma})^{1/2}=o_p(1) and \bar{\sigma}\mathcal{R}(\hat{\alpha})^{1/2}=o_p(1). If either \gamma_{\mathcal{G}}=\gamma_0 or \alpha_{\mathcal{A}}=\alpha_0, then \hat{\theta}=\theta_0+o_p(1).

Proof of Proposition 3

Suppose \gamma_{\mathcal{G}}=\gamma_0. Then by the law of iterated expectations,

\theta_{\textsc{mis}}=\mathbb{E} [m(W,\gamma_0)+\alpha_{\mathcal{A}}(W)\{Y-\gamma_0(W)\}]=\mathbb{E}\{m(W,\gamma_0)\}=\theta_0

and \sigma_{\textsc{mis}}^2=var\{\psi(W,\theta_0,\gamma_0,\alpha_{\mathcal{A}})\}.

Suppose instead that \alpha_{\mathcal{A}}=\alpha_0. Then by Riesz representation and the law of iterated expectations,

\begin{aligned} \theta_{\textsc{mis}}&=\mathbb{E} [m(W,\gamma_{\mathcal{G}})+\alpha_0(W)\{Y-\gamma_{\mathcal{G}}(W)\}]=\mathbb{E}\{\alpha_0(W)Y\}\\ &=\mathbb{E}\{\alpha_0(W)\gamma_0(W)\}=\mathbb{E}\{m(W,\gamma_0)\}=\theta_0 \end{aligned}

and \sigma_{\textsc{mis}}^2=var\{\psi(W,\theta_0,\gamma_{\mathcal{G}},\alpha_0)\}.

Finally appeal to Proposition 6.

Lemma

Lemma 8 (generalization error to H norm; c.f. Lemma 12)

X6HZ

Suppose the conditions of Lemma 3 hold. Then \mathcal{R}(\tilde{\alpha})=\|T^{\frac{1}{2}}(\tilde{\alpha}-\alpha_H)\|^2_H.

Uses Lemma 3 · Used by Proof
Proof of Lemma 8

The result follows from Parseval’s identity.

Definition

Definition 5 (Standard learning theory quantities)

O8A3

Define the residual \mathcal{A}(\lambda)=\|T^{1/2}(\alpha_{\lambda}-\alpha_H)\|_H^2, the reconstruction error \mathcal{B}(\lambda)= \|\alpha_{\lambda}-\alpha_H\|^2_H, and the actual effective dimension \mathcal{N}(\lambda)= Tr\{(T+\lambda)^{-1}T\}.

Definition

Definition 6 (New learning theory quantity)

RN6M

Define the counterfactual effective dimension \mathcal{N}^{(m)}(\lambda)=Tr\{(T+\lambda)^{-1}T^{(m)}\}, where T^{(m)}=\mathbb{E}[\phi^{(m)}(W)\otimes \{\phi^{(m)}(W)\}^*] is the counterfactual covariance operator.

Lemma

Lemma 9 (New learning theory bound)

AHL2

Suppose the conditions of Lemma 3 hold. Then \mathcal{N}^{(m)}(\lambda)\leq \bar{M} \cdot \mathcal{N}(\lambda).

Uses Lemma 3 · Used by Proof
Proof of Lemma 9

By Assumption 1, there exists some \bar{M}<\infty such that for all f\in H, \mathbb{E}\{m(W,f)^2\}\leq \bar{M} \mathbb{E}\{f(W)^2\}. By the reproducing property, boundedness of the counterfactual kernel, and the definition of T^{(m)},

\begin{aligned} \mathbb{E}\{m(W,f)^2\}&=\mathbb{E}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\langle f,\phi^{(m)}(W) \rangle^2_H\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}=\mathbb{E}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\langle f,[\phi^{(m)}(W)\otimes \{\phi^{(m)}(W)\}^*]f \rangle_H\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\\ &=\langle f,T^{(m)}f \rangle_H. \end{aligned}

By the reproducing property, boundedness of the actual kernel, and the definition of T,

\mathbb{E}\{f(W)^2\}=\mathbb{E}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\langle f,\phi(W) \rangle^2_H\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}=\mathbb{E}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\langle f,\{\phi(W)\otimes \phi(W)^*\}f \rangle_H\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right]=\langle f,Tf \rangle_H.

In summary, for all f\in H, \langle f,T^{(m)}f \rangle_H \leq \bar{M} \langle f,Tf \rangle_H i.e. \bar{M} T-T^{(m)} \succeq 0 in the sense of Loewner ordering. Therefore by properties of trace,

\bar{M} \mathcal{N}(\lambda)-\mathcal{N}^{(m)}(\lambda)=Tr[(T+\lambda)^{-1}\{\bar{M} T-T^{(m)}\}] \geq 0.
Remark

Remark 5 (Avoiding an auxiliary approximation assumption)

DJGG

Lemma 9 demonstrates, by a simple argument, that a standard assumption in semiparametric theory (Assumption 1) implies that the counterfactual effective dimension (Definition 6) is not much greater than the actual effective dimension (Definition 5). This connection between semiparametric theory and learning theory appears to be new, and it powers the main result.

Previous works that study generalization error, for different estimators of Riesz representers, appear to place auxiliary approximation assumptions to limit the counterfactual effective dimension, e.g. [chernozhukov2018global, Assumption SC(a)], [chernozhukov2018learning, Assumption 2], and [chernozhukov2020adversarial, Assumption 2].

Definition

Definition 7 (High probability events in H)

8EF7

Let \|\cdot\|_{\mathcal{L}_2(H,H)} be the Hilbert-Schmidt norm for operators that map from H to H. Define the events

\begin{aligned} \mathcal{E}_1&=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\|(T+\lambda)^{-1}(\hat{T}-T)\|_{\mathcal{L}_2(H,H)}\leq 2\ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{2\kappa}{\lambda n}+\sqrt{\frac{\kappa\mathcal{N}(\lambda)}{\lambda n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right], \\ \mathcal{E}_2&=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\|(T-\hat{T})(\alpha_{\lambda}-\alpha_H)\|_H\leq 2\ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{2\kappa\sqrt{\mathcal{B}(\lambda)}}{n}+\sqrt{\frac{\kappa\mathcal{A}(\lambda)}{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right], \\ \mathcal{E}_3&=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\|(T+\lambda)^{-\frac{1}{2}}(\hat{\mu}^{(m)}-\hat{T}\alpha_H)\|_H \leq 2\ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{1}{n}\sqrt{\Upsilon^2\frac{\kappa'}{\lambda}}+\sqrt{\frac{\Sigma^2\mathcal{N}(\lambda)}{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right], \end{aligned}

where \Upsilon=2(1+\sqrt{\kappa}\|\alpha_H\|_H), \Sigma^2=2(\bar{M}+\kappa \|\alpha_H\|^2_H), and \kappa'=\max\{\kappa,\kappa^{(m)}\}.

Proposition

Proposition 4 (High probability events in H)

3UVZ

Suppose the conditions of Lemma 3 hold. Then \mathbb{P}(\mathcal{E}_j^c)\leq \delta/3 for each j\in\{1,2,3\}.

Uses Lemma 3 · Used by Proof
Proof of Proposition 4

The argument for \mathcal{E}_1 immediately precedes [caponnetto2007optimal, eq. 41]. The argument for \mathcal{E}_2 is identical to [caponnetto2007optimal, eq. 43]. I prove a new result for \mathcal{E}_3 appealing to Lemmas 9 and 15, the former of which appears to be new.

Define

\xi_i=(T+\lambda)^{-\frac{1}{2}}[\phi^{(m)}(W_i)-\{\phi(W_i)\otimes \phi(W_i)^*\}\alpha_H].

By Lemma 4, \mathbb{E}(\xi_i)=(T+\lambda)^{-\frac{1}{2}}\{\mu^{(m)}-T\alpha_H\}=0. Towards an application of Lemma 15, I analyze \|\xi\|_H and \mathbb{E}(\|\xi\|^2_H).

  1. To bound \|\xi\|_H, write

    \begin{aligned} \|\xi_i\|_H &\leq\|(T+\lambda)^{-\frac{1}{2}}\|_{op}\|\phi^{(m)}(W_i)-\{\phi(W_i)\otimes \phi(W_i)^*\}\alpha_H\|_H \\ &\leq \lambda^{-1/2}\{\sqrt{\kappa^{(m)}}+\kappa\|\alpha_H\|_H\} \end{aligned}

    i.e. \bar{a}=2\lambda^{-1/2}\{\sqrt{\kappa^{(m)}}+\kappa\|\alpha_H\|_H\}.

  2. To bound \mathbb{E}(\|\xi\|^2_H), write

    \xi_i=A-B,\quad A=(T+\lambda)^{-\frac{1}{2}}\phi^{(m)}(W_i),\quad B=(T+\lambda)^{-\frac{1}{2}}\alpha_H(W_i)\phi(W_i)

    where I use the reproducing property to simplify

    \{\phi(W_i)\otimes \phi(W_i)^*\}\alpha_H=\phi(W_i) \langle \phi(W_i),\alpha_H\rangle_H=\alpha_H(W_i)\phi(W_i).

    Hence \|\xi_i\|^2_H\leq 2\|A\|_H^2+2\|B\|_H^2. Focusing on the former term, by properties of trace,

    \begin{aligned} \|A\|_H^2 &=\langle (T+\lambda)^{-\frac{1}{2}}\phi^{(m)}(W_i),(T+\lambda)^{-\frac{1}{2}}\phi^{(m)}(W_i) \rangle_H \\ &=\langle (T+\lambda)^{-1}\phi^{(m)}(W_i),\phi^{(m)}(W_i) \rangle_H \\ &=Tr\{\langle (T+\lambda)^{-1}\phi^{(m)}(W_i),\phi^{(m)}(W_i) \rangle_H\} \\ &=Tr((T+\lambda)^{-1} [\phi^{(m)}(W_i) \otimes \{\phi^{(m)}(W_i)\}^*]). \end{aligned}

    Therefore by linearity of trace, the definition of T^{(m)}, and Lemma 9,

    \mathbb{E}(\|A\|^2_H)=\mathcal{N}^{(m)}(\lambda)\leq \bar{M} \mathcal{N}(\lambda).

    Focusing on the latter term, by properties of trace,

    \begin{aligned} \|B\|_H^2 &=\langle (T+\lambda)^{-\frac{1}{2}}\alpha_H(W_i)\phi(W_i),(T+\lambda)^{-\frac{1}{2}}\alpha_H(W_i)\phi(W_i) \rangle_H \\ &=\langle (T+\lambda)^{-1}\alpha_H(W_i)\phi(W_i),\alpha_H(W_i)\phi(W_i) \rangle_H \\ &=Tr\{\langle (T+\lambda)^{-1}\alpha_H(W_i)\phi(W_i),\alpha_H(W_i)\phi(W_i) \rangle_H\} \\ &=\{\alpha_H(W_i)\}^2 Tr[(T+\lambda)^{-1} \{\phi(W_i) \otimes \phi(W_i)^*\}]. \end{aligned}

    By the reproducing property, Cauchy-Schwartz inequality, and boundedness of the kernel,

    |\alpha_H(W_i)|=\langle \alpha_H,\phi(W_i)\rangle_H \leq \|\alpha_H\|_H\cdot \|\phi(W_i)\|_H\leq \sqrt{\kappa}\|\alpha_H\|_H.

    Therefore by linearity of trace and the definition of T,

    \mathbb{E}(\|B\|_H^2) \leq \kappa \|\alpha_H\|^2_H \mathcal{N}(\lambda).

    In summary,

    \mathbb{E}(\|\xi\|^2_H)\leq 2(\bar{M}+\kappa \|\alpha_H\|^2_H ) \cdot \mathcal{N}(\lambda),

    i.e. \bar{b}=\sqrt{2(\bar{M}+\kappa \|\alpha_H\|^2_H)} \sqrt{\mathcal{N}(\lambda)}.

  3. Finally I appeal to Lemma 15. With probability 1-\delta/3,

    \begin{aligned} &\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|(T+\lambda)^{-\frac{1}{2}}(\hat{\mu}^{(m)}-\hat{T}\alpha_H)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|_H \leq 2 \ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{\bar{a}}{n}+\frac{\bar{b}}{\sqrt{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right) \\ &= 2 \ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{2(\sqrt{\kappa^{(m)}}+\kappa\|\alpha_H\|_H)}{n\sqrt{\lambda}}+\frac{\sqrt{2(\bar{M}+\kappa \|\alpha_H\|^2_H)}\sqrt{\mathcal{N}(\lambda)}}{\sqrt{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\} \\ &\leq2\ln(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{1}{n}\sqrt{\Upsilon^2\frac{\kappa'}{\lambda}}+\sqrt{\frac{\Sigma^2\mathcal{N}(\lambda)}{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}. \end{aligned}
Proposition

Proposition 5 (Nonasymptotic bound)

UK5W

Suppose the conditions of Lemma 3 hold. If n is sufficiently large that n\geq 192 \ln^2(6/\delta) \kappa' \mathcal{N}(\lambda) \lambda^{-1} and \lambda\leq \|T\|_{op}, then with probability 1-\delta, \|T^{1/2}(\tilde{\alpha}-\alpha_H)\|^2_H\leq 96 \ln^2(6/\delta) \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\mathcal{A}(\lambda)+\frac{\kappa^2\mathcal{B}(\lambda)}{n^2\lambda}+\frac{\kappa \mathcal{A}(\lambda)}{ n\lambda}+\frac{\kappa' \Upsilon^2}{n^2\lambda}+\frac{\Sigma^2 \mathcal{N}(\lambda)}{n}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}.

Uses Lemma 3 · Used by Proof
Proof of Proposition 5

I follow the structure of [caponnetto2007optimal, Proof of Theorem 4], appealing to Proposition 4 and the union bound.

  1. By a decomposition that mirrors [caponnetto2007optimal, eq. 36],

    \|T^{1/2}(\tilde{\alpha}-\alpha_H)\|^2_H\leq 3\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\mathcal{A}(\lambda)+\mathcal{S}_1\{\lambda,(W_i)\}+\mathcal{S}_2\{\lambda,(W_i)\}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right].

    The middle and final terms depend on data (W_i), and the work of this proposition is to bound them. They are

    \begin{aligned} \mathcal{S}_1\{\lambda,(W_i)\}&=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}\{\hat{\mu}^{(m)}-\hat{T}\alpha_H\}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|^2_H; \\ \mathcal{S}_2\{\lambda,(W_i)\}&=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}(T-\hat{T})(\alpha_{\lambda}-\alpha_H)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|^2_H. \end{aligned}
  2. Consider the final term. Write

    \mathcal{S}_2\{\lambda,(W_i)\}\leq\|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}\|^2_{op}\|(T-\hat{T})(\alpha_{\lambda}-\alpha_H)\|^2_H.

    By [caponnetto2007optimal, eq. 39], under \mathcal{E}_1, \|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}\|_{op} \leq \lambda^{-1/2}. The event \mathcal{E}_2 directly guarantees a bound on the other factor. In summary, under \mathcal{E}_1 and \mathcal{E}_2,

    \begin{aligned} \mathcal{S}_2\{\lambda,(W_i)\}& \leq \frac{1}{\lambda}\cdot \{2\ln(6/\delta)\}^2\cdot2\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left[\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{2\kappa\sqrt{\mathcal{B}(\lambda)}}{n}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}^2+\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\sqrt{\frac{\kappa\mathcal{A}(\lambda)}{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}^2\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right] \\ &=8\ln^2(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{4\kappa^2\mathcal{B}(\lambda)}{n^2\lambda }+\frac{\kappa\mathcal{A}(\lambda)}{n\lambda}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}. \end{aligned}
  3. Consider the middle term. Write

    \mathcal{S}_1\{\lambda,(W_i)\}\leq \|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}(T+\lambda)^{\frac{1}{2}}\|^2_{op}\|(T+\lambda)^{-\frac{1}{2}}\{\hat{\mu}^{(m)}-\hat{T}\alpha_H\}\|^2_H.

    By [caponnetto2007optimal, eq. 47], under \mathcal{E}_1, \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|T^{\frac{1}{2}}(\hat{T}+\lambda)^{-1}(T+\lambda)^{\frac{1}{2}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|_{op} \leq 2. The event \mathcal{E}_3 directly guarantees a bound on the other factor, and it is designed to resemble [caponnetto2007optimal, eq. 48]. Therefore by [caponnetto2007optimal, eq. 49], under \mathcal{E}_1 and \mathcal{E}_3,

    \mathcal{S}_1\{\lambda,(W_i)\}\leq 32 \ln^2(6/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{\kappa' \Upsilon^2}{n^2\lambda}+\frac{\Sigma^2 \mathcal{N}(\lambda)}{n}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}.
Lemma

Lemma 10 (Characterization of key quantities; Theorem 2)

NIHB

If noise has finite variance then \bar{\sigma}^2<\infty. Suppose bounded moment, heteroscedasticity, density, and derivative conditions of [chernozhukov2021simple, Supplement 2.4] hold. Then the moments satisfy

\zeta_h/\sigma_h \lesssim h^{-1/6},\quad \sigma_h \asymp \bar{M}_h \asymp h^{-1/2},\quad \zeta_h\lesssim h^{-2/3},\quad \chi_h\lesssim h^{-3/4}.

The other key quantities in Lemma 7 satisfy

\bar{M}_h\lesssim h^{-2},\quad \bar{\alpha}_h \lesssim h^{-1},\quad \mathcal{R}(\hat{\alpha}_h)\lesssim h^{-2}\mathcal{R}(\tilde{\alpha}),\quad \Delta_h \lesssim n^{1/2} h^{\mathsf{v}+1/2}

where \mathsf{v} is the order of differentiability of \mathcal{K}.

Used by Corollary 9
Corollary

Corollary 9 (Inference for causal functions; Corollary S1)

CL0U

Suppose the conditions of Lemmas 7 and 10 hold as well as correct specification. As n\rightarrow \infty, suppose the bandwidth satisfies h=o(1) and n^{-1/2}h^{-3/2}=o(1). Suppose the following learning rate conditions hold:

  1. \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(h^{-1}+\bar{\alpha}'\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\mathcal{R}(\hat{\gamma})^{1/2}=o_p(1);

  2. \bar{\sigma}h^{-1}\mathcal{R}(\tilde{\alpha})^{1/2}=o_p(1);

  3. h^{-1/2}\{n \mathcal{R}(\hat{\gamma}) \mathcal{R}(\tilde{\alpha})\}^{1/2} =o_p(1).

Finally suppose the approximation error condition \Delta_h \rightarrow 0. Then \sigma_h^{-1}n^{1/2}(\hat{\theta}^h-\theta^{\lim}_0)\leadsto \mathcal{N}(0,1) and \mathbb{P} \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\theta^{\lim}_0 \in \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\hat{\theta}^h\pm c_a\hat{\sigma} n^{-1/2} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}\rightarrow 1-a.

Lemma

Lemma 11 (Standard learning theory bounds; Proposition 3)

4VRX

Under Assumption 2, \mathcal{A}(\lambda)\leq \lambda^cR and \mathcal{B}(\lambda)\leq \lambda^{c-1}R, where R=\|T^{\frac{1-c}{2}}\alpha_H\|^2_H. Under Assumption 3(i), \mathcal{N}(\lambda)\leq \bar{B}^{\frac{1}{b}}\frac{\pi/b}{\sin(\pi/b)}\lambda^{-\frac{1}{b}} if b<\infty, and \mathcal{N}(\lambda)\leq J if b=\infty .

Used by Proof
Proof of Lemma 11

Both \mathcal{A}(\lambda) and \mathcal{B}(\lambda) are standard. See [sutherland2017fixing, Section 1] for \mathcal{N}(\lambda).

Definition

Definition 8 (Foldwise target and oracle)

KC9X
\begin{aligned} \hat{\theta}_{\ell}&=\mathbb{E}_{\ell} [m(W,\hat{\gamma}_{\ell})+\hat{\alpha}_{\ell}(W)\{Y-\hat{\gamma}_{\ell}(W)\}],\; \bar{\theta}_{\ell}=\mathbb{E}_{\ell} [m(W,\gamma_{\mathcal{G}})+\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\}]. \end{aligned}
Definition

Definition 9 (Overall target and oracle)

IZ7E
\hat{\theta}=\frac{1}{L}\sum_{\ell=1}^L \hat{\theta}_{\ell},\quad \bar{\theta}=\frac{1}{L}\sum_{\ell=1}^L \bar{\theta}_{\ell},\quad \theta_{\textsc{mis}}=\mathbb{E} [m(W,\gamma_{\mathcal{G}})+\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\}].
Lemma

Lemma 12 (Residual expansion)

U6G1

Let u=\hat{\gamma}_{\ell}-\gamma_{\mathcal{G}} and v=\hat{\alpha}_{\ell}-\alpha_{\mathcal{A}}. Then |\hat{\theta}_{\ell}-\bar{\theta}_{\ell}|\leq \sum_{j=1}^3 \Delta_{j{\ell}}^{1/2} where

\begin{aligned} \Delta_{1{\ell}}&=\mathbb{E}_{\ell}\{m(W,u)^2\},\quad \Delta_{2{\ell}}=\mathbb{E}_{\ell}\{u(W)^2\hat{\alpha}_{\ell}(W)^2\},\quad \Delta_{3{\ell}}=\mathbb{E}_{\ell}[v(W)^2\{Y-\gamma_{\mathcal{G}}(W)\}^2]. \end{aligned}
Used by Proof
Proof of Lemma 12

Write the difference within the empirical mean as

\begin{aligned} &m(W,\hat{\gamma}_{\ell})+\hat{\alpha}_{\ell}(W)\{Y-\hat{\gamma}_{\ell}(W)\}-[m(W,\gamma_{\mathcal{G}})+\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\}] \\ &=m(W,\hat{\gamma}_{\ell}-\gamma_{\mathcal{G}}) +\hat{\alpha}_{\ell}(W)\{Y-\hat{\gamma}_{\ell}(W)\} - \hat{\alpha}_{\ell}(W)\{Y-\gamma_{\mathcal{G}}(W)\} \\ &\quad + \hat{\alpha}_{\ell}(W)\{Y-\gamma_{\mathcal{G}}(W)\} -\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\} \\ &=m(W,\hat{\gamma}_{\ell}-\gamma_{\mathcal{G}}) +\hat{\alpha}_{\ell}(W)\{\gamma_{\mathcal{G}}-\hat{\gamma}_{\ell}(W)\} +\{\hat{\alpha}_{\ell}(W)-\alpha_{\mathcal{A}}(W)\}\{Y-\gamma_{\mathcal{G}}(W)\}. \end{aligned}

Averaging over observations in I_{\ell},

\hat{\theta}_{\ell}-\bar{\theta}_{\ell}=\mathbb{E}_{\ell}\{m(W,u)\}+\mathbb{E}_{\ell}\{-u(W)\hat{\alpha}_{\ell}(W)\}+\mathbb{E}_{\ell}[v(W)\{Y-\gamma_{\mathcal{G}}(W)\}].

Finally, apply Jensen’s inequality to each term.

Lemma

Lemma 13 (Residuals; Proposition S9)

1897

Suppose Assumption 1 holds and

\mathbb{E}[\{Y-\gamma_{\mathcal{G}}(W)\} \mid W ]^2\leq \bar{\sigma}^2,\quad \|\hat{\alpha}_{\ell}\|_{\infty}\leq\bar{\alpha}'.

Then with probability 1-\epsilon/(2L), for a fixed \ell \in [L], jointly

\begin{aligned} \Delta_{1\ell}&\leq t_1=\frac{6L}{\epsilon}\bar{M}\mathcal{R}(\hat{\gamma}_{\ell}),\quad \Delta_{2\ell}\leq t_2=\frac{6L}{\epsilon}(\bar{\alpha}')^2\mathcal{R}(\hat{\gamma}_{\ell}),\quad \Delta_{3\ell}\leq t_3=\frac{6L}{\epsilon}\bar{\sigma}^2\mathcal{R}(\hat{\alpha}_{\ell}). \end{aligned}
Definition

Definition 10 (Shorter notation)

FVT8

Define the notation

\psi_{\textsc{mis}}(w)=\psi(w,\theta_0,\gamma_{\mathcal{G}},\alpha_{\mathcal{A}}), \quad \psi(w,\theta,\gamma,\alpha)=m(w,\gamma)+\alpha(w)\{y-\gamma(w)\}-\theta,

where \gamma\mapsto m(w,\gamma) is linear. Let \sigma_{\textsc{mis}}^2=var\{\psi_{\textsc{mis}}(W)\}.

Used by Proof, Proof, Proof
Lemma

Lemma 14 (Oracle concentration)

X20G

Suppose \sigma_{\textsc{mis}}^2<\infty. Then with probability 1-\epsilon/2,

|\bar{\theta}-\theta_{\textsc{mis}}|=\Delta \leq t=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{2}{\epsilon}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)^{1/2}\frac{\sigma_{\textsc{mis}}}{n^{1/2}}.
Proof of Lemma 14

I proceed in steps.

  1. Decomposition. Write

    \begin{aligned} &\bar{\theta}-\theta_{\textsc{mis}} \\ &=\mathbb{E}_n [m(W,\gamma_{\mathcal{G}})+\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\}]-\mathbb{E}[m(W,\gamma_{\mathcal{G}})+\alpha_{\mathcal{A}}(W)\{Y-\gamma_{\mathcal{G}}(W)\}] \\ &=\mathbb{E}_n \{\psi_{\textsc{mis}}(W)\}-\mathbb{E}\{\psi_{\textsc{mis}}(W)\}=\mathbb{E}_n\{\xi(W)\} \end{aligned}

    where \xi(W)=\psi_{\textsc{mis}}(W)-\mathbb{E}\{\psi_{\textsc{mis}}(W)\} is mean zero by construction.

  2. Bounding moments. Since observations are independent and identically distributed,

    \begin{aligned} \mathbb{E}(\Delta^2) &=\mathbb{E} ([\mathbb{E}_n\{\xi(W)\}]^2) =\mathbb{E} \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{1}{n^2} \sum_{i,j\in [n]} \xi(W_i)\xi(W_j)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\} \\ &=\frac{1}{n^2} \sum_{i,j\in [n]} \mathbb{E} \{\xi(W_i)\xi(W_j)\} =\frac{1}{n^2} \sum_{i\in [n]} \mathbb{E} \{\xi(W_i)^2\} =\frac{1}{n}\mathbb{E} \{\xi(W)^2\}=\frac{1}{n}\sigma_{\textsc{mis}}^2. \end{aligned}
  3. Markov inequality implies

    \mathbb{P}(\Delta>t)\leq \frac{\mathbb{E}(\Delta^2)}{t^2}= \frac{\sigma_{\textsc{mis}}^2}{nt^2} =\frac{\epsilon}{2}.

    Solving for t gives the desired result.

Proposition

Proposition 6 (Nonasymptotic bound under mis-specification)

O1PV

Suppose the conditions of Lemmas 13 and 14 hold. Then with probability 1-\epsilon,

|\hat{\theta}-\theta_{\textsc{mis}}|\leq \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{6L}{\epsilon}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)^{1/2}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{ (\bar{M}^{1/2}+\bar{\alpha}')\mathcal{R}(\hat{\gamma}_{\ell})^{1/2} +\bar{\sigma}\mathcal{R}(\hat{\alpha}_{\ell})^{1/2} +\frac{\sigma_{\textsc{mis}}}{n^{1/2}} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}.
Uses Lemma 13, Lemma 14 · Used by Proof
Proof of Proposition 6

I proceed in steps.

  1. Decomposition. By Lemma 12 and the triangle inequality, write

    \begin{aligned} |\hat{\theta}-\theta_{\textsc{mis}}| &\leq |\hat{\theta}-\bar{\theta}|+|\bar{\theta}-\theta_{\textsc{mis}}| \leq\frac{1}{L} \sum_{\ell=1}^L \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left| \hat{\theta}_{\ell}-\bar{\theta}_{\ell}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right|+|\bar{\theta}-\theta_{\textsc{mis}}| \leq \frac{1}{L} \sum_{\ell=1}^L \sum_{j=1}^3 \Delta_{j\ell}^{1/2}+\Delta. \end{aligned}
  2. Union bound.

    Define the event \mathcal{E}=(\Delta \leq t). By Lemma 14, \mathbb{P}(\mathcal{E}^c)\leq \frac{\epsilon}{2}. Define the events

    \mathcal{E}'_{\ell}=(\Delta_{1\ell}\leq t_1, \Delta_{2\ell}\leq t_2, \Delta_{3\ell}\leq t_3),\quad \mathcal{E}'=\cap_{\ell=1}^L \mathcal{E}'_{\ell},\quad (\mathcal{E}')^c=\cup_{\ell=1}^L (\mathcal{E}_{\ell}')^c.

    Hence by the union bound and Lemma 13,

    \mathbb{P}\{(\mathcal{E}')^c\}\leq \sum_{\ell=1}^L \mathbb{P}\{(\mathcal{E}_{\ell}')^c\} \leq L\frac{\epsilon}{2L}=\frac{\epsilon}{2}.
  3. Collecting results. Therefore with probability 1-\epsilon,

    \begin{aligned} &|\hat{\theta}-\theta_{\textsc{mis}}| \leq \frac{1}{L} \sum_{\ell=1}^L \sum_{j=1}^3 \Delta_{j\ell}^{1/2}+\Delta \leq \frac{1}{L} \sum_{\ell=1}^L \sum_{j=1}^3 t_j^{1/2}+t =\sum_{j=1}^3 t_j^{1/2}+t \\ &=\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{6L}{\epsilon}\bar{M}\mathcal{R}(\hat{\gamma}_{\ell})\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}^{1/2} +\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{6L}{\epsilon}(\bar{\alpha}')^2\mathcal{R}(\hat{\gamma}_{\ell})\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}^{1/2} +\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\frac{6L}{\epsilon}\bar{\sigma}^2\mathcal{R}(\hat{\alpha}_{\ell})\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}^{1/2} +\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{2}{\epsilon}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)^{1/2}\frac{\sigma_{\textsc{mis}}}{n^{1/2}} \\ &\leq \mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{6L}{\epsilon}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right)^{1/2}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{ \bar{M}^{1/2}\mathcal{R}(\hat{\gamma}_{\ell})^{1/2} +\bar{\alpha}'\mathcal{R}(\hat{\gamma}_{\ell})^{1/2} +\bar{\sigma}\mathcal{R}(\hat{\alpha}_{\ell})^{1/2} +\frac{\sigma_{\textsc{mis}}}{n^{1/2}} \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}. \end{aligned}
Lemma

Lemma 15 (Concentration; Proposition 2)

8LCN

Let (\Omega,\mathcal{F},p) be a probability space. Let \xi be a random variable on \Omega taking value in a real separable Hilbert space \mathcal{H}. Assume there exist \bar{a},\bar{b}>0 such that \|\xi(\omega)\|_{\mathcal{H}}\leq \frac{\bar{a}}{2} and \mathbb{E}(\|\xi\|^2_{\mathcal{H}})\leq \bar{b}^2. Then for all n\in\mathbb{N} and \delta\in(0,1),

\mathbb{P}_{(\omega_i)\sim p^{n}}\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\{\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left\|\frac{1}{n}\sum_{i=1}^n \xi(\omega_i)-\mathbb{E}(\xi)\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\|_{\mathcal{H}}\leq 2 \ln(2/\delta)\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\mathopen{}\mathclose\bgroup\left(\frac{\bar{a}}{n}+\frac{\bar{b}}{\sqrt{n}}\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right) \aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\aftergroup\egroup\right\}\geq 1-\delta.
Used by Proof

Source and licence

The text on this page is from arXiv:2102.11076v4 by Rahul Singh, under the licence CC BY 4.0. The corpus made these changes: it read the items out of the LaTeX source, it expanded the macros of the paper, it put each proof after its statement, and it converted the text to HTML. The formulas are the TeX of the paper, and KaTeX renders them. A formula or a reference can look different from the PDF. The page is published under CC BY 4.0.

The record of each item is in analysis/objects/2102/2102.11076.jsonl.