ReLU让深度网络可训练,其他激活函数则导致计算困难。
Why ReLU? A Bit-Model Dichotomy for Deep Network Training
- 用比特模型分析训练复杂度,发现非线性激活会导致难题
- 多项式激活下训练是#P难,梯度符号判断也难
- ReLU等分段线性激活使训练在NP内,适合实际应用
经典理论以实数计算模型分析经验风险最小化(ERM),认为简单神经网络训练已是∃ℝ完全问题,远超NP难度。但真实硬件为有限精度计算。本文引入比特级模型(ERM_bit),约束参数和输入为位长多项式有理数。在此模型下,我们揭示了激活函数决定的显著二分现象:任意具有有理系数且次数≥2的多项式激活函数,其训练问题是#P难,判断单个偏导数符号也极难(不属BPP),梯度某一位判定为#P难;这从复杂性角度解释了梯度爆炸/消失现象。相反,对于分段线性激活如ReLU,ERM_bit属于NP(具体为NP完全),标准反向传播可在多项式时间内完成。结果表明,有限精度不仅是实现细节,更是学习可行性的根本因素。
原文摘要 · Abstract (English)
Theoretical analyses of Empirical Risk Minimization (ERM) are standardly framed within the Real-RAM model of computation. In this setting, training even simple neural networks is known to be $\exists \mathbb{R}$-complete -- a complexity class believed to be harder than NP, that characterizes the difficulty of solving systems of polynomial inequalities over the real numbers. However, this algebraic framework diverges from the reality of digital computation with finite-precision hardware. In this work, we analyze the theoretical complexity of ERM under a realistic bit-level model ($\mathsf{ERM}_{\text{bit}}$), where network parameters and inputs are constrained to be rational numbers with polynomially bounded bit-lengths. Under this model, we reveal a sharp dichotomy in tractability governed by the network's activation function. We prove that for deep networks with {\em any} polynomial activations with rational coefficients and degree at least $2$, the bit-complexity of training is severe: deciding $\mathsf{ERM}_{\text{bit}}$ is $\#P$-Hard, hence believed to be strictly harder than NP-complete problems. Furthermore, we show that determining the sign of a single partial derivative of the empirical loss function is intractable (unlikely in BPP), and deciding a specific bit in the gradient is $\#P$-Hard. This provides a complexity-theoretic perspective for the phenomenon of exploding and vanishing gradients. In contrast, we show that for piecewise-linear activations such as ReLU, the precision requirements remain manageable: $\mathsf{ERM}_{\text{bit}}$ is contained within NP (specifically NP-complete), and standard backpropagation runs in polynomial time. Our results demonstrate that finite-precision constraints are not merely implementation details but fundamental determinants of learnability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。