English

Spectral Selection in Symmetric Self-Attention Dynamics

Dynamical Systems 2026-04-30 v1 Classical Analysis and ODEs

Abstract

We study self-attention dynamics on the unit sphere as an interacting particle system arising from an idealized Transformer-type update. Under a symmetry assumption on weight matrices given by QK=V=VQ^\top K=V=V^\top, the flow admits a gradient-flow structure and an exact reformulation in the eigenbasis of VV, revealing a spectral mode-selection mechanism. We show that the dynamics exhibits two distinct asymptotic scenarios: homogeneous alignment toward the dominant eigendirection when one positive eigenvalue strictly dominates all others in modulus, and sign-split polarization toward the most negative eigendirection when VV is negative definite. In particular, we obtain local stability criteria for pure-mode equilibria and global selection results in both regimes. These results provide a rigorous finite-particle description of how the spectrum of the weight matrices organizes asymptotic patterns in a symmetric self-attention flow, and highlight how the symmetric setting renders the dynamics amenable to mathematical analysis.

Keywords

Cite

@article{arxiv.2604.26085,
  title  = {Spectral Selection in Symmetric Self-Attention Dynamics},
  author = {Christian Kuehn and Jaeyoung Yoon},
  journal= {arXiv preprint arXiv:2604.26085},
  year   = {2026}
}

Comments

49 pages, 5 figures

R2 v1 2026-07-01T12:40:06.275Z