Showing 1–1 of 1 results for author: Henry, N W

Geometry of Lightning Self-Attention: Identifiability and Dimension

Authors: Nathan W. Henry, Giovanni Luca Marchetti, Kathlén Kohn

Abstract: We consider function spaces defined by self-attention networks without normalization, and theoretically analyze their geometry. Since these networks are polynomial, we rely on tools from algebraic geometry. In particular, we study the identifiability of deep attention by providing a description of the generic fibers of the parametrization for an arbitrary number of layers and, as a consequence, co… ▽ More We consider function spaces defined by self-attention networks without normalization, and theoretically analyze their geometry. Since these networks are polynomial, we rely on tools from algebraic geometry. In particular, we study the identifiability of deep attention by providing a description of the generic fibers of the parametrization for an arbitrary number of layers and, as a consequence, compute the dimension of the function space. Additionally, for a single-layer model, we characterize the singular and boundary points. Finally, we formulate a conjectural extension of our results to normalized self-attention networks, prove it for a single layer, and numerically verify it in the deep case. △ Less

Submitted 19 February, 2025; v1 submitted 30 August, 2024; originally announced August 2024.

Comments: Accepted at ICLR 2025