揭示神经网络在叠加态下计算的复杂性极限,首次给出参数下界。
On the Complexity of Neural Computation in Superposition
- 提出计算叠加态特征所需最小神经元数和参数量的理论下界。
- 证明计算m'个特征至少需Ω(√(m' log m'))个神经元和Ω(m' log m')参数。
- 适用于模型压缩与蒸馏,适合关注大模型效率的研究者。
叠加态使神经网络能表示超过神经元数量的特征,被视为大模型高效的关键。本文研究叠加态下的神经计算理论基础,建立了显式且可证明正确的算法复杂度下界。对包括排列和成对逻辑运算在内的广泛问题,计算m'个叠加特征至少需要Ω(√(m' log m'))个神经元和Ω(m' log m')个参数。这表明模型稀疏化或蒸馏存在明确限制,并补充经验缩放定律,首次给出容量的次指数上界:含n个神经元的网络最多可计算O(n² / log n)个特征。反之,我们提供了近似紧致的构造性上界:成对逻辑运算(如AND)可用O(√(m') log m')个神经元和O(m' log² m')个参数实现。计算叠加态的复杂度与仅表示特征的复杂度之间存在指数级差距——后者仅需O(log m')个神经元(基于Johnson-Lindenstrauss引理)。本工作首次从理论上确立参数数量是网络实际计算特征数的良好估计器。
原文摘要 · Abstract (English)
Superposition, the ability of neural networks to represent more features than neurons, is increasingly seen as key to the efficiency of large models. This paper investigates the theoretical foundations of computing in superposition, establishing complexity bounds for explicit, provably correct algorithms. We present the first lower bounds for a neural network computing in superposition, showing that for a broad class of problems, including permutations and pairwise logical operations, computing $m'$ features in superposition requires at least $Ω(\sqrt{m' \log m'})$ neurons and $Ω(m' \log m')$ parameters. This implies an explicit limit on how much one can sparsify or distill a model while preserving its expressibility, and complements empirical scaling laws by implying the first subexponential bound on capacity: a network with $n$ neurons can compute at most $O(n^2 / \log n)$ features. Conversely, we provide a nearly tight constructive upper bound: logical operations like pairwise AND can be computed using $O(\sqrt{m'} \log m')$ neurons and $O(m' \log^2 m')$ parameters. There is thus an exponential gap between the complexity of computing in superposition (the subject of this work) versus merely representing features, which can require as little as $O(\log m')$ neurons based on the Johnson-Lindenstrauss Lemma. Our work analytically establishes that the number of parameters is a good estimator of the number of features a neural network computes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。