Tensor Switching Networks

By Chuan-Yung Tsai, Andrew M. Saxe, David D. Cox. 2016

10 citations recorded by the source corpus.

We present a novel neural network algorithm, the Tensor Switching (TS) network, which generalizes the Rectified Linear Unit (ReLU) nonlinearity to tensor-valued hidden units. The TS network copies its entire input vector to different locations in an expanded representation, with the location determined by its hidden unit activity. In this way, even a simple linear readout from the TS representation can implement a highly expressive deep-network-like function. The TS network hence avoids the vanishing gradient problem by construction, at the cost of larger representation size. We develop several methods to train the TS network, including equivalent kernels for infinitely wide and deep TS networks, a one-pass linear learning algorithm, and two backpropagation-inspired representation learning algorithms. Our experimental results demonstrate that the TS network is indeed more expressive and consistently learns faster than standard ReLU networks.

Subfields: Computational Mathematics, Computer Vision and Pattern Recognition, Statistical and Nonlinear Physics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Mathematics

  1. Ameet Talwalkar
  2. Razvan Pascanu
  3. Michael W. Mahoney

In Computer Vision and Pattern Recognition

  1. Ameet Talwalkar
  2. Razvan Pascanu
  3. Virginia Smith
  4. Tom Schaul
  5. Maziar Sanjabi
  6. Diana Borsa
  7. Joseph Modayil
  8. Luc Van Gool

In Statistical and Nonlinear Physics

  1. Ameet Talwalkar
  2. Tom Schaul
  3. Razvan Pascanu
  4. Diana Borsa
  5. Joseph Modayil
  6. Yoshua Bengio
  7. Kyunghyun Cho
  8. Li Deng

Citations in the corpus