Outlier-Spectral Landmark Attention
Implementation & benchmark of arXiv:2608.00840 — Sublinear Time Eigenvector Approximation via Column Sampling
Source paper: Sublinear Time Eigenvector Approximation via Column Sampling arXiv:2608.00840 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Construct a symmetric token affinity matrix and approximate only its spectrally outlying token-mixing modes using a small set of sampled landmark columns. The resulting low-rank operator replaces an \(O(n^2)\) dense mixer by two skinny matrix multiplications, while the paper's residual guarantee predicts that large-magnitude global interaction modes are preserved.
Formulas
Mathematical statement
The paper assumes a symmetric matrix \(A\in\mathbb R^{n\times n}\) with entrywise bound \(\|A\|_\infty\le 1\), and for \(\epsilon,\delta\in(0,1)\) samples \(s=\frac{c\log n}{\epsilon^4\delta}\log^3\frac{1}{\epsilon\delta}\) columns. With probability at least \(1-\delta\), every eigenvalue \(\lambda\) satisfying \(|\lambda|\ge \epsilon n\) has an estimated eigenvalue \(\tilde\lambda\) and normalized vector \(v\) obeying \(|\lambda-\tilde\lambda|\le \epsilon n\) and \(\|Av-\lambda v\|_2\le \epsilon n\). There are at most \(1/\epsilon^2\) such eigenvalues because \(\|A\|_F\le n\). For an implementable neural version, let \(X\in\mathbb R^{n\times d}\) be token features, define a bounded symmetric affinity \(A_{ij}=\tanh(x_i^Tx_j/\sqrt d)\) or a clipped Gaussian kernel, sample landmark indices \(S\) of size \(s\), form \(C=A[:,S]\) and \(W=A[S,S]\), and use the Nyström operator \(\widehat A=CW^\dagger C^T\). Its nonzero eigenvectors lie in \(\operatorname{span}(C)\), matching the paper's column-span construction.
Implementation notes
Integrate this into a transformer as a token-mixing residual branch, either after the value projection or as a replacement for one dense self-attention block. For token matrix X with shape [batch,n,d], first compute a symmetric bounded affinity A implicitly: use normalized tokens Z=X/(||X||_2+10^-6), compute landmark affinities C between every token and s sampled landmarks, and compute W, the s-by-s landmark affinity matrix. Sample S uniformly without replacement during training; for cheaper inference, fix S using a deterministic hash or reservoir sampling. Pseudocode: (1) choose s=ceil(c log(n)/(epsilon^4 delta) log(1/(epsilon delta))^3), clipped to [r_min,n]; (2) compute C_ij=tanh(z_i^T z_{S_j}/sqrt(d)) and W_ij=tanh(z_{S_i}^T z_{S_j}/sqrt(d)); (3) symmetrize W=(W+W^T)/2; (4) compute W=U diag(mu) U^T and set P=U diag(1/mu_k if |mu_k|>=tau_W, else 0) U^T; (5) for values V, output Y=X+alpha C P(C^T V), where alpha is a learned scalar. The paper's mathematical contribution is the sampled-column span and outlier threshold; Nyström is the concrete reconstruction used here because the extracted material does not expose Algorithm (1)'s full estimator. Estimate the approximation residual on validation batches by explicitly forming A for n<=1024 and computing ||(A-Ahat)V||_F/||AV||_F. First test on a 6-layer transformer with 512 and 1024 tokens using CIFAR-100 patch tokens or Long Range Arena, comparing full attention, standard Nyström attention, and this module. Record accuracy, wall-clock time, peak memory, rank, and spectral residual. Success means at least 1.5x lower attention memory or latency with less than one percentage point accuracy loss, while increasing s should reduce the measured residual and preserve the dominant eigenmodes.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.