mHC-lite: You Don't Need 20 Sinkhorn-Knopp Iterations
Not on the map. Why?
Abstract (arXiv)
Hyper-Connections (HC) generalizes residual connections by introducing dynamic residual matrices that mix information across multiple residual streams, accelerating convergence in deep neural networks. However, unconstrained residual matrices can compromise training stability. To address this, DeepSeek's Manifold-Constrained Hyper-Connections (mHC) approximately projects these matrices onto the Birkhoff polytope via iterative Sinkhorn--Knopp (SK) normalization. We identify two limitations of this approach: (i) finite SK iterations do not guarantee exact doubly stochasticity, leaving an approximation gap that can accumulate through network depth and undermine stability; (ii) efficient SK implementation requires highly specialized CUDA kernels, raising engineering barriers and reducing portability. Motivated by the Birkhoff--von Neumann theorem, we propose mHC-lite, a simple reparameterization that explicitly constructs doubly stochastic matrices as convex combinations of permutation matrices. This approach guarantees exact doubly stochasticity by construction and can be implemented using only native matrix operations. Extensive experiments demonstrate that mHC-lite matches or exceeds mHC in performance while achieving higher training throughput with a naive implementation and eliminating the residual instabilities observed in both HC and mHC. The code is publicly available at https://github.com/FFTYYY/mhc-lite.
Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). Builds on 13 of its 23 references that are in the map. First posted on arXiv 9 Jan 2026.
Builds on
13 of 23 references.
References · 13
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2016Identity Mappings in Deep Residual Networkscited by 1,965
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2015Highway Networkscited by 260
- 2024Hyper-Connectionscited by 84
- 20251000 Layer Networks for Self-Supervised RL: Scaling Depth Can Enable New Goal-Reaching Capabilitiescited by 45
- 2023ResiDual: Transformer with Dual Residual Connectionscited by 31
- 2018Better and Simpler Error Analysis of the Sinkhorn-Knopp Algorithm for Matrix Scalingcited by 14
- 2025Residual Matrix Transformers: Scaling the Size of the Residual Streamcited by 12
1 more
Built on by
Citing papers per year: 2026: 17.
17 papers.
Sort:
By area:
- 2026Attention Residualscited by 65
- 2026KromHC: Manifold-Constrained Hyper-Connections with Kronecker-Product Residual Matricescited by 11
- 2026Spectral-Sphere-Constrained Hyper-Connectionscited by 4
- 2026go-mHC: Direct Parameterization of Manifold-Constrained Hyper-Connections via Generalized Orthostochastic Matricescited by 4
- 2026Analyzing Stream Collapse in Hyper-Connections: From Diagnosis to Mitigationcited by 4
- 2026xHC: Expanded Hyper-Connectionscited by 4
- 2026A Unified Foundation Model for All-in-One Multi-Modal Remote Sensing Image Restoration and Fusion with Language Promptingcited by 3
- 2026TBP-mHC: full expressivity for manifold-constrained hyper connections through transportation polytopescited by 3
- 2026Multi-Gate Residualscited by 1
- 2026Manifold-Constrained Hyper-Connections for Parameter-Efficient Finetuningcited by 1
- 2026TEMPER: Tensorized Efficient Manifold-constrained Parameterization for Expressive Residual Routingcited by 1
- 2026oHC: Orthogonal Hyper-Connections on SO(4) via Quaternionscited by 1
5 more
- 2026CliffSearch: Structured Agentic Co-Evolution over Theory and Code for Scientific Algorithm Discoverycited by 0
- 2026The Birkhoff Geometry of Manifold-Constrained Hyper-Connections: Two Channels, Vertex Viscosity, and Sinkhorn as a Retractioncited by 0
- 2026Multi-Stream Attention Residualscited by 0
- 2026Multi-Stream Residual Computation in Diffusion Language Modelscited by 0
- 2026Depth Memorycited by 0
- 2026Depth Memory13 Oct 2026
- 2026Multi-Stream Residual Computation in Diffusion Language Models13 Oct 2026
- 2026Multi-Stream Attention Residuals9 Oct 2026
- 2026The Birkhoff Geometry of Manifold-Constrained Hyper-Connections: Two Channels, Vertex Viscosity, and Sinkhorn as a Retraction5 Oct 2026
- 2026oHC: Orthogonal Hyper-Connections on SO(4) via Quaternions2 Sep 2026
- 2026TEMPER: Tensorized Efficient Manifold-constrained Parameterization for Expressive Residual Routing8 Aug 2026
- 2026Manifold-Constrained Hyper-Connections for Parameter-Efficient Finetuning20 Jul 2026
- 2026xHC: Expanded Hyper-Connections16 Jul 2026
- 2026Analyzing Stream Collapse in Hyper-Connections: From Diagnosis to Mitigation2 Jun 2026
- 2026Multi-Gate Residuals22 May 2026
- 2026TBP-mHC: full expressivity for manifold-constrained hyper connections through transportation polytopes20 May 2026
- 2026A Unified Foundation Model for All-in-One Multi-Modal Remote Sensing Image Restoration and Fusion with Language Prompting7 Apr 2026
5 more
- 2026go-mHC: Direct Parameterization of Manifold-Constrained Hyper-Connections via Generalized Orthostochastic Matrices2 Apr 2026
- 2026CliffSearch: Structured Agentic Co-Evolution over Theory and Code for Scientific Algorithm Discovery1 Apr 2026
- 2026Spectral-Sphere-Constrained Hyper-Connections21 Mar 2026
- 2026Attention Residuals16 Mar 2026
- 2026KromHC: Manifold-Constrained Hyper-Connections with Kronecker-Product Residual Matrices29 Jan 2026
- 2026xHC: Expanded Hyper-Connectionscited by 4
- 2026xHC: Expanded Hyper-Connections16 Jul 2026
- 2026Attention Residualscited by 65
- 2026KromHC: Manifold-Constrained Hyper-Connections with Kronecker-Product Residual Matricescited by 11
- 2026Spectral-Sphere-Constrained Hyper-Connectionscited by 4
- 2026go-mHC: Direct Parameterization of Manifold-Constrained Hyper-Connections via Generalized Orthostochastic Matricescited by 4
- 2026Analyzing Stream Collapse in Hyper-Connections: From Diagnosis to Mitigationcited by 4
- 2026A Unified Foundation Model for All-in-One Multi-Modal Remote Sensing Image Restoration and Fusion with Language Promptingcited by 3
- 2026TBP-mHC: full expressivity for manifold-constrained hyper connections through transportation polytopescited by 3
- 2026Multi-Gate Residualscited by 1
- 2026Manifold-Constrained Hyper-Connections for Parameter-Efficient Finetuningcited by 1
- 2026TEMPER: Tensorized Efficient Manifold-constrained Parameterization for Expressive Residual Routingcited by 1
- 2026oHC: Orthogonal Hyper-Connections on SO(4) via Quaternionscited by 1
- 2026The Birkhoff Geometry of Manifold-Constrained Hyper-Connections: Two Channels, Vertex Viscosity, and Sinkhorn as a Retractioncited by 0
12 most cited of 14 All 14 (JSON)
- 2026Depth Memory13 Oct 2026
- 2026Multi-Stream Attention Residuals9 Oct 2026
- 2026The Birkhoff Geometry of Manifold-Constrained Hyper-Connections: Two Channels, Vertex Viscosity, and Sinkhorn as a Retraction5 Oct 2026
- 2026oHC: Orthogonal Hyper-Connections on SO(4) via Quaternions2 Sep 2026
- 2026TEMPER: Tensorized Efficient Manifold-constrained Parameterization for Expressive Residual Routing8 Aug 2026
- 2026Manifold-Constrained Hyper-Connections for Parameter-Efficient Finetuning20 Jul 2026
- 2026Analyzing Stream Collapse in Hyper-Connections: From Diagnosis to Mitigation2 Jun 2026
- 2026Multi-Gate Residuals22 May 2026
- 2026TBP-mHC: full expressivity for manifold-constrained hyper connections through transportation polytopes20 May 2026
- 2026A Unified Foundation Model for All-in-One Multi-Modal Remote Sensing Image Restoration and Fusion with Language Prompting7 Apr 2026
- 2026go-mHC: Direct Parameterization of Manifold-Constrained Hyper-Connections via Generalized Orthostochastic Matrices2 Apr 2026
- 2026Spectral-Sphere-Constrained Hyper-Connections21 Mar 2026
12 newest of 14 All 14 (JSON)