Self-Expanding Neural Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Mitchell, Rupert, Menzenbach, Robin, Kersting, Kristian, Mundt, Martin
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910323954417664
author Mitchell, Rupert
Menzenbach, Robin
Kersting, Kristian
Mundt, Martin
author_facet Mitchell, Rupert
Menzenbach, Robin
Kersting, Kristian
Mundt, Martin
contents The results of training a neural network are heavily dependent on the architecture chosen; and even a modification of only its size, however small, typically involves restarting the training process. In contrast to this, we begin training with a small architecture, only increase its capacity as necessary for the problem, and avoid interfering with previous optimization while doing so. We thereby introduce a natural gradient based approach which intuitively expands both the width and depth of a neural network when this is likely to substantially reduce the hypothetical converged training loss. We prove an upper bound on the ``rate'' at which neurons are added, and a computationally cheap lower bound on the expansion score. We illustrate the benefits of such Self-Expanding Neural Networks with full connectivity and convolutions in both classification and regression problems, including those where the appropriate architecture size is substantially uncertain a priori.
format Preprint
id arxiv_https___arxiv_org_abs_2307_04526
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Self-Expanding Neural Networks
Mitchell, Rupert
Menzenbach, Robin
Kersting, Kristian
Mundt, Martin
Machine Learning
I.2.6
The results of training a neural network are heavily dependent on the architecture chosen; and even a modification of only its size, however small, typically involves restarting the training process. In contrast to this, we begin training with a small architecture, only increase its capacity as necessary for the problem, and avoid interfering with previous optimization while doing so. We thereby introduce a natural gradient based approach which intuitively expands both the width and depth of a neural network when this is likely to substantially reduce the hypothetical converged training loss. We prove an upper bound on the ``rate'' at which neurons are added, and a computationally cheap lower bound on the expansion score. We illustrate the benefits of such Self-Expanding Neural Networks with full connectivity and convolutions in both classification and regression problems, including those where the appropriate architecture size is substantially uncertain a priori.
title Self-Expanding Neural Networks
topic Machine Learning
I.2.6
url https://arxiv.org/abs/2307.04526