93128bdb2313d5df.tex
1: \begin{proof}
2: Conditioning on $x^l$ and taking expectation with respect to the weights in the $(l+1)$-layer,  
3: \begin{align}\label{eq_cov}
4: \Sigma^{l+1}_{\alpha, \alpha'}= \mathbb E\,\left[ h^{l+1}_j(\alpha)h^{l+1}_j(\alpha')\middle|x^l \right] 
5: = \frac{\sigma^2_\omega}{2k+1}\sum_{\beta\in\ker} \frac 1 {c}\sum_{i=1}^c (x^{l}_i(\alpha+\beta)x^{l}_i(\alpha'+\beta)) + \sigma_b^2.
6: \end{align}
7: We used the fact that 
8: \begin{align*}
9: \mathbb E\, \left[\omega_{ij}^{l+1}(\beta) \omega_{i'j'}^{l+1}(\beta')\right] = 
10: \begin{cases}
11: \frac{\sigma_\omega^2}{c(2k+1)}, \quad &{\rm if }\quad (i, j, \beta) = (i', j', \beta')
12: \\
13: \\
14: 0, \quad &otherwise. 
15: \end{cases}
16: \end{align*} 
17: For any fixed pair $(\alpha, \alpha')$, $\{x_i(\alpha)x_i(\alpha')\}_{i}$ are i.i.d., the law of large number gives 
18: \begin{align}\label{eq_lim}
19: \lim_{c\to \infty}\frac 1 {c}\sum_{i=1}^c (x^{l}_i(\alpha)x^{l}_i(\alpha'))  = 
20: \mathbb E\,\left[x^{l}_i(\alpha)x^{l}_i(\alpha') \right] 
21: \end{align}
22: Thus by \eq.~\ref{eq:cmap},  eq.~\eqref{eq_cov} can be written as 
23: \begin{align}\label{eq_cov2}
24: \Sigma^{l+1}_{\alpha, \alpha'}
25: = \frac{1}{2k+1}\sum_{\beta\in\ker} \mathcal C(\bm \Sigma^l)_{\alpha+\beta, \alpha'+\beta} .
26: \end{align}
27: Let 
28: \begin{align*}
29: \mathcal C(q^l_{\alpha, \alpha'}) = \sigma_\omega^2 \mathbb E\, \left[\phi(z_1)\phi(z_2)\right] + \sigma_b^2 
30: \end{align*}
31: with $(z_1, z_2)^T$ drawn from a mean zero Gaussian with covariance matrix 
32: $[ [q^l_{\alpha,\alpha}, q^l_{\alpha,\alpha'}], [q^l_{\alpha,\alpha'}, q^l_{\alpha',\alpha'}]]$\footnote{Rigorously, ${\mathcal C}$ is a function of $q^l_{\alpha,\alpha}, q^l_{\alpha,\alpha'}$ and $ q^l_{\alpha',\alpha'}$.}. 
33: Applying \eqref{eq_lim} to \eqref{eq_cov} and letting $c\to\infty$ give 
34: \begin{align*}
35:     q^{l+1}_{\alpha, \alpha'} = \frac 1 {2k+1 }\sum_{\beta=-k}^k \mathcal C(q_{\alpha+\beta, \alpha'+\beta}) = \mathcal A \circ \mathcal C(q_{\alpha, \alpha'}).
36: \end{align*}
37: \end{proof}