Bloch-Husimi attention
Implementation & benchmark of arXiv:2608.12248 — Isospectral majorization and isoperimetric inequalities for coherent states on the Bloch sphere
Source paper: Isospectral majorization and isoperimetric inequalities for coherent states on the Bloch sphere arXiv:2608.12248 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace unconstrained attention score vectors by normalized SU(2) coherent-state responses of a positive operator on an (N+1)-dimensional spin space. Each query produces a smooth bounded response over a fixed spherical grid, while values are aggregated normally. The coherent-state kernel imposes geometric structure and exposes a controllable concentration parameter N.
Formulas
Mathematical statement
Use the Hilbert space \(\mathcal P_N\) of analytic polynomials of degree at most \(N\), with orthonormal basis \(e_n^{(N)}(z)=\binom{N}{n}^{1/2}z^n\), \(0\le n\le N\). Its normalized coherent state is \(\kappa_{N,z}(w)=(1+w\bar z)^N/(1+|z|^2)^{N/2}\). For a density operator \(\rho\succeq0\), \(\operatorname{tr}\rho=1\), define \(Q_\rho(z)=\langle\kappa_{N,z},\rho\kappa_{N,z}\rangle\). With normalized spherical measure \(dm(z)=dA(z)/(\pi(1+|z|^2)^2)\), the paper proves that if \(\rho^\downarrow\) places the eigenvalues of \(\rho\) in decreasing order on the monomial basis, then \(\int\Phi(Q_\rho)dm\le\int\Phi(Q_{\rho^\downarrow})dm\) for every convex \(\Phi:[0,1]\to\mathbb R\). This is an isospectral majorization result: basis alignment controls concentration of the coherent response.
Implementation notes
Integrate this at the attention-score stage of a small spherical attention layer. Choose spin dimension N+1, a fixed grid of M points z_j on the Riemann sphere, and precompute U in C^{M x (N+1)} with U_{jn}=binom(N,n)^{1/2} z_j^n/(1+|z_j|^2)^{N/2}. For each query token x, produce a Hermitian matrix A(x) using a linear head, symmetrize A as (A+A*)/2, and form rho=expm(A)/trace(expm(A)); for an MVP, use rho=LL*/trace(LL*) with L lower triangular and positive diagonal. Compute q_j=real(u_j* rho u_j), logits l_j=log(q_j+epsilon)/tau, probabilities p=softmax(l), and output y=sum_j p_j V_j. Pseudocode: A=head(x); rho=positive_trace_one(A); q=diag(U rho U*); p=softmax(log(q+eps)/tau); y=p @ V. The paper supplies the coherent-state map and convex ordering; quadrature, temperature, and matrix parameterization are engineering choices. First compare N=3 or 7, M=64 against same-width softmax attention on CIFAR-100 or a WikiText-2 Transformer slice. Log validation loss, attention entropy, gradient variance, and wall-clock cost. Success is improved validation accuracy or smoother loss at matched FLOPs, with concentration increasing predictably as N grows.
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.