English

Self-Expanding Neural Networks

Machine Learning 2024-02-12 v3

Abstract

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.

Keywords

Cite

@article{arxiv.2307.04526,
  title  = {Self-Expanding Neural Networks},
  author = {Rupert Mitchell and Robin Menzenbach and Kristian Kersting and Martin Mundt},
  journal= {arXiv preprint arXiv:2307.04526},
  year   = {2024}
}

Comments

17 pages, 7 figures

R2 v1 2026-06-28T11:25:55.479Z