Assumption 1 (Mean square continuity)
3BJPThe 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.
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.
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.
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).
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}\}.
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.
Assume \alpha_H\in H^c for some c\in[1,2].
(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.
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.
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 ?.
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.
Let \alpha_H\in\operatorname*{\arg\!\min}_{\alpha\in H} \|\alpha-\alpha_0\|_2.
W takes values in a Polish space \mathcal{W}, i.e. a separable and completely metrizable topological space. Y takes values in \mathcal{Y}\subset \mathbb{R}.
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.
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.
To begin, write
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
By the reproducing property and boundedness of the actual kernel, the third term is
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.
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)}.
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.
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
and \hat{\mu}^{(m)}=\mathbb{E}_n\{\phi^{(m)}(W)\}.
\tilde{\alpha}=\tilde{\gamma} when m(w,f)=yf(w).
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).
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.
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.
I restate the proof of [bruns2023augmented, p. 5] for completeness. Define
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
Since H is self-dual,
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
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,
and (\Sigma^2n^{-1}+\lambda)^{-1}\Sigma=\Sigma (\Sigma^2 n^{-1}+\lambda)^{-1} by diagonality of each factor.
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)\}.
By the reproducing property, boundedness of the counterfactual kernel, and the well known expression for kernel ridge regression, write
By the reproducing property, boundedness of the actual kernel, and Lemma 4, write
Finally observe that (\hat{T}+\lambda)^{-1} is self adjoint.
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)\}.
See the proof of Corollary 5.
Suppose the conditions of Lemma 3 hold. There exists some \rho\in\mathbb{R}^{2n} such that \tilde{\alpha}=\Psi^*\rho.
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,
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.
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.
Calculate K^{(j)}\in\mathbb{R}^{n\times n} as defined above, for j\in\{1,2,3,4\}.
Calculate \Omega^{2n\times 2n} and u(w),v\in\mathbb{R}^{2n} by
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.
Set \tilde{\alpha}(w)=v^{\top}(\Omega+n\lambda K)^{-1} u(w).
Using Definition 2 and Lemma 5, we have
The first order condition yields
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).
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
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.
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.
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.
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].
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.
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.
The conclusion is immediate.
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}.
For each fold \ell, estimate \hat{\gamma}_{\ell} and \hat{\alpha}_{\ell} from observations in I_{\ell}^c.
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)\}].
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.
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.
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:
\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);
[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.
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).
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}.
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).
Suppose \gamma_{\mathcal{G}}=\gamma_0. Then by the law of iterated expectations,
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,
and \sigma_{\textsc{mis}}^2=var\{\psi(W,\theta_0,\gamma_{\mathcal{G}},\alpha_0)\}.
Finally appeal to Proposition 6.
Suppose the conditions of Lemma 3 hold. Then \mathcal{R}(\tilde{\alpha})=\|T^{\frac{1}{2}}(\tilde{\alpha}-\alpha_H)\|^2_H.
The result follows from Parseval’s identity.
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\}.
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.
Suppose the conditions of Lemma 3 hold. Then \mathcal{N}^{(m)}(\lambda)\leq \bar{M} \cdot \mathcal{N}(\lambda).
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)},
By the reproducing property, boundedness of the actual kernel, and the definition of T,
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,
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].
Let \|\cdot\|_{\mathcal{L}_2(H,H)} be the Hilbert-Schmidt norm for operators that map from H to H. Define the events
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)}\}.
Suppose the conditions of Lemma 3 hold. Then \mathbb{P}(\mathcal{E}_j^c)\leq \delta/3 for each j\in\{1,2,3\}.
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
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).
To bound \|\xi\|_H, write
i.e. \bar{a}=2\lambda^{-1/2}\{\sqrt{\kappa^{(m)}}+\kappa\|\alpha_H\|_H\}.
To bound \mathbb{E}(\|\xi\|^2_H), write
where I use the reproducing property to simplify
Hence \|\xi_i\|^2_H\leq 2\|A\|_H^2+2\|B\|_H^2. Focusing on the former term, by properties of trace,
Therefore by linearity of trace, the definition of T^{(m)}, and Lemma 9,
Focusing on the latter term, by properties of trace,
By the reproducing property, Cauchy-Schwartz inequality, and boundedness of the kernel,
Therefore by linearity of trace and the definition of T,
In summary,
i.e. \bar{b}=\sqrt{2(\bar{M}+\kappa \|\alpha_H\|^2_H)} \sqrt{\mathcal{N}(\lambda)}.
Finally I appeal to Lemma 15. With probability 1-\delta/3,
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\}.
I follow the structure of [caponnetto2007optimal, Proof of Theorem 4], appealing to Proposition 4 and the union bound.
By a decomposition that mirrors [caponnetto2007optimal, eq. 36],
The middle and final terms depend on data (W_i), and the work of this proposition is to bound them. They are
Consider the final term. Write
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,
Consider the middle term. Write
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,
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
The other key quantities in Lemma 7 satisfy
where \mathsf{v} is the order of differentiability of \mathcal{K}.
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:
\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);
\bar{\sigma}h^{-1}\mathcal{R}(\tilde{\alpha})^{1/2}=o_p(1);
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.
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 .
Both \mathcal{A}(\lambda) and \mathcal{B}(\lambda) are standard. See [sutherland2017fixing, Section 1] for \mathcal{N}(\lambda).
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
Write the difference within the empirical mean as
Averaging over observations in I_{\ell},
Finally, apply Jensen’s inequality to each term.
Suppose Assumption 1 holds and
Then with probability 1-\epsilon/(2L), for a fixed \ell \in [L], jointly
Define the notation
where \gamma\mapsto m(w,\gamma) is linear. Let \sigma_{\textsc{mis}}^2=var\{\psi_{\textsc{mis}}(W)\}.
Suppose \sigma_{\textsc{mis}}^2<\infty. Then with probability 1-\epsilon/2,
I proceed in steps.
Decomposition. Write
where \xi(W)=\psi_{\textsc{mis}}(W)-\mathbb{E}\{\psi_{\textsc{mis}}(W)\} is mean zero by construction.
Bounding moments. Since observations are independent and identically distributed,
Markov inequality implies
Solving for t gives the desired result.
Suppose the conditions of Lemmas 13 and 14 hold. Then with probability 1-\epsilon,
I proceed in steps.
Decomposition. By Lemma 12 and the triangle inequality, write
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
Hence by the union bound and Lemma 13,
Collecting results. Therefore with probability 1-\epsilon,
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),
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.