Parallel Algorithms for Exact Enumeration of Deep Neural Network Activation Regions

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Drammis, Sabrina, Zheng, Bowen, Srinivasan, Karthik, Berwick, Robert C., Lynch, Nancy A., Ajemian, Robert
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914699376852992
author Drammis, Sabrina
Zheng, Bowen
Srinivasan, Karthik
Berwick, Robert C.
Lynch, Nancy A.
Ajemian, Robert
author_facet Drammis, Sabrina
Zheng, Bowen
Srinivasan, Karthik
Berwick, Robert C.
Lynch, Nancy A.
Ajemian, Robert
contents A feedforward neural network using rectified linear units constructs a mapping from inputs to outputs by partitioning its input space into a set of convex regions where points within a region share a single affine transformation. In order to understand how neural networks work, when and why they fail, and how they compare to biological intelligence, we need to understand the organization and formation of these regions. Step one is to design and implement algorithms for exact region enumeration in networks beyond toy examples. In this work, we present parallel algorithms for exact enumeration in deep (and shallow) neural networks. Our work has three main contributions: (1) we present a novel algorithm framework and parallel algorithms for region enumeration; (2) we implement one of our algorithms on a variety of network architectures and experimentally show how the number of regions dictates runtime; and (3) we show, using our algorithm's output, how the dimension of a region's affine transformation impacts further partitioning of the region by deeper layers. To our knowledge, we run our implemented algorithm on networks larger than all of the networks used in the existing region enumeration literature. Further, we experimentally demonstrate the importance of parallelism for region enumeration of any reasonably sized network.
format Preprint
id arxiv_https___arxiv_org_abs_2403_00860
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Parallel Algorithms for Exact Enumeration of Deep Neural Network Activation Regions
Drammis, Sabrina
Zheng, Bowen
Srinivasan, Karthik
Berwick, Robert C.
Lynch, Nancy A.
Ajemian, Robert
Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
A feedforward neural network using rectified linear units constructs a mapping from inputs to outputs by partitioning its input space into a set of convex regions where points within a region share a single affine transformation. In order to understand how neural networks work, when and why they fail, and how they compare to biological intelligence, we need to understand the organization and formation of these regions. Step one is to design and implement algorithms for exact region enumeration in networks beyond toy examples. In this work, we present parallel algorithms for exact enumeration in deep (and shallow) neural networks. Our work has three main contributions: (1) we present a novel algorithm framework and parallel algorithms for region enumeration; (2) we implement one of our algorithms on a variety of network architectures and experimentally show how the number of regions dictates runtime; and (3) we show, using our algorithm's output, how the dimension of a region's affine transformation impacts further partitioning of the region by deeper layers. To our knowledge, we run our implemented algorithm on networks larger than all of the networks used in the existing region enumeration literature. Further, we experimentally demonstrate the importance of parallelism for region enumeration of any reasonably sized network.
title Parallel Algorithms for Exact Enumeration of Deep Neural Network Activation Regions
topic Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
url https://arxiv.org/abs/2403.00860