Literature DB >> 32782423

Reverse engineering recurrent networks for sentiment classification reveals line attractor dynamics.

Niru Maheswaranathan1, Alex H Williams2, Matthew D Golub2, Surya Ganguli3, David Sussillo1.   

Abstract

Recurrent neural networks (RNNs) are a widely used tool for modeling sequential data, yet they are often treated as inscrutable black boxes. Given a trained recurrent network, we would like to reverse engineer it-to obtain a quantitative, interpretable description of how it solves a particular task. Even for simple tasks, a detailed understanding of how recurrent networks work, or a prescription for how to develop such an understanding, remains elusive. In this work, we use tools from dynamical systems analysis to reverse engineer recurrent networks trained to perform sentiment classification, a foundational natural language processing task. Given a trained network, we find fixed points of the recurrent dynamics and linearize the nonlinear system around these fixed points. Despite their theoretical capacity to implement complex, high-dimensional computations, we find that trained networks converge to highly interpretable, low-dimensional representations. In particular, the topological structure of the fixed points and corresponding linearized dynamics reveal an approximate line attractor within the RNN, which we can use to quantitatively understand how the RNN solves the sentiment analysis task. Finally, we find this mechanism present across RNN architectures (including LSTMs, GRUs, and vanilla RNNs) trained on multiple datasets, suggesting that our findings are not unique to a particular architecture or dataset. Overall, these results demonstrate that surprisingly universal and human interpretable computations can arise across a range of recurrent networks.

Entities:  

Year:  2019        PMID: 32782423      PMCID: PMC7416638     

Source DB:  PubMed          Journal:  Adv Neural Inf Process Syst        ISSN: 1049-5258


  5 in total

1.  Nonlinear dimensionality reduction by locally linear embedding.

Authors:  S T Roweis; L K Saul
Journal:  Science       Date:  2000-12-22       Impact factor: 47.728

2.  Long short-term memory.

Authors:  S Hochreiter; J Schmidhuber
Journal:  Neural Comput       Date:  1997-11-15       Impact factor: 2.026

3.  How the brain keeps the eyes still.

Authors:  H S Seung
Journal:  Proc Natl Acad Sci U S A       Date:  1996-11-12       Impact factor: 11.205

4.  Opening the black box: low-dimensional dynamics in high-dimensional recurrent neural networks.

Authors:  David Sussillo; Omri Barak
Journal:  Neural Comput       Date:  2012-12-28       Impact factor: 2.026

5.  LSTMVis: A Tool for Visual Analysis of Hidden State Dynamics in Recurrent Neural Networks.

Authors:  Hendrik Strobelt; Sebastian Gehrmann; Hanspeter Pfister; Alexander M Rush
Journal:  IEEE Trans Vis Comput Graph       Date:  2017-08-29       Impact factor: 4.579

  5 in total
  2 in total

1.  Training deep neural density estimators to identify mechanistic models of neural dynamics.

Authors:  Pedro J Gonçalves; Jan-Matthis Lueckmann; Michael Deistler; Marcel Nonnenmacher; Kaan Öcal; Giacomo Bassetto; Chaitanya Chintaluri; William F Podlaski; Sara A Haddad; Tim P Vogels; David S Greenberg; Jakob H Macke
Journal:  Elife       Date:  2020-09-17       Impact factor: 8.140

2.  Synaptic balancing: A biologically plausible local learning rule that provably increases neural network noise robustness without sacrificing task performance.

Authors:  Christopher H Stock; Sarah E Harvey; Samuel A Ocko; Surya Ganguli
Journal:  PLoS Comput Biol       Date:  2022-09-19       Impact factor: 4.779

  2 in total

北京卡尤迪生物科技股份有限公司 © 2022-2023.