U-centered relational attention
Implementation & benchmark of arXiv:2608.01364 — U-centering as subset ANOVA: edge regression and higher-order theory
Source paper: U-centering as subset ANOVA: edge regression and higher-order theory arXiv:2608.01364 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace raw pairwise attention or graph-edge scores by exact U-centered residuals, removing additive effects attributable to either endpoint. The resulting scores represent interaction beyond independent source and destination biases and satisfy zero row sums, preventing a few high-degree or high-activation tokens from dominating relational aggregation.
Formulas
Mathematical statement
For a symmetric hollow edge-score matrix A in R^{n x n}, define A_{i\\cdot}=\\sum_{k\
e i}A_{ik} and A_{\\cdot\\cdot}=\\sum_{i}\\sum_{j\
e i}A_{ij}, where the second sum is over ordered distinct pairs. U-centering is the least-squares residual after fitting the additive model A_{ij} approximately equals alpha_i+alpha_j on i unequal to j: \\widetilde A_{ij}=A_{ij}-(A_{i\\cdot}+A_{j\\cdot})/(n-2)+A_{\\cdot\\cdot}/((n-1)(n-2)). For n at least 3, \\sum_{j\
e i}\\widetilde A_{ij}=0, so transformed edge scores are orthogonal to endpoint main effects. The residual space has dimension n(n-3)/2 for unordered edges. In a neural layer, A_{ij} can be a symmetric similarity score or symmetrized attention logit.
Implementation notes
Integrate this into a graph neural network or transformer attention block immediately after computing pairwise scores and before masking and softmax. For a batch of n valid tokens, compute symmetric logits A_{ij}=(q_i^T k_j+q_j^T k_i)/(2 sqrt(d)) for i unequal to j, set A_{ii}=0, and exclude padding tokens from every sum. Compute row sums r_i=sum_{j not equal to i} A_{ij} and total t=sum_i r_i. Form R_{ij}=A_{ij}-(r_i+r_j)/(n-2)+t/((n-1)(n-2)). Use R directly as the relational logit, or add a gated term to ordinary attention: L_{ij}=L^{vanilla}_{ij}+gamma R_{ij}, with gamma initialized to zero and learned. Apply the usual mask after constructing R. The paper's exact computation is the residualization formula and its zero-row-sum guarantee; no pseudoinverse is needed for pairwise arrays. Estimate only gamma, the temperature, and network parameters by backpropagation. First test a six-layer small ViT on CIFAR-10 or a graph-attention model on Cora, comparing vanilla attention, ordinary row-centering, and U-centering at matched FLOPs. Track validation accuracy, attention entropy, gradient-norm variance, and performance under duplicated or high-degree tokens. Success means improved accuracy or robustness at equal compute, fewer dominant-token failures, and stable loss without pathological attention concentration.
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.