中文

ReLU神经网络中的稳定未激活神经元

机器学习 2024-12-18 v2 概率论 机器学习

摘要

神经网络架构的选择决定了该神经网络可以实现哪些函数,因此研究所选架构的表达能力受到了广泛关注。在ReLU神经网络中,稳定未激活神经元的存在会降低网络的表达能力。在本文中,我们研究了当权重和偏置从对称概率分布初始化时,此类神经网络第二隐藏层中神经元被稳定未激活的概率。对于输入维度为 n0n_0 的网络,我们证明如果第一隐藏层有 n0+1n_0+1 个神经元,则该概率精确为 2n0+14n0+1\frac{2^{n_0}+1}{4^{n_0+1}};如果第一隐藏层有 n1n_1 个神经元且 n1n0n_1 \le n_0,则该概率为 12n1+1\frac{1}{2^{n_1+1}}。最后,针对第一隐藏层神经元数量超过 n0+1n_0+1 的情况,提出了一个猜想并给出了理由。计算证据被提供以支持该猜想。

关键词

引用

@article{arxiv.2412.06829,
  title  = {Stably unactivated neurons in ReLU neural networks},
  author = {Natalie Brownlowe and Christopher R. Cornwell and Ethan Montes and Gabriel Quijano and Grace Stulman and Na Zhang},
  journal= {arXiv preprint arXiv:2412.06829},
  year   = {2024}
}