Path-Normalized Optimization of Recurrent Neural Networks with ReLU Activations
Neyshabur, Behnam, Wu, Yuhuai, Salakhutdinov, Russ R., Srebro, Nati
–Neural Information Processing Systems
We investigate the parameter-space geometry of recurrent neural networks (RNNs), and develop an adaptation of path-SGD optimization method, attuned to this geometry, that can learn plain RNNs with ReLU activations. On several datasets that require capturing long-term dependency structure, we show that path-SGD can significantly improve trainability of ReLU RNNs compared to RNNs trained with SGD, even with various recently suggested initialization schemes. Papers published at the Neural Information Processing Systems Conference.
Neural Information Processing Systems
Feb-14-2020, 13:58:26 GMT
- Technology: