We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cond-mat.dis-nn

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Condensed Matter > Disordered Systems and Neural Networks

Title: Theory of gating in recurrent neural networks

Abstract: Recurrent neural networks (RNNs) are powerful dynamical models, widely used in machine learning (ML) for processing sequential data, and in neuroscience, to understand the emergent properties of networks of real neurons. Prior theoretical work in understanding the properties of RNNs has focused on networks with additive interactions. However, gating -- i.e. multiplicative -- interactions are ubiquitous in real neurons, and gating is also the central feature of the best-performing RNNs in ML. Here, we study the consequences of gating for the dynamical behavior of RNNs. We show that gating leads to slow modes and a novel, marginally-stable state. The network in this marginally-stable state can function as a robust integrator, and unlike previous approaches, gating permits this function without parameter fine-tuning or special symmetries. We study the long-time behavior of the gated network using its Lyapunov spectrum, and provide a novel relation between the maximum Lyapunov exponent and the relaxation time of the dynamics. Gating is also shown to give rise to a novel, discontinuous transition to chaos, where the proliferation of critical points (topological complexity) is decoupled from the appearance of chaotic dynamics (dynamical complexity), in contrast to a seminal result for additive RNNs. The rich dynamical behavior is summarized in a phase diagram indicating critical surfaces and regions of marginal stability -- thus, providing a map for principled parameter choices to ML practitioners. Finally, we develop a field theory for gradients that arise in training, by combining the adjoint formalism from control theory with the dynamical mean-field theory. This paves the way for the use of powerful field theoretic techniques to study training and gradients in large RNNs.
Comments: 11 figures
Subjects: Disordered Systems and Neural Networks (cond-mat.dis-nn); Statistical Mechanics (cond-mat.stat-mech); Machine Learning (cs.LG); Chaotic Dynamics (nlin.CD); Neurons and Cognition (q-bio.NC)
Cite as: arXiv:2007.14823 [cond-mat.dis-nn]
  (or arXiv:2007.14823v4 [cond-mat.dis-nn] for this version)

Submission history

From: Kamesh Krishnamurthy [view email]
[v1] Wed, 29 Jul 2020 13:20:58 GMT (5249kb,D)
[v2] Sat, 29 Aug 2020 21:48:52 GMT (5301kb,D)
[v3] Thu, 3 Sep 2020 20:02:16 GMT (5250kb,D)
[v4] Thu, 21 Jan 2021 03:03:56 GMT (5779kb,D)
[v5] Wed, 1 Dec 2021 17:43:29 GMT (4292kb,D)

Link back to: arXiv, form interface, contact.