cs.LGFeb 17, 2026

A unified theory of feature learning in RNNs and DNNs

Authors: Jan P. Bauer, Kirsten Fischer, Moritz Helias, Agostina Palmigiano

Organizations: Gatsby Computational Neuroscience Unit, UCL, London, UK · IAS-6, Forschungszentrum Juelich, Juelich, Germany

Abstract

Recurrent and deep neural networks (RNNs/DNNs) are cornerstone architectures in machine learning. Remarkably, RNNs differ from DNNs only by weight sharing, as can be shown through unrolling in time. How does this structural similarity fit with the distinct functional properties these networks exhibit? To address this question, we here develop a unified mean-field theory for RNNs and DNNs in terms of representational kernels, describing fully trained networks in the feature learning (μμP) regime. This theory casts training as Bayesian inference over sequences and patterns, directly revealing the functional implications induced by the RNNs' weight sharing. In DNN-typical tasks, we identify a phase transition when the learning signal overcomes the noise due to randomness in the weights: below this threshold, RNNs and DNNs behave identically; above it, only RNNs develop correlated representations across timesteps. For sequential tasks, the RNNs' weight sharing furthermore induces an inductive bias that aids generalization by interpolating unsupervised time steps. Overall, our theory offers a way to connect architectural structure to functional biases.

Explore similar work

CardsList
  1. Sublinearly Structured Deep Neural Networks Achieve Feature Learning Consistency for Compositional Functions

    Jun 22, 2026Sehwan Kim, Yan Sun, Faming LiangNeural Network GeneralizationHierarchical Representation Learning

  2. Pointwise Generalization in Deep Neural Networks

    May 18, 2026Shaojie Li, Yunbei XuNeural Network GeneralizationRepresentation Learning

  3. The Weight Gram Matrix Captures Sequential Feature Linearization in Deep Networks

    May 7, 2026Taehun Cha, Daniel Beaglehole, Adityanarayanan Radhakrishnan +1Neural Network Training DynamicsGradient Descent Dynamics