Effect of the block-orthogonal multi-head constraint

Determine whether enforcing mutual orthogonality among all transformer attention heads through the block-orthogonal Stiefel variant improves or degrades performance relative to independent Stiefel frames.

Background

The paper introduces a block-orthogonal multi-head extension that stacks the query frames of all heads into a single orthogonal matrix, thereby forcing every head subspace to be orthogonal to every other head subspace. This is a stronger constraint than maintaining independent Stiefel frames and is motivated by the observed redundancy among attention heads.

Although the variant is implemented and unit-tested, it is not used to produce the reported experimental results. Its effect on model performance therefore remains unresolved: the stronger constraint could either help by preventing redundancy or hurt by restricting useful overlap between heads.

References

The block-orthogonal variant is implemented and unit-tested but is not what produced the reported accuracies, and we have not measured whether the stronger constraint helps or hurts.

Stiefel Attention: When the Geometry of Transformer Projection Matrices Dominates Optimizer Choice---and When It Does Not  (2609.19363 - Guerrero, 16 Sep 2026) in Section 2, subsection “Block-Orthogonal Multi-Head Extension”