中文

张量切换网络

神经与进化计算 2016-11-01 v1 机器学习 机器学习

摘要

我们提出了一种新的神经网络算法,即张量切换 (TS) 网络,它将整流线性单元 (ReLU) 非线性推广到张量值隐藏单元。TS 网络将其整个输入向量复制到扩展表示中的不同位置,位置由其隐藏单元活动决定。通过这种方式,即使是来自 TS 表示的简单线性读出也能实现高度表达式的深网络类函数。因此,TS 网络通过构造避免了梯度消失问题,代价是更大的表示尺寸。我们开发了多种训练 TS 网络的方法,包括用于无限宽和深 TS 网络的等效核、一次性线性学习算法以及两种受反向传播启发的表示学习算法。我们的实验结果表明,TS 网络确实具有更强的表达能力,并且比标准 ReLU 网络学习得更快。

关键词

引用

@article{arxiv.1610.10087,
  title  = {Tensor Switching Networks},
  author = {Chuan-Yung Tsai and Andrew Saxe and David Cox},
  journal= {arXiv preprint arXiv:1610.10087},
  year   = {2016}
}