深度神经网络离散训练极难,甚至超出了经典难题范畴。
On the Hardness of Training Deep Neural Networks Discretely
- 证明深度网络离散参数训练不在NP类中
- 即使固定维度和数据量,深度结构仍使问题不可解
- 为实际训练提供理论边界,适合算法与复杂性研究者
我们研究神经网络训练(NNT):在给定数据集上优化网络参数以最小化损失。尽管对两层网络的连续参数情形(C-NNT)已有广泛理论研究,但深层网络的离散参数情形(D-NNT)——即参数来自有限集合——的复杂性仍知之甚少。本文证明,在标准复杂性假设下,即使维度和数据集大小固定,具有深度架构的D-NNT也不属于NP类,从而超越所有NP完全问题。通过多项式归约,该结论也适用于更结构化的C-NNT。此外,我们为两层网络的D-NNT建立了全面的NP难下界,表明固定维度、数据集大小或隐藏层神经元数量仍无法简化问题。最后,我们给出一个针对固定数据集大小的两层网络的伪多项式算法。
原文摘要 · Abstract (English)
We study neural network training (NNT): optimizing a neural network's parameters to minimize the training loss over a given dataset. NNT has been studied extensively under theoretic lenses, mainly on two-layer networks with linear or ReLU activation functions where the parameters can take any real value (here referred to as continuous NNT (C-NNT)). However, less is known about deeper neural networks, which exhibit substantially stronger capabilities in practice. In addition, the complexity of the discrete variant of the problem (D-NNT in short), in which the parameters are taken from a given finite set of options, has remained less explored despite its theoretical and practical significance. In this work, we show that the hardness of NNT is dramatically affected by the network depth. Specifically, we show that, under standard complexity assumptions, D-NNT is not in the complexity class NP even for instances with fixed dimensions and dataset size, having a deep architecture. This separates D-NNT from any NP-complete problem. Furthermore, using a polynomial reduction we show that the above result also holds for C-NNT, albeit with more structured instances. We complement these results with a comprehensive list of NP-hardness lower bounds for D-NNT on two-layer networks, showing that fixing the number of dimensions, the dataset size, or the number of neurons in the hidden layer leaves the problem challenging. Finally, we obtain a pseudo-polynomial algorithm for D-NNT on a two-layer network with a fixed dataset size.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。