中文

缩减神经网络训练的参数空间

机器学习 2020-01-30 v3 计算机视觉与模式识别 机器学习 神经与进化计算

摘要

对于具有修正线性单元(ReLU)或二值激活函数的神经网络(NNs),我们表明其训练可在缩减的参数空间中完成。具体而言,每个神经元中的权重可在单位球面上训练,而非整个空间,阈值可在有界区间内训练,而非实直线。我们表明缩减参数空间中的NNs与具有全空间参数的标准NNs在数学上等价。缩减的参数空间将促进网络训练的优化过程,因为搜索空间变得(更)小。我们通过数值例子展示了改进的训练性能。

关键词

引用

@article{arxiv.1805.08340,
  title  = {Reducing Parameter Space for Neural Network Training},
  author = {Tong Qin and Ling Zhou and Dongbin Xiu},
  journal= {arXiv preprint arXiv:1805.08340},
  year   = {2020}
}

备注

17 pages, 8 figures