Dynamical similarity analysis can identify compositional dynamics developing in RNNs

TL;DR

This paper introduces a new dynamical similarity metric, DSA, which reliably detects compositional dynamics in RNNs, outperforming prior metrics and providing a benchmark for analyzing neural representations.

Contribution

The paper develops a test case based on compositional learning in RNNs to evaluate and improve dynamical representation metrics, introducing DSA as a more robust alternative.

Findings

DSA outperforms prior metrics like Procrustes and CKA in noise robustness.

DSA reliably identifies behaviorally relevant, compositional dynamical motifs in RNNs.

Modern state space models may not exhibit the same dynamical changes as RNNs, as shown by DSA analysis.

Abstract

Methods for analyzing representations in neural systems have become a popular tool in both neuroscience and mechanistic interpretability. Having measures to compare how similar activations of neurons are across conditions, architectures, and species, gives us a scalable way of learning how information is transformed within different neural networks. In contrast to this trend, recent investigations have revealed how some metrics can respond to spurious signals and hence give misleading results. To identify the most reliable metric and understand how measures could be improved, it is going to be important to identify specific test cases which can serve as benchmarks. Here we propose that the phenomena of compositional learning in recurrent neural networks (RNNs) allows us to build a test case for dynamical representation alignment metrics. By implementing this case, we show it enables us to test whether metrics can identify representations which gradually develop throughout learning and probe whether representations identified by metrics are relevant to computations executed by networks. By building both an attractor- and RNN-based test case, we show that the new Dynamical Similarity Analysis (DSA) is more noise robust and identifies behaviorally relevant representations more reliably than prior metrics (Procrustes, CKA). We also show how test cases can be used beyond evaluating metrics to study new architectures. Specifically, results from applying DSA to modern (Mamba) state space models, suggest that, in contrast to RNNs, these models may not exhibit changes to their recurrent dynamics due to their expressiveness. Overall, by developing test cases, we show DSA's exceptional ability to detect compositional dynamical motifs, thereby enhancing our understanding of how computations unfold in RNNs.