On the Local Complexity of Linear Regions in Deep ReLU Networks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Patel, Niket, Montufar, Guido
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912480244006912
author Patel, Niket
Montufar, Guido
author_facet Patel, Niket
Montufar, Guido
contents We define the local complexity of a neural network with continuous piecewise linear activations as a measure of the density of linear regions over an input data distribution. We show theoretically that ReLU networks that learn low-dimensional feature representations have a lower local complexity. This allows us to connect recent empirical observations on feature learning at the level of the weight matrices with concrete properties of the learned functions. In particular, we show that the local complexity serves as an upper bound on the total variation of the function over the input data distribution and thus that feature learning can be related to adversarial robustness. Lastly, we consider how optimization drives ReLU networks towards solutions with lower local complexity. Overall, this work contributes a theoretical framework towards relating geometric properties of ReLU networks to different aspects of learning such as feature learning and representation cost.
format Preprint
id arxiv_https___arxiv_org_abs_2412_18283
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On the Local Complexity of Linear Regions in Deep ReLU Networks
Patel, Niket
Montufar, Guido
Machine Learning
We define the local complexity of a neural network with continuous piecewise linear activations as a measure of the density of linear regions over an input data distribution. We show theoretically that ReLU networks that learn low-dimensional feature representations have a lower local complexity. This allows us to connect recent empirical observations on feature learning at the level of the weight matrices with concrete properties of the learned functions. In particular, we show that the local complexity serves as an upper bound on the total variation of the function over the input data distribution and thus that feature learning can be related to adversarial robustness. Lastly, we consider how optimization drives ReLU networks towards solutions with lower local complexity. Overall, this work contributes a theoretical framework towards relating geometric properties of ReLU networks to different aspects of learning such as feature learning and representation cost.
title On the Local Complexity of Linear Regions in Deep ReLU Networks
topic Machine Learning
url https://arxiv.org/abs/2412.18283