mlsanityData to 14 Oct 2026

mHC-lite: You Don't Need 20 Sinkhorn-Knopp Iterations

arXiv 2601.05732 ↗First posted 9 Jan 2026cs.LG cs.AIcited by 17

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2020Language Models are Few-Shot Learnerscited by 32,419
  4. 2016Densely Connected Convolutional Networkscited by 6,720
  5. 2016Identity Mappings in Deep Residual Networkscited by 1,965
  6. 2019Root Mean Square Layer Normalizationcited by 1,480
  7. 2015Highway Networkscited by 260
  8. 2024Hyper-Connectionscited by 84
  9. 20251000 Layer Networks for Self-Supervised RL: Scaling Depth Can Enable New Goal-Reaching Capabilitiescited by 45
  10. 2023ResiDual: Transformer with Dual Residual Connectionscited by 31
  11. 2018Better and Simpler Error Analysis of the Sinkhorn-Knopp Algorithm for Matrix Scalingcited by 14
  12. 2025Residual Matrix Transformers: Scaling the Size of the Residual Streamcited by 12
1 more
  1. 2025Thoughtbubbles: an Unsupervised Method for Parallel Thinking in Latent Spacecited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 17.

17 papers.

Sort:

By area: