Two-cluster convergence conjecture for causal transformer dynamics with simple dominant eigendirection

From papers

Let QQ and KK be arbitrary matrices, and let VV be diagonalizable with dd different positive real eigenvalues. Denote its largest eigenvalue by λmax\lambda_{\max}, and let ξ\xi be a unit eigenvector satisfying

Vξ=λmaxξ.V\xi=\lambda_{\max}\xi.

For the causal transformer dynamics, two-cluster convergence conjecture. For almost every starting point (x1(0),,xn(0))(x_1(0),\ldots,x_n(0)) with respect to the volume measure on (Sd1)n(\mathbb{S}^{d-1})^n,

k[n],limtxk(t)ξ,ξ.\forall k\in[n],\qquad \lim_{t\to\infty}x_k(t)\in\\{\xi,-\xi\\}.

This conjectures that, when the dominant eigenvalue is positive and has a one-dimensional eigendirection under the stated diagonalizability assumptions, causal attention generically drives all tokens into two antipodal clusters. The claim is presented as part of the paper's conjectural extension beyond the proved identity-matrix case; its resolution is not supplied here.

Progress summary

Nothing recorded yet. Refresh searches the literature and the public web for attempts on this problem, and writes the first summary here.

Sources & referencesView supporting material

Primary source

Nikita Karagodin, Yury Polyanskiy and Philippe Rigollet, “Clustering in Causal Attention Masking”, arXiv:2411.04990 (2024).

Solutions 0

No solutions have been posted yet.