中文

基于ReLU激活的循环神经网络路径归一化优化

机器学习 2016-05-24 v1 神经与进化计算

摘要

我们研究了循环神经网络(RNN)的参数空间几何,并开发了一种适应此几何的路径-SGD优化方法变体,能够学习带有ReLU激活的普通RNN。在几个需要捕捉长期依赖结构的数据集上,我们证明,与使用SGD训练的RNN相比,即使采用各种近期提出的初始化方案,路径-SGD也能显著提高ReLU RNN的可训练性。

关键词

引用

@article{arxiv.1605.07154,
  title  = {Path-Normalized Optimization of Recurrent Neural Networks with ReLU Activations},
  author = {Behnam Neyshabur and Yuhuai Wu and Ruslan Salakhutdinov and Nathan Srebro},
  journal= {arXiv preprint arXiv:1605.07154},
  year   = {2016}
}

备注

15 pages