Training Neural Networks is NP-Hard in Fixed Dimension

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Froese, Vincent, Hertrich, Christoph
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914644437762048
author Froese, Vincent
Hertrich, Christoph
author_facet Froese, Vincent
Hertrich, Christoph
contents We study the parameterized complexity of training two-layer neural networks with respect to the dimension of the input data and the number of hidden neurons, considering ReLU and linear threshold activation functions. Albeit the computational complexity of these problems has been studied numerous times in recent years, several questions are still open. We answer questions by Arora et al. [ICLR '18] and Khalife and Basu [IPCO '22] showing that both problems are NP-hard for two dimensions, which excludes any polynomial-time algorithm for constant dimension. We also answer a question by Froese et al. [JAIR '22] proving W[1]-hardness for four ReLUs (or two linear threshold neurons) with zero training error. Finally, in the ReLU case, we show fixed-parameter tractability for the combined parameter number of dimensions and number of ReLUs if the network is assumed to compute a convex map. Our results settle the complexity status regarding these parameters almost completely.
format Preprint
id arxiv_https___arxiv_org_abs_2303_17045
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Training Neural Networks is NP-Hard in Fixed Dimension
Froese, Vincent
Hertrich, Christoph
Computational Complexity
Data Structures and Algorithms
Machine Learning
Neural and Evolutionary Computing
We study the parameterized complexity of training two-layer neural networks with respect to the dimension of the input data and the number of hidden neurons, considering ReLU and linear threshold activation functions. Albeit the computational complexity of these problems has been studied numerous times in recent years, several questions are still open. We answer questions by Arora et al. [ICLR '18] and Khalife and Basu [IPCO '22] showing that both problems are NP-hard for two dimensions, which excludes any polynomial-time algorithm for constant dimension. We also answer a question by Froese et al. [JAIR '22] proving W[1]-hardness for four ReLUs (or two linear threshold neurons) with zero training error. Finally, in the ReLU case, we show fixed-parameter tractability for the combined parameter number of dimensions and number of ReLUs if the network is assumed to compute a convex map. Our results settle the complexity status regarding these parameters almost completely.
title Training Neural Networks is NP-Hard in Fixed Dimension
topic Computational Complexity
Data Structures and Algorithms
Machine Learning
Neural and Evolutionary Computing
url https://arxiv.org/abs/2303.17045