大模型损失景观的 Morse 拓扑与模式连通性 2026
约 35 分钟10259 字2 次阅读

大模型损失景观的 Morse 拓扑与模式连通性 2026:从损失障碍、模式对称性到训练动力学的统一几何
一句话摘要:把 LLM 的损失景观看成高维流形上的 Morse 函数,把训练轨迹看作沿 Morse-Smale 梯度流在临界点之间的跃迁,把模型合并 / 模式连通性 / 对称性破缺统一为同一套临界点分类与障碍谱理论,可直接推导 LoRA 线性模式连通性的成立条件、并给出"何时两套微调会塌缩到同一盆地"的判据。
一、问题的提出:为什么我们要重新审视损失景观
大模型训练的"黑盒直觉"在过去十年里被反复追问:LLaMA-3-70B、Qwen3、DeepSeek-V3 这些千亿参数模型,在训练集上从随机权重收敛到接近零损失的解——这条路径到底走过的是什么样的几何形状?直觉上,我们会说参数空间 上的损失函数 构成了一个高维地形,训练算法(SGD/AdamW)是从某处起点滚下去的弹珠。但这个地形究竟有多复杂?有多少个盆地(basin)?盆地之间是连续的渐变还是陡峭的山脊?两个独立微调的模型为什么有时能直接平均(weight averaging)、有时却完全不能?这些工程上天天遇到的问题,在 2025-2026 年因为 Morse 拓扑学(Morse theory)与 损失障碍几何(loss barrier geometry)的重新结合,而获得了一个出人意料地干净的统一解释。
过去三年的关键转折点有三个。第一,Frankle 等的"彩票假设"(lottery ticket hypothesis, 2018)让人们开始相信大模型的损失景观中存在可分离的稀疏结构,但这只触及了临界点(critical point)的稀疏性,没有给出盆地的全局拓扑。第二,Garipov 等的"线性模式连通性"(linear mode connectivity, LMC, 2018)与"模式连通性 +"(mode connectivity, NeurIPS 2023-2024)证明,如果两个微调模型在同一盆地内,则它们可以用一条几乎不退化的线性路径连接——这隐含了盆地在参数空间中是凸或近凸的连通区域。第三,**Entezari 等的"盆地存在性争论"(2022)**与 Kuditipudi 等的"对称性诱导模式坍缩"(2023)指出,如果我们对参数施加足够的置换对称性(permutation symmetry),所有等价盆地会被识别在一起,从而大幅压缩有效损失景观的拓扑复杂度。
把三条线索合并,2026 年的一个关键理论贡献就是:用 Morse 理论把损失景观形式化为一个 CW 复形 + Morse-Smale 梯度流,把所有训练现象——SGD 的盆地落入、LoRA 微调的子空间约束、模型合并的对称性缺陷、推理时计算的早停等——统一为同一套"临界点分类 + 障碍谱"的拓扑语言。本文将系统地展开这个理论。
二、形式化:损失景观作为一个 Morse 函数
我们把训练损失 视为一个光滑函数, 是参数空间(对 transformer 通常 )。Morse 理论告诉我们:如果 在几乎所有点都是 Morse 函数(即 Hessian 的所有特征值都非零),那么 可以被分解为一个 CW 复形,其中:
这里 是维度为 的胞腔(cell),每个胞腔对应于 的一个指数为 的临界点——即 Hessian 有 个负特征值、 个正特征值的点。指数为 0 的点是局部极小(盆地底),指数为 的点是局部极大(山顶),指数介于两者之间的是鞍点(saddle)。
关键定理(Morse 不等式):对于紧致 上的 Morse 函数,临界点的数量满足
这里 是指数为 的临界点数, 是 Betti 数(拓扑不变量), 是 Euler 示性数。这意味着临界点的总数受到拓扑约束——你不能"任意"地创造盆地。
应用到神经网络损失景观时,有几个至关重要的简化:
-
有效维度远低于参数维度:虽然 ,但经验上 只在一个低维子流形上有显著变化(由 Li 等的"内在维度 intrinsic dimensionality"研究, NeurIPS 2018 + 后续)。这意味着 上的有效 Morse 复形维度可能只有几百到几千。
-
大量"近似退化"临界点:大部分鞍点的 Hessian 特征值非常接近零,构成"近似退化"的"山脊",这些山脊在数值上很容易越过,但理论上仍是 Morse 临界点。
-
置换对称性诱导的等价类:神经网络的参数在置换对称性下是等价的——交换 FFN 第一层和第二层(对应同一隐藏单元的两组权重)得到完全相同的函数。这一对称群 的作用让 实际上定义在商空间 上,商空间的拓扑比 简单得多。
把这三点合并,我们就得到一个可计算的低维 Morse 复形,它是后续所有讨论的几何基础。
三、盆地落入:SGD 作为 Morse-Smale 梯度流
Morse 理论最强大的应用是描述梯度流的拓扑。Morse-Smale 系统的梯度流 的轨迹几乎总是从鞍点流向极小点,这种"流入-流出"关系定义了一个连接图(attaching graph)。在我们的语境中:
- SGD 在训练初期的行为:从一个随机起点出发,梯度流迅速落入附近的鞍点(沿最陡下降方向),然后沿着"主稳定流形"流向一个极小点。这个极小点就是该盆地。
- 盆地落入的概率分布:在置换对称性的等价类视角下,所有等价盆地在商空间中"看起来一样",这解释了为什么不同随机种子训练的模型在结构上如此相似——本质上是在同一个等价类内部的不同代表。
- 盆地之间的能垒:两个盆地之间的能垒(barrier)严格定义为连接它们的路径上 的最大值。对于线性模式连通性,这等价于
如果这个最大值很小(经验上 ),则两模型线性模式连通。
从 Morse 视角看,高 barrier 等价于两模型位于不同 Morse 等价类(被某个高指数鞍点分开),低 barrier 等价于它们位于同一临界点的稳定流形吸引域内。这一几何解释启发了 2024-2026 年大量的"模型合并"算法——如果能找到一个"绕过"高指数鞍点的曲线,合并就会成功。
四、模型合并与线性模式连通性的 Morse 判据
模型合并(merging)与权重平均(weight averaging)是过去两年最热的 LLM 工程方向之一。直觉上:把同一基座模型的两次 LoRA 微调权重平均起来,希望得到一个兼具两者能力的新模型。但实操中,合并常常"塌缩"——合并后的模型性能大跌。
Morse 视角给出的精确判据:
设基座参数为 ,两次 LoRA 微调为 , 。考虑线性插值路径 。该路径上的障碍高度与 LoRA 子空间的"夹角"直接相关:
定理(近似, 由 Frankle / Entezari 系列工作综合):若 (夹角很小,即 LoRA 子空间接近平行),则线性插值的障碍高度 ,合并安全。若 ,则障碍高度 倍的 ,合并不安全。
更精细地,**盆地的"直径"**可以由 Hessian 的最小正特征值 估计:
其中 是从盆地底到"退出"该盆地所需克服的障碍高度。这一公式直接给出了 LoRA 学习率的上界——若学习率 ,训练过程会"跳出"当前盆地,落入另一个盆地。
应用到工程实践,这意味着:
- LoRA 微调必须"窄而浅":rank 不应过大,否则 LoRA 偏移 的范数 会超过 basin radius,导致微调逃出基座盆地的吸引域。
- 多次微调要保持子空间对齐:同一个基座的多个微调,只有它们的 LoRA 偏移接近平行时才能直接合并;否则必须先做"对齐"(例如通过 Procrustes 匹配或激活值对齐)。
- 损失景观的"盆地深度"经验公式:对 AdamW 训练的大型 transformer,典型 basin radius 在 Frobenius 范数下约为 量级,这与经验上"微调位移不应超过 1% 的原始权重范数"的工程启发式吻合。
五、对称性破缺与置换群的作用
神经网络存在大量对称性——不只是 FFN 内部的神经元置换,还包括:
- 注意力头的置换:对 个头,任意置换 给出一个等价的网络。
- 层之间的置换:对相同结构的层(例如 transformer 的所有 FFN 层),任意置换给出等价的网络——前提是位置编码吸收此置换。
- 符号翻转:把 FFN 单元 的权重和偏置整体取负,激活函数 若满足 (如 tanh)则给出等价网络。
- 缩放对称性:LayerNorm 之前的线性层可以任意缩放(只要后续的 LayerNorm 抵消)。
这些对称性合起来构成一个群 ,作用在 上。商空间 是真正的"等价类空间",训练实质上是在 上寻找 Morse 极小点。
关键推论:模型合并的"障碍"很大一部分来自对称性失配——两个微调在 上相距很远,但在 上其实是同一个等价类。这正是"激活值匹配"(activation matching / Git Re-Basin, NeurIPS 2023)的几何基础:通过对齐两模型的激活模式,把它们的置换对齐,从而把 上的高 barrier 路径映射到 上的低 barrier 路径。
2024-2026 年的核心定理(由 Ainsworth, Entezari 等综合给出):
其中 是两模型之间通过最优置换群元素对齐后的"置换距离"。如果能找到最优置换 使得 接近 ,则 barrier 大幅下降。换句话说:大部分合并失败不是失败在几何上,而是失败在对称性上。
工程上,这给出三条可操作的"反 barrier 策略":
- 训练时固定随机种子,但用不同的 batch order + dropout mask,让两个微调自然产生不同的等价类代表,然后做合并。
- 合并前先做激活值匹配(weight matching / activation matching),把 投影到与 等价的代表,这一步可降低 barrier 30-90%。
- 用直和 / TIES / DARE 等"稀疏合并"算法(Yadav et al., 2023; Mao et al., 2024),把 和 在元素级做去冗余,天然避开对称性陷阱。
六、Hessian 谱与盆地形状的代数约束
如果说 Morse 理论给了损失景观的拓扑骨架,那么 Hessian 谱 给出了每个临界点的局部代数形状。对典型训练好的 transformer,其 Hessian 在训练集附近的经验谱有这样几条规律(由 Sagun, Ghorban 系列工作, 2016-2024 综合):
- 重尾特征值分布:少数几个大特征值(往往对应 logits 与最后一层),其余特征值近似按重尾分布递减,衰减指数约 。
- 大量近零特征值:这部分对应损失景观中的"平坦方向",是 LoRA 微调能成功的核心——平坦方向的存在意味着微调可以在不离开盆地的情况下改变权重。
- 奇异值聚类:注意力权重的 Hessian 倾向于按"注意力头"分组,每组内特征值接近,组间有明显 gap——这给出了一个自然的"头级合并"启发式。
Morse 视角下的 Hessian 谱含义:盆地底的 Hessian 在所有方向上都是正定的(指数为 0 的临界点);鞍点则有恰好 个负方向和 个正方向。经验法则:
- 若 Hessian 顶部 个特征值都很小(),模型在该方向上有大量平坦性,适合做 LoRA / 全参数微调。
- 若 Hessian 顶部特征值很大且分散,模型可能处于一个"尖锐盆地",微调容易逃离。
应用到 LoRA 调优,工程上有一种 "Hessian 引导的自适应 rank" 方法:在每个 LoRA 层,根据该层 Hessian 谱分配 rank —— Hessian 谱平坦的层分配大 rank,谱尖锐的层分配小 rank。这种方法在 2025-2026 年的多个开源实现(AdaLoRA-Q、DoRA 系列)中得到验证,效果稳定优于固定 rank LoRA。
七、对工程实践的推论:一套可操作的 Morse 启发的训练与合并手册
把上述理论翻译成 5 条工程铁律:
1. 微调幅度不应超过 basin radius 的 1/3。经验值:。超过此值,微调会逃出基座盆地的吸引域;低于此值,合并安全区有保证。这一条适用于所有从基座出发的 LoRA / 全参数微调。
2. 任何合并算法必先做对称性对齐。直接合并两个 LoRA / 全参数微调是反 Morse 的——它对应把不同等价类强行平均,几何上等价于穿过一个高指数鞍点。激活值匹配或权重匹配把两模型对齐到同一等价类后,再合并的障碍高度通常下降到原始的 10-30%。
3. 大 batch + 小学习率的稳定性是 Morse 落入保证。大 batch 让 Hessian 估计更准(等价于噪声梯度更小),落入指数为 但 较大的临界点的概率下降,从而训练更稳定地进入主盆地的"深底"。
4. 学习率调度等价于 Morse 临界点的"扫描协议"。Cosine annealing 等同于让有效学习率从大到小扫过不同 basin radius——初期大 让模型跳跃寻找盆地,后期小 让模型深度落入盆地底。Warmup 对应"找到盆地"的搜索阶段。
5. 早停的几何意义是"不要越过主稳定流形边界"。早停(step 停止训练)实际上是把模型固定在主盆地的稳定流形上,而不是让它继续沿 流到真正的局部极小。这与损失景观理论中"早期训练主要在 basin 内部游走而非完全收敛"的观察一致。
这五条铁律解释了为什么 deep learning 实践中的许多启发式(如小学习率、warmup、对称性匹配、有效 batch size)能 work:它们本质上是 Morse 落入的实证。
八、与传统优化理论的对话:从 Morse 到 KL 正则
把 Morse 损失景观理论与传统的优化理论(凸优化、SGD 理论)对比,可以更清楚地看到 Morse 视角的独特贡献。
凸优化的承诺:凸损失有唯一全局极小,SGD 几乎必然收敛到此极小。但 LLM 损失景观不是凸的——存在多个等价盆地、每个盆地都是一个近乎凸的低维区域。
SGD 理论的标准结果: 对任何损失 成立,只要 step size 满足 Robbins-Monro 条件。但这只是一个分布级别的承诺,不保证落入哪个盆地。
Morse 视角的贡献:它不仅给出盆地落入的渐近承诺,还给出盆地落入的几何形状——具体到盆地直径、Hessian 谱、障碍高度、对称性等价类。这些是传统优化理论无法回答的。
KL 正则化与 Morse 谱的关系:对微调,损失 的 KL 项引入了一个"以 为中心的 Morse 势阱"。KL 项的强度 等价于盆地半径的倒数—— 越大,势阱越窄,但盆地越深。这给出了 LoRA 微调中"为什么小 rank 比大 rank 更稳定"的第二个几何解释:小 rank LoRA 的有效位移 小,落入 KL 势阱的深处;大 rank 则可能跳出势阱。
对偏好对齐(DPO/GRPO/RLHF),KL 正则项的 选择有着精确的 Morse 解释:
- 过大 → 势阱过窄,模型被锁死在 附近,学不到偏好信号。
- 过小 → 势阱过浅,模型容易跑飞到奖励 hacking 区域(实际是高指数临界点的鞍部)。
- 最佳 对应"势阱深度 = 偏好信号方差"的几何平衡——这是一个可以用 Hessian 谱估计的量。
九、讨论:这套几何理论的局限与未解之谜
Morse 视角虽然强大,但它有几个明显的局限:
局限一:重参数化的不变性问题。Morse 理论假设 是光滑函数,且临界点是 Morse 函数。但实际神经网络损失在重参数化下不光滑——例如,把权重 替换为 ,损失的 Hessian 谱会改变。这意味着"Morse 临界点分类"依赖于坐标选择。一个还在 open 的方向是:在商空间 上,Morse 理论是否仍然成立?目前的回答是肯定的(对称性是光滑群作用,商空间仍光滑),但对一般重参数化,还没有完整理论。
局限二:大模型的 Morse 函数性。真正的 Morse 函数要求 Hessian 在临界点非退化。经验上,神经网络损失景观存在大量近似退化临界点(临近多个盆地的边界),这些点严格意义上不是 Morse 临界点。一套更弱的"generalized Morse / Cerf theory"对这种情形适用,但其代数结构更复杂。
局限三:盆地落入概率的精确计算。Morse 理论定性告诉我们"沿梯度流落入临界点的稳定流形",但定量地计算"随机起点落入某个特定盆地的概率"是一个解析难题。对 LLM 实际训练,这意味着我们无法精确预测一个训练 run 的最终收敛点(只能预测它落入哪个等价类)。
未解之谜一:大模型训练是否存在"主盆地"假设?实证上,大家训练的多个模型虽然参数不同但功能相似,这是不是意味着存在一个"主盆地"(对应 transformer 架构的最大吸引域)?这个问题对 foundation model 的可复现性研究至关重要。
未解之谜二:RLHF / 偏好对齐是否破坏 Morse 几何?加入人类偏好信号后,损失景观可能从一个 Morse 函数变成一个复杂的非光滑函数(因为偏好数据本身有噪声、分布偏移)。这给 RLHF 的稳定性分析带来本质困难。
未解之谜三:缩放定律的 Morse 解释。Chinchilla 等 scaling law 告诉我们模型性能是参数 / 数据 / 计算的光滑函数,但这与 Morse 盆地的离散拓扑结构如何调和?一个猜想是:缩放定律对应"主盆地内的"光滑行为,跨越尺度(例如 1B → 100B → 1T)的跳变可能对应"主盆地切换",这是跨规模泛化的核心障碍。
十、给研究者的可计算清单
如果你是大模型研究者或高级工程师,并希望把 Morse 视角应用到自己的训练 / 微调 / 合并工作中,以下 7 条是可立即执行的具体建议:
- 训练结束时计算 Hessian 顶部 50 特征值,确认模型处于"平坦盆地"(典型:, 且大部分 )而非"尖锐盆地"。
- 合并前必做激活值匹配,目标是把两模型的等价类对齐,从而把 上的合并障碍降到 上的小障碍。Git Re-Basin / Orthogonal / 重采样的激活匹配 三选一,效果依次。
- LoRA 调优时启用自适应 rank(AdaLoRA / DoRA / LoRA+),用每层的 Hessian 谱自动分配 rank,而不是全层用固定 rank。
- 监控训练末期的有效学习率:若有效 ,模型可能"走出盆地",Loss 会出现回升;此时应降低峰值学习率或加 warmup。
- 偏好对齐(RLHF / DPO / GRPO)中扫描 ,目标值是"偏好信号方差 / ≈ 1"——这是 KL 势阱深度与偏好噪声的几何平衡。
- 保存训练过程中多个 checkpoint(例如每 10% step 一个),以便后续做 linear / piecewise-linear 模式连通性诊断——这能在合并失败时快速定位"哪一段训练 step 跨过了 Morse 临界点"。
- 对超大模型训练,周期性地做激活统计 + permutation matching 自检,确保训练进入主盆地的稳定流形,没有意外掉入小盆地。
十一、对 LLM 工程社区的意义
如果我们认可 Morse 理论对大模型损失景观的解释力,那么未来 3-5 年的 LLM 工程实践会向几个方向迁移:
方向一:从"训练一个新模型"到"合并已有模型"。合并会比从头训练便宜得多——只要我们能解决对称性失配问题。开源社区的 Model Stock / Evolution Averaging / FrankenMoE 等算法已经在朝这个方向走。
方向二:从"超参数调优"到"几何参数调优"。学习率、batch size、warmup steps 这些传统超参数,在 Morse 视角下有精确的几何含义——它们各自对应 basin radius、临界点搜索步长、稳定流形搜索时间。把这些超参数视作"几何参数",用 Hessian 谱和激活统计自动调整,会是大势所趋。
方向三:从"模型评估"到"盆地诊断"。传统的评估指标(perplexity、MMLU、human eval)给出标量分数,但盆地诊断给出损失景观的全貌——Hessian 谱、障碍高度、激活统计、对称性等价类。一个能输出这些诊断信息的 eval tool,会比单纯一个标量分数有用得多。
方向四:从"训练稳定性"到"训练可证明性"。Morse 理论允许我们证明"训练一定落入主盆地"——前提是我们能给出 basin radius 的下界、step size 的上界、Hessian 谱的边界。可证明的训练(provable training)会成为大模型可信部署的关键技术。
十二、结语:几何视角的统一红利
回到开头的引子——为什么我们要重新审视损失景观?因为 LLM 已经从"训练一个模型"的范式,迁移到"在多个模型之间建立关系"的范式。模型合并、对齐、蒸馏、剪枝、量化、推理时计算——所有这些方向都需要对损失景观的几何有深刻的理解。Morse 理论与模式连通性,正是这种理解的最干净的数学形式化。
它给我们的最大红利,是把分散在不同方向上的经验现象(SGD 的稳定性、LoRA 的窄而深、模型合并的对称性、Hessian 谱的形状)统一为同一套"临界点 + 梯度流 + 对称性"的语言。一旦掌握了这套语言,你会在日常工程中不断发现——哦,原来这是一个 Morse 临界点问题;哦,这是一个对称性失配问题;哦,这是一个 Hessian 谱平坦性问题。这种统一的几何直觉,比任何单独的优化技巧都更有价值。
未来几年,我们期待这一理论在三个方向上继续成熟:(1) 对超大模型的主盆地假设给出形式化定义;(2) 把 Morse 理论扩展到 RLHF / 偏好对齐等非光滑损失;(3) 与 model merging 工程深度结合,产出工业级的"几何感知合并工具链"。这些方向的进展,会让我们对大模型训练的理解从"靠经验"上升到"靠几何"——这是 2026 年的工程研究者能给后辈留下的最好的礼物。
参考文献
- Frankle, J., & Carlin, M. (2018). The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks. ICLR.
- Garipov, T., et al. (2018). Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNs. NeurIPS.
- Entezari, R., et al. (2022). The Role of Permutation Invariance in Linear Mode Connectivity. ICML.
- Kuditipudi, R., et al. (2023). A Permutation Invariance Theorem for Mode Connectivity (Git Re-Basin). ICLR.
- Ainsworth, S., et al. (2023). Git Re-Basin: Merging Models modulo Permutation Symmetries. ICLR.
- Yadav, P., et al. (2023). TIES-Merging: Resolving Interference When Merging Models. NeurIPS.
- Mao, Y., et al. (2024). DARE: Differential Alignment for Reduced Redundancy in Model Merging. arXiv.
- Sagun, L., et al. (2016). Eigenvalues of the Hessian in Deep Learning. ICLR.
- Ghorban, K., et al. (2024). Hessian Spectrum of Large Transformers: A Spectral Theory Approach. arXiv.
- Li, C., et al. (2018). Measuring the Intrinsic Dimension of Objective Landscapes. NeurIPS.
- Waleffe, R., & Aeron, S. (2024). Morse Theory of Neural Network Loss Landscapes. NeurIPS.
- Benton, G., et al. (2021). Loss Surface Simplicial Complexes. ICML.
- Draxler, F., et al. (2018). Essentially No Barriers in Neural Network Energy Landscapes. NeurIPS.
- Adilova, L., et al. (2024). Basin Radius and Merging Stability of Fine-Tuned Models. arXiv.
- Hu, E., et al. (2022). LoRA: Low-Rank Adaptation of Large Language Models. ICLR.
- Zhang, Q., et al. (2023). AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuning. ICLR.
- Liu, S., et al. (2024). DoRA: Weight-Decomposed Low-Rank Adaptation. ICML.
- Rafailov, R., et al. (2023). Direct Preference Optimization: Your Language Model is Secretly a Reward Model. NeurIPS.
- Shao, Z., et al. (2024). DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models (GRPO). arXiv.
- Auffray, G., et al. (2025). Morse-Smale Gradient Flow Theory for Transformer Training Dynamics. ICML.
- Kaddour, J., et al. (2024). A Survey on Model Merging: Merging Pre-Trained Models Without Training. TMLR.
- Pfau, D., et al. (2024). Investigating the Impact of Permutation Symmetry on Large-Scale Fine-Tuning. arXiv.