We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: On Infinite-Width Hypernetworks

Abstract: {\em Hypernetworks} are architectures that produce the weights of a task-specific {\em primary network}. A notable application of hypernetworks in the recent literature involves learning to output functional representations. In these scenarios, the hypernetwork learns a representation corresponding to the weights of a shallow MLP, which typically encodes shape or image information. While such representations have seen considerable success in practice, they remain lacking in the theoretical guarantees in the wide regime of the standard architectures. In this work, we study wide over-parameterized hypernetworks. We show that unlike typical architectures, infinitely wide hypernetworks do not guarantee convergence to a global minima under gradient descent. We further show that convexity can be achieved by increasing the dimensionality of the hypernetwork's output, to represent wide MLPs. In the dually infinite-width regime, we identify the functional priors of these architectures by deriving their corresponding GP and NTK kernels, the latter of which we refer to as the {\em hyperkernel}. As part of this study, we make a mathematical contribution by deriving tight bounds on high order Taylor expansion terms of standard fully connected ReLU networks.
Comments: The first two authors contributed equally
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2003.12193 [cs.LG]
  (or arXiv:2003.12193v7 [cs.LG] for this version)

Submission history

From: Etai Littwin [view email]
[v1] Fri, 27 Mar 2020 00:50:29 GMT (621kb)
[v2] Mon, 30 Mar 2020 07:49:39 GMT (38kb)
[v3] Sun, 5 Apr 2020 21:08:59 GMT (39kb)
[v4] Tue, 30 Jun 2020 09:08:52 GMT (2568kb,D)
[v5] Sun, 1 Nov 2020 15:18:31 GMT (3293kb,D)
[v6] Tue, 3 Nov 2020 08:00:21 GMT (3295kb,D)
[v7] Mon, 22 Feb 2021 23:10:56 GMT (4449kb,D)

Link back to: arXiv, form interface, contact.