RNNs perform task computations by dynamically warping neural representations
Arthur Pellegrino, Angus Chadwick
Abstract
Analysing how neural networks represent data features in their activations can help interpret how they perform tasks. Hence, a long line of work has focused on mathematically characterising the geometry of such "neural representations." In parallel, machine learning has seen a surge of interest in understanding how dynamical systems perform computations on time-varying input data. Yet, the link between computation-through-dynamics and representational geometry remains poorly understood. Here, we hypothesise that recurrent neural networks (RNNs) perform computations by dynamically warping their representations of task variables. To test this hypothesis, we develop a Riemannian geometric framework that enables the derivation of the manifold topology and geometry of a dynamical system from the manifold of its inputs. By characterising the time-varying geometry of RNNs, we show that dynamic warping is a fundamental feature of their computations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on16
- Photorealistic Text-to-Image Diffusion Models with Deep Language UnderstandingChitwan Saharia, William Chan, Saurabh Saxena, Lala Li et al.NeurIPS 2022 · 8,965 citations
- Organizing recurrent network dynamics by task-computation to enable continual learningLea Duncker, Laura Driscoll, Krishna V. Shenoy, Maneesh Sahani et al.NeurIPS 2020 · 108 citations
- The interplay between randomness and structure during learning in RNNsFriedrich Schüßler, Francesca Mastrogiuseppe, Alexis M. Dubreuil, Srdjan Ostojic et al.NeurIPS 2020 · 91 citations
- Extracting computational mechanisms from neural data using low-rank RNNsAdrian Valente, Jonathan W. Pillow, Srdjan OstojicNeurIPS 2022 · 71 citations
- Reverse-engineering recurrent neural network solutions to a hierarchical inference task for miceRylan Schaeffer, Mikail Khona, Leenoy Meshulam, International Brain Laboratory et al.NeurIPS 2020 · 49 citations
Related papers
- Temporal superposition and feature geometry of RNNs under memory demandsPratyaksh Sharma, Alexandra Maria Proca, Lucas Prieto, Pedro A. M. MedianoICLR 2026
- Exploring the Enigma of Neural Dynamics Through A Scattering-Transform Mixer Landscape for Riemannian ManifoldTingting Dan, Ziquan Wei, Won Hwa Kim, Guorong WuICML 2024 · 7 citations
- Neural Feature Geometry Evolves as Discrete Ricci FlowMoritz Hehl, Max von Renesse, Melanie WeberICML 2026 · 2 citations
- Riemannian Liquid Spatio-Temporal Graph NetworkLiangsi Lu, Jingchao Wang, Zhaorong Dai, Hanqian Liu et al.WWW 2026 · 1 citation
- Beyond Geometry: Comparing the Temporal Structure of Computation in Neural Circuits with Dynamical Similarity AnalysisMitchell Ostrow, Adam Eisen, Leo Kozachkov, Ila FieteNeurIPS 2023 · 60 citations
