Understanding Recurrent Neural Networks Using Nonequilibrium Response Theory
This provides a theoretical foundation for interpreting RNN behavior, which is incremental but useful for researchers in machine learning and statistical mechanics.
The authors tackled the problem of understanding how recurrent neural networks process sequential data by applying nonequilibrium response theory to derive an interpretable Volterra series representation for stochastic RNN outputs, showing that these networks can be viewed as kernel machines with fixed hidden weights.
Recurrent neural networks (RNNs) are brain-inspired models widely used in machine learning for analyzing sequential data. The present work is a contribution towards a deeper understanding of how RNNs process input signals using the response theory from nonequilibrium statistical mechanics. For a class of continuous-time stochastic RNNs (SRNNs) driven by an input signal, we derive a Volterra type series representation for their output. This representation is interpretable and disentangles the input signal from the SRNN architecture. The kernels of the series are certain recursively defined correlation functions with respect to the unperturbed dynamics that completely determine the output. Exploiting connections of this representation and its implications to rough paths theory, we identify a universal feature -- the response feature, which turns out to be the signature of tensor product of the input signal and a natural support basis. In particular, we show that SRNNs, with only the weights in the readout layer optimized and the weights in the hidden layer kept fixed and not optimized, can be viewed as kernel machines operating on a reproducing kernel Hilbert space associated with the response feature.