quant-phJun 10, 2026

Higher-Order Token Interactions via Quantum Attention

Authors: Jian XuChao LiDelu ZengJohn PaisleyQibin Zhao

Organizations: 1RIKEN iTHEMS · 2RIKEN AIP · 3South China University of Technology · 4Columbia University

Abstract

Standard dot-product self-attention computes, in a single layer, only pairwise (order-2) interactions between tokens; representing a generic order-kk interaction is known to require either super-quadratic resources in one layer or composition across depth. We introduce \textbf{Quantum Higher-Order Attention (QHA)}, a shallow, hardware-realizable quantum attention head that, via data re-uploading and an all-to-all non-Clifford entangler, synthesizes order-kk token interactions inside the circuit and exposes them through a local single-qubit read-out. We prove (i) an expressivity separation: any single standard self-attention layer with embedding dimension mm, HH heads and pp-bit precision satisfying mHp=o(N/loglogN)mHp=o(N/\log\log N) cannot represent the order-kk correlation family that one QHA head represents with circuit depth O(logk)O(\log k) (O(k)O(k) two-qubit gates); and (ii) a trainability guarantee for its local-design instantiation: with a local read-out and O(logn)O(\log n) depth the gradient variance is Ω(1/poly(n))Ω(1/\mathrm{poly}(n)) (no barren plateau), which we confirm empirically -- while being explicit that the more expressive all-to-all instantiation we benchmark is trained empirically and shows exponentially decaying gradients. Empirically, at a 6.5×6.5\times smaller parameter budget, QHA generalizes hidden-subset parity of every order k6k\le6 from disjoint inputs, whereas the larger classical attention head collapses past order~2; consistent with theory, the size of the advantage tracks the target's Fourier degree - largest for parity and shrinking when low-order structure is present. As an application, QHA serves as a compact high-order interaction detector across three domains - genetic epistasis, learning-parity-with-noise, and graph triangle detection - reaching the noise ceiling at the smallest parameter budget where field-standard linear methods fail.

Explore similar work

CardsList