Emergence of Globally Attracting Fixed Points in Deep Neural Networks With Nonlinear Activations

Amir Joudaki, Thomas Hofmann
10/26/2024
cs.LGcs.AIstat.ML

No confident code implementation found for this paper.

Abstract

Understanding how neural networks transform input data across layers is fundamental to unraveling their learning and generalization capabilities. Although prior work has used insights from kernel methods to study neural networks, a global analysis of how the similarity between hidden representations evolves across layers remains underexplored. In this paper, we introduce a theoretical framework for the evolution of the kernel sequence, which measures the similarity between the hidden representation for two different inputs. Operating under the mean-field regime, we show that the kernel sequence evolves deterministically via a kernel map, which only depends on the activation function. By expanding activation using Hermite polynomials and using their algebraic properties, we derive an explicit form for kernel map and fully characterize its fixed points. Our analysis reveals that for nonlinear activations, the kernel sequence converges globally to a unique fixed point, which can correspond to orthogonal or similar representations depending on the activation and network architecture. We further extend our results to networks with residual connections and normalization layers, demonstrating similar convergence behaviors. This work provides new insights into the implicit biases of deep neural networks and how architectural choices influence the evolution of representations across layers.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{joudaki2024emergence,
  title  = {Emergence of Globally Attracting Fixed Points in Deep Neural Networks With Nonlinear Activations},
  author = {Amir Joudaki and Thomas Hofmann},
  year   = {2024},
  eprint = {2410.20107},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.20107v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.20107.svg)](https://codeofpaper.com/paper/2410.20107)

Discussion