Skip to yearly menu bar Skip to main content


Poster

Geometry of Lightning Self-Attention: Identifiability and Dimension

Nathan Henry · Giovanni Luca Marchetti · Kathlén Kohn

Hall 3 + Hall 2B #142
[ ]
Fri 25 Apr 7 p.m. PDT — 9:30 p.m. PDT

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, 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.

Live content is unavailable. Log in and register to view live content