Reverse Engineering the Neural Tangent Kernel

Autor:	Simon, James B., Anand, Sajant, DeWeese, Michael R.
Rok vydání:	2021
Předmět:	Computer Science - Machine Learning
Druh dokumentu:	Working Paper
Popis:	The development of methods to guide the design of neural networks is an important open challenge for deep learning theory. As a paradigm for principled neural architecture design, we propose the translation of high-performing kernels, which are better-understood and amenable to first-principles design, into equivalent network architectures, which have superior efficiency, flexibility, and feature learning. To this end, we constructively prove that, with just an appropriate choice of activation function, any positive-semidefinite dot-product kernel can be realized as either the NNGP or neural tangent kernel of a fully-connected neural network with only one hidden layer. We verify our construction numerically and demonstrate its utility as a design tool for finite fully-connected networks in several experiments. Comment: 15 pages, 5 figures
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2106.03186 Zobrazit plný text záznamu View this record from Arxiv