突破神经网络训练的可计算性边界,找到两类新可解架构。
New Complexity-Theoretic Frontiers of Tractability for Neural Network Training
- 针对ReLU网络,证明出度为1的隐藏层结构可多项式时间求解。
- 首次发现线性激活网络在数据吞吐量条件下可多项式时间优化。
- 为复杂度理论下的训练可解性开辟新方向,适合理论研究者。
尽管神经网络在现代机器学习中具有基础性作用,但我们对最优训练神经网络的计算复杂性理解仍不完整,即使在最简单的激活函数情况下亦然。近年来已有研究不断收紧线性和ReLU激活函数下该问题的下界,但在识别新的多项式时间可解网络架构方面进展有限。本文首次获得训练线性和ReLU激活神经网络至最优的算法上界,将可解性边界推向新的前沿。对于ReLU网络,我们证明了所有隐藏神经元出度为1的架构均可多项式时间求解,优于此前Arora、Basu、Mianjy和Mukherjee提出的算法。对于线性激活网络,我们首次提出满足新型数据吞吐量条件的网络架构可被最优训练,给出了首个非平凡的多项式时间可解类。
原文摘要 · Abstract (English)
In spite of the fundamental role of neural networks in contemporary machine learning research, our understanding of the computational complexity of optimally training neural networks remains incomplete even when dealing with the simplest kinds of activation functions. Indeed, while there has been a number of very recent results that establish ever-tighter lower bounds for the problem under linear and ReLU activation functions, less progress has been made towards the identification of novel polynomial-time tractable network architectures. In this article we obtain novel algorithmic upper bounds for training linear- and ReLU-activated neural networks to optimality which push the boundaries of tractability for these problems beyond the previous state of the art. In particular, for ReLU networks we establish the polynomial-time tractability of all architectures where hidden neurons have an out-degree of $1$, improving upon the previous algorithm of Arora, Basu, Mianjy and Mukherjee. On the other hand, for networks with linear activation functions we identify the first non-trivial polynomial-time solvable class of networks by obtaining an algorithm that can optimally train network architectures satisfying a novel data throughput condition.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。