从ResNet到mHC:DeepSeek重构残差连接,额外开销仅6.7%,附复现代码

news2026/3/29 18:01:12
2015年由微软亚洲研究院的何恺明团队提出ResNetResNet引入残差连接的概念用以解决深层神经网络训练中的梯度消失/爆炸和网络退化问题使得训练极深的网络成为可能。1(,)xl1​xl​F(xl​,Wl​)​​在公式1中∈1×xl​∈R1×d为 l层网络输入F为对 l层进行的非线性变换如卷积、Attention或MLP等1∈1×xl1​∈R1×d为该层的输出传统的网络试图对每层的输入 x 直接学习目标映射 ()H(x)。而残差网络的设计思想是既然直接学习很难不如让网络去学习每层的残差()()−F(x)H(x)−x​​公式1和公式2本质是一样的公式2中的 ()F(x)即为公式1的 F公式2中的()H(x)即为公式1的1xl1​公式2中的 x 即为公式1的xl​这就像是把原始文件x复印了一份直接交给下一个人同时附上一张便利贴上面写着这一层所做的修改()F(x)。下一个人收到的是「原件 修改意见」该设计的关键特性是恒等映射Identity Mapping能力。在网络初始化的早期阶段权重通常很小()≈0F(x)≈0。此时1≈0xl1​≈xl​0xl​​​这意味着信号像是在高速公路上一样毫无阻碍地从第一层直通最后一层。梯度也可以沿着这条高速公路无损地回传。正是这一特性使得训练成百上千层的网络如GPT-4, DeepSeek-V3成为可能。超连接Hyper-Connect标准的残差连接强制要求输入信号与经过变换的信号以 1:1 的比例叠加虽然保证了梯度的高速公路但也带来了两个问题信息流瓶颈原来只有一条残差通道所有信息不管是简单细节还是高层抽象都挤在同一条路上传。这就像所有车都走一条车道没法灵活分配路线模型没法根据需要把信息送到最合适的地方。表示坍塌网络特别深的时候为了不崩、训练稳定很多层其实学不到有用的新东西只能改一点点、几乎等于没改。结果就是白白浪费算力提取出来的特征都长得差不多没有多样性表达能力变弱。在上面的背景下 字节提出了Hyper-Connections的模型结构来改进传统的残差连接结构。通过扩展残差流宽度和多样化连接模式拓展了过去十年中广泛应用的残差连接范式。HC的核心思想是将原本单一维度的残差流扩展为4n4个并行的流然后乘以一个权重矩阵Hlres​。这增加了信息的带宽可以更多地捕获输入不同维度之间的融合信息。在HC架构中信息不再是简单的标量加法而是通过矩阵运算进行复杂的混合数学表达为1 (,)xl1​Hlres​xl​Hlpost T​Fl​(Hlpre​xl​,Wl​)​​对于公式4Hpre预融合/汇聚将4n4个并行流的信息汇聚起来压缩成适合Transformer层Attention或MLP处理的输入维度。Hpost后融合/分发将Transformer层的计算结果重新分发回4n4个并行流中。Hres流内混合这是最激进也是最关键的组件。它允许4n4个并行流在不经过Transformer层计算的情况下直接进行内部的信息交换和混合。理论上HC允许模型学习出任意的连接模式① 模型认为某一层应该保持恒等映射可以学习将Hres变为单位矩阵 I② 如果模型认为需要剧烈改变信息流向可以学习复杂的非对角矩阵。这种灵活性极大地增强了模型的拓扑结构复杂度。mHCHC的问题根据公式1传统的残差连接结构模型第L层和第l层的关系表示如下Σ−1(,)xL​xl​ΣilL−1​Fi​(xi​,Wi​)​​上式表明xl​的梯度信息可以1比1传递给xL​不会梯度爆炸或者梯度消失保证训练过程的稳定性。根据公式4可推导出HC结构下第L层和第l层的关系(∏1−1−) Σ−1(∏_1−1−−) (,)xL​(i1∏L−1​HL−ires​) xl​ΣilL−1​(∏_j1L−1−iHL−jres​) Hipost T​F(Hipre​xi​,Wi​)​​这会导致 ∂∂(∏1−1−)∂∂...∂xL​∂loss​(∏i1L−1​HL−ires​)∂xl​∂loss​...。由于 ∏1−1−∏i1L−1​HL−ires​一个典型的连乘过程矩阵连乘的性质取决于矩阵的谱范数即矩阵最大特征值的模。我们来比较Resnet中残差连接和HC连乘的情况Resnet中的残差连接残差路径是恒等映射相当于乘以单位矩阵I。单位矩阵的谱范数严格为 1。无论乘多少次1100111001信号始终稳定。HC在HC中Hres是自由学习的参数矩阵不可控。如果Hres的平均谱范数略大于 1例如 1.05在 60 层的网络中信号会被放大 1.0560≈18.61.0560≈18.6倍。如果谱范数更大或者网络更深放大倍数会呈指数级爆炸破坏残差结构中梯度回传的稳定性从而导致训练不稳定具体可见下图。上图显示27B模型训练到12k step左右HC的loss突然飙升。与此同时梯度范数也开始疯狂震荡。另外由于 −HL−ires​矩阵是无约束的可以往任意方向发散。论文测量了一个叫「Amax Gain Magnitude」的指标在残差流里被放大了3000倍。在大规模训练中这就是爆炸的前奏。mHC的诞生面对HC的问题DeepSeek提出了mHC即Manifold-Constrained Hyper-Connections流形约束超连接其核心思路为将不可控的Hres矩阵通过数学的手段转换为可控的双随机矩阵Doubly Stochastic Matrices。下面为大家解释mHC几个重要概念。双随机矩阵Doubly Stochastic Matrices双随机矩阵定义如下res(res):{res∈×∣res11,1_⊤ℎres1⊤, res≥0}PMres​(Hlres​):{Hlres​∈Rn×n∣Hlres​1n​1n​,1_n⊤mathcalHlres​1n⊤​, Hlres​≥0}​​其中11n表示所有元素均为 1 的n维向量。从公式7中可看出双随机矩阵的元素都大于等于0每一行、每一列的值相加都等于1。双随机矩阵具备2个重要特性① 范数值为1②多个双随机矩阵的乘积还是双随机矩阵。从而可以推出 ∣∏_1−1−∣1∣∏_i1L−1HL−ires​∣1 解决了上面提到的范数不可控的问题。那么怎么把_Hres_l矩阵变换成双随机矩阵呢DeepSeek团队采用的是Sinkhorn-Knopp算法。Sinkhorn-Knopp算法① 通过公式8计算变换前的Hlres​其中输入∈×,pre,post∈×,∈,2xl​∈Rn×C,φlpre​,φlpost​∈RnC×n,φlres​∈RnC,n2先将 xl​转换为 1×R1×nC的向量 ~x~l​并通过mat(·)操作从1×2R1×n2空间转换到×Rn×n。② 通过(0)(~)M(0)exp(H~lres​)得到元素值都大于0的矩阵作为迭代起始矩阵。③ 通过下面的公式迭代做normalization使其满足每行之和和每列之和接近1()(((−1)))M(t)Tr​(Tc​(M(t−1)))​​其中Tr​ 和Tc​分别代表按行和按列做归一化根据Sinkhorn-Knopp算法原理 ()M(t)会收敛成双随机矩阵DeepSeek论文中一般迭代20步。Birkhoff多胞体流形论文标题中的Manifold流形指的就是由所有双随机矩阵构成的几何空间被称为Birkhoff多胞体Birkhoff Polytope, 记为Bn​。多胞体Polytope想象一个多维空间中的多面体类似于3D空间中的钻石或立方体。这个多面体的每一个点都代表一个合法的双随机矩阵。顶点Vertices这个多面体的顶点是所有的置换矩阵Permutation Matrices。置换矩阵是只包含0和1的矩阵且每行每列只有一个1。它们的作用仅仅是交换信息的顺序比如把通道1的信息换到通道 2而不改变信息的大小。内部InteriorBirkhoff-von Neumann定理告诉我们这个多面体内部的任何一个点即任何一个双随机矩阵都可以表示为这些顶点的加权平均。DeepSeek的做法实际上是将神经网络原本在整个欧几里得空间中乱跑的参数强行拉回到了这个 Birkhoff多胞体的表面或内部。在这个几何体内游走无论怎么走都是安全的。关于mHC的更多知识可参考https://www.k-a.in/mHC-math.html、https://arxiv.org/pdf/2512.24880系统级实现与工程优化虽然Sinkhorn-Knopp迭代在理论上很美但在计算上却很昂贵。如果在每一层、每一步训练中都进行 20次矩阵迭代训练速度会大打折扣。DeepSeek为mHC量身定制了基础设施设计将额外开销仅增加6.7%。算子融合 (Kernel Fusion)DeepSeek利用了自研的TileLang编程语言开发了定制化的CUDA内核。融合操作将指数化、20次Sinkhorn迭代、以及后续的矩阵乘法全部融合进了一个单一的GPU Kernel中。SRAM驻留在这个Kernel执行期间中间数据如迭代过程中的矩阵一直保留在GPU的高速缓存SRAM/Register中而不需要反复写回慢速的全局显存。这大大减少了内存 I/O 次数使得 20 次迭代的计算时间几乎可以忽略不计。DualPipe通信重叠在大模型训练中由于模型太大往往需要跨多个GPU进行流水线并行Pipeline Parallelism。 DeepSeek设计了一种名为DualPipe的调度策略。打时间差当GPU的计算单元Tensor Cores正在全力计算mHC的Sinkhorn投影时GPU的通信单元NVLink并未闲着。重叠执行DualPipe巧妙地安排了任务利用mHC计算的时间窗口同时进行不同GPU之间的数据传输。结果Sinkhorn带来的额外计算延迟被通信时间完美掩盖了。对于整体训练流程来说mHC的计算几乎是免费的。选择性重计算Selective Recomputation由于mHC引入了 4n4 的扩展流中间激活值的显存占用会增加。如果全部存储会导致显存不足。DeepSeek 采用了选择性重计算策略在前向传播时不存储所有Sinkhorn迭代的中间结果。在反向传播时利用这一层极快的计算速度重新计算出所需的中间变量。这种以计算换显存的策略结合TileLang的高效率使得 mHC 在显存占用上也保持了高效。实验结果研究首先考察27B模型的训练稳定性与收敛性。如图5 (a) 所示mHC有效缓解了在HC中观察到的训练不稳定性与基线相比最终损失降低了0.021。图5 (b) 中的梯度范数分析进一步证实了这种稳定性的提升其中mHC表现出明显优于HC的行为其稳定的表现与基线相当。表 4展示了模型在多种下游基准测试中的性能表现。相比于基线模型mHC实现了全面的性能提升并在大多数任务上超过了HC。值得注意的是与HC相比mHC 进一步增强了模型的推理能力在 BBH上实现了2.1%的性能提升在 DROP上实现了2.3%的性能提升

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2462232.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…