Training Neural Networks as Learning Data-adaptive Kernels: Provable Representation and Approximation Benefits
Xialiang Dou,Tengyuan Liang
DOI: https://doi.org/10.1080/01621459.2020.1745812
IF: 4.369
2020-04-23
Journal of the American Statistical Association
Abstract:Consider the problem: given the data pair <span class="NLM_disp-formula inline-formula"><math>(x,y)</math></span> drawn from a population with <span class="NLM_disp-formula inline-formula"><math>f*(x)=E[y|x=x]</math></span>, specify a neural network model and run gradient flow on the weights over time until reaching any stationarity. How does <i>f<sub>t</sub></i>, the function computed by the neural network at time <i>t</i>, relate to <span class="NLM_disp-formula inline-formula"><math>f*</math></span>, in terms of approximation and representation? What are the provable benefits of the adaptive representation by neural networks compared to the prespecified fixed basis representation in the classical nonparametric literature? We answer the above questions via a dynamic reproducing kernel Hilbert space (RKHS) approach indexed by the training process of neural networks. First, we show that when reaching any local stationarity, gradient flow learns an adaptive RKHS representation and performs the global least-squares projection onto the adaptive RKHS, simultaneously. Second, we prove that as the RKHS is data-adaptive and task-specific, the residual for <span class="NLM_disp-formula inline-formula"><math>f*</math></span> lies in a subspace that is potentially much smaller than the orthogonal complement of the RKHS. The result formalizes the representation and approximation benefits of neural networks. Finally, we show that the neural network function computed by gradient flow converges to the kernel ridgeless regression with an adaptive kernel, in the limit of vanishing regularization. The adaptive kernel viewpoint provides new angles of studying the approximation, representation, generalization, and optimization advantages of neural networks. <a class="ext-link" href="https://doi.org/10.1080/01621459.2020.1745812">Supplementary materials</a> for this article are available online.
statistics & probability