cs.LGOct 5, 2026

The Birkhoff Geometry of Manifold-Constrained Hyper-Connections: Two Channels, Vertex Viscosity, and Sinkhorn as a Retraction

Authors: Xiaoyu Li, Zhizhou Sha, Chiwun Yang

Organizations: University of New South Wales · University of Texas at Austin · City University of Hong Kong

Abstract

Hyper-connections widen the residual stream of a Transformer to nn parallel streams. Their manifold-constrained version (mHC) mixes the streams at each layer with a doubly stochastic matrix, which it computes by Sinkhorn normalization of exponentiated logits. We give a geometric theory of this design on the Birkhoff polytope. First, a doubly stochastic mixer splits the stream into a mean channel, on which mHC is exactly a residual network, and a difference channel, which each layer contracts by its second singular value σ2≤1−nmin⁡ijHijσ_2 \le 1 - n \min_{ij} H_{ij}. Thus the extra width is a fading memory with a horizon of 1/(1−σ2)1/(1-σ_2) layers, and among nonnegative mixers only the permutations do not collapse. Second, the Sinkhorn-logit map is a global chart, and its logit gradient is exactly the Fisher-Rao gradient. Thus logit gradient flow follows a squared Fisher-Rao metric, and the straight-through update is exactly entropic mirror descent. Third, under logit gradient flow the logarithm of each entry moves at a rate of at most 4n3∥∇f∥∞ε4n^3\|\nabla f\|_\infty \varepsilon, where ε\varepsilon is the distance to the nearest permutation. Thus gradient flow approaches and leaves the vertices only at rate 1/t1/t, but mirror descent moves at an exponential rate. Fourth, the local convergence factor of Sinkhorn is σ22σ_2^2, so a fixed iteration budget limits the horizon. Experiments confirm the predicted rates.

Figures & tables

Appendix figures & tables5 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Accelerating Birkhoff Projection for Manifold-Constrained Hyper-Connections

    May 26, 2026Chenrui Wang, Yixuan QiuSolver IterationsProjection