Husimi spectral concentration regularizer
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
Apply a convex Husimi functional as a differentiable regularizer to positive matrices used by attention heads, routers, or feature covariances. Penalizing the squared response suppresses sharp spherical peaks and can prevent collapsed routing or unstable attention without directly forcing uniform eigenvalues.
Formulas
Mathematical statement
Let \(\rho\succeq0\) with \(\operatorname{tr}\rho=1\), and let \(Q_\rho(z)=\langle\kappa_{N,z},\rho\kappa_{N,z}\rangle\in[0,1]\). Define \(F_\Phi(\rho)=\int\Phi(Q_\rho(z))dm(z)\). The paper’s isospectral majorization theorem states \(F_\Phi(\rho)\le F_\Phi(\rho^\downarrow)\) for every convex \(\Phi\), where \(\rho^\downarrow=\sum_{n=0}^N\lambda_n^\downarrow|e_n^{(N)}\rangle\langle e_n^{(N)}|\) and \(\lambda_n^\downarrow\) are the eigenvalues sorted decreasingly. Choose \(\Phi(t)=t^2\) to obtain the concentration functional \(F_2(\rho)=\int Q_\rho(z)^2dm(z)\); higher \(t^p\), \(p>1\), emphasizes peaks. On a quadrature grid, use \(\widehat F_p(\rho)=\sum_jw_j(u_j^*\rho u_j)^p\), where \(w_j\ge0\) and \(\sum_jw_j=1\). The result gives a basis-sensitive concentration diagnostic and a principled family of convex penalties.
Implementation notes
Integrate the regularizer into a router or attention module that maintains a small positive matrix per head, layer, or expert. Parameterize an unconstrained matrix B as a lower-triangular L, set L_ii=softplus(B_ii)+epsilon, and construct rho=LL*/(trace(LL*)+epsilon). Precompute M spherical points and normalized quadrature weights w_j. Compute q_j=real(u_j* rho u_j), then F2=sum_j w_j q_j^2. Add lambda F2 to the task loss to discourage sharp responses, or add lambda(F2-c)^2 when a target concentration c is desired. Pseudocode: B -> L -> rho; q=real(diag(U @ rho @ U.conj().T)); F2=(w*q*q).sum(); loss=task_loss+lambda*F2; backpropagate. The theorem supplies the convex spectral ordering; quadrature accuracy, lambda, and c are empirical. Test first on a 6-layer ViT-Tiny or four-expert MoE Transformer on CIFAR-100 or WikiText-2, comparing no regularizer, standard router load-balancing entropy, and this penalty at matched parameters and FLOPs. Track validation loss, router entropy, maximum q_j, effective expert count, and gradient variance. Success is fewer collapsed experts or attention spikes and better validation accuracy without a material training-speed penalty.
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.