Locally linear attributes of ReLU neural networks

Autor: Ben Sattelberg, Renzo Cavalieri, Michael Kirby, Chris Peterson, Ross Beveridge
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Frontiers in Artificial Intelligence, Vol 6 (2023)
Druh dokumentu: article
ISSN: 2624-8212
DOI: 10.3389/frai.2023.1255192
Popis: A ReLU neural network functions as a continuous piecewise linear map from an input space to an output space. The weights in the neural network determine a partitioning of the input space into convex polytopes, where each polytope is associated with a distinct affine mapping. The structure of this partitioning, together with the affine map attached to each polytope, can be analyzed to investigate the behavior of the associated neural network. We investigate simple problems to build intuition on how these regions act and both how they can potentially be reduced in number and how similar structures occur across different networks. To validate these intuitions, we apply them to networks trained on MNIST to demonstrate similarity between those networks and the potential for them to be reduced in complexity.
Databáze: Directory of Open Access Journals