arXiv:2601.21579cs.CLcs.LG2026-01被引 9

用克罗内克积构建双随机残差矩阵,解决超连接网络训练不稳和参数爆炸问题。

KromHC: Manifold-Constrained Hyper-Connections with Kronecker-Product Residual Matrices

  • 通过克罗内克积分解小双随机矩阵,构造残差连接
  • 理论保证双随机性,参数量降至O(n²C)(原为O(n³C))
  • 在相同性能下参数减少90%以上,适合轻量化模型部署

超连接(HC)在神经网络中表现优异,但存在训练不稳和扩展性受限的问题。曼ifold约束超连接(mHC)通过将残差空间投影到Birkhoff多面体缓解此问题,但其迭代Sinkhorn-Knopp算法无法保证精确双随机性,且参数复杂度高达O(n³C),其中n为残差流宽度,C为特征维度。最近的mHC-lite虽通过Birkhoff-von-Neumann定理重构残差矩阵以确保双随机性,却面临因子阶乘级的参数爆炸,复杂度达O(nC·n!)。本文提出KromHC,利用更小双随机矩阵的克罗内克积参数化残差矩阵,并沿张量化残差流各模式施加流形约束,确保残差矩阵精确双随机性,同时将参数复杂度降低至O(n²C)。实验表明,KromHC在性能上匹配或超越现有SOTA mHC变体,但可训练参数显著减少。

原文摘要 · Abstract (English)

The success of Hyper-Connections (HC) in neural networks (NN) has also highlighted issues related to training instability and restricted scalability. The Manifold-Constrained Hyper-Connections (mHC) mitigate these challenges by projecting the residual connection space onto a Birkhoff polytope, however, it faces two issues: 1) its iterative Sinkhorn-Knopp (SK) algorithm does not always yield exactly doubly stochastic residual matrices; 2) mHC incurs a prohibitive $O(n^3C)$ parameter complexity with $n$ as the width of the residual stream and $C$ as the feature dimension. The recently proposed mHC-lite reparametrizes the residual matrix via the Birkhoff-von-Neumann theorem to guarantee double stochasticity, but also faces a factorial explosion in its parameter complexity, $O \left( nC \cdot n! \right)$. To address both challenges, we propose KromHC, which uses the Kronecker products of smaller doubly stochastic matrices to parametrize the residual matrix in mHC. By enforcing manifold constraints across the factor residual matrices along each mode of the tensorized residual stream, KromHC guarantees exact double stochasticity of the residual matrices while reducing parameter complexity to only $O(n^2C)$. Experiments show that KromHC matches or even outperforms other state-of-the-art (SOTA) mHC variants, while requiring significantly fewer trainable parameters. The code is at https://github.com/wz1119/KromHC.

神经网络残差连接参数压缩双随机矩阵

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。