面向持续学习:通过子空间调优减轻遗忘
🧭 持续学习 · 子空间约束微调 ★ 深度解读
UC Berkeley · Mila · McGill University · Université de Montréal · UMass Amherst
通过将权重更新限制在低能量奇异子空间,SST 有效减轻 LLM 微调中的灾难性遗忘
📌 工作参考价值(4 点)
- 在业务中微调 LLM 以适应新领域或任务时,可引入 SST 保护已有通用能力,避免上线后基础能力下降,尤其适用于需要持续更新的推荐/广告模型。
- SST 与优化器无关,只需对优化器最终更新做投影,可直接嵌入现有 SFT/RLHF 流程,无需修改损失函数或数据,工程实现成本低。
- 保护主导奇异方向可显著减少对基模型的 KL 漂移(2-4 倍),有助于保持输出分布的稳定性,对在线学习或 A/B 测试更友好。
- 在顺序学习多领域任务时,SST 能够累积能力而非互相覆盖,适合构建多能力融合的模型或专家合并场景。
DeltaMomentum:基于键值的各向异性动量更新
🧭 优化器动量更新 · Delta Rule
Electrical and Computer Engineering, Carnegie Mellon University
把线性层梯度拆成 key-value,用 delta rule 做方向感知动量更新来加速训练收敛
📌 工作参考价值(4 点)
- 推荐系统特征通常极度稀疏/长尾,例如 user/item id、类目等输入方向的频率高度不均。DeltaMomentum 可直接替换现有优化器的动量 buffer,对高频方向自动降低遗忘、低频方向加快遗忘,特别适合 embedding 层和 wide 部分;无额外持久内存,可在排序/召回模型训练中验证步数节省。
- 若业务在用 LLM 做生成式推荐或领域模型预训练,可在 AdamW 上启用 DeltaMomentum,论文在 67M/370M 语言模型上分别减少 46%/22% 达到同验证损失的步数,对固定算力预算下扩大模型或数据量有直接价值。
- 工程实现注意:额外计算约为 gated-MLP 线性层成本的 22-25%,无 persistent memory;但方法基于线性层 key-value 结构,对 attention 等非简单线性部分需适配,建议先在 MLP/embedding 层替换。
- 调参经验:系数在 μP 下可跨宽度迁移,若业务使用 μP 训练超大规模模型,能用小模型调的系数直接放大,降低调参成本。
Dion3:Muon 优化器全栈正交更新加速
🧭 Muon 优化器全栈加速与分布式优化 ★ 深度解读
New York University · Princeton University · NVIDIA · Yale University · Microsoft Research
提出 Gram Newton-Schulz、对称 GEMM 核、分数行选择更新和 megabatching,将 Muon 优化器步时最多降低 6 倍且保持/提升模型质量
📌 工作参考价值(4 点)
- Gram Newton-Schulz 可作为标准 Newton-Schulz 的 drop-in 替换,在权重矩阵 aspect ratio 高(如 MLP/MoE 权重)时显著减少 FLOP,适合推荐/广告模型中常见的宽矩阵,无需改变训练流程。
- 分数行选择 + error feedback 是实用的加速技巧:每次只正交化动量矩阵中 ℓ1 范数最大的 f 行,学习率按 η/√f 缩放,在资源受限时可将优化器计算量降低约 1/f²,且不损害甚至轻微提升模型质量。
- Megabatching 策略对 FSDP 下的通信优化很有价值:按矩阵形状分组做 all-to-all,减少通信轮次至 O(1),在中小模型通信瓶颈场景(如 1B 模型 8 卡)可降低 35% 优化器步时,值得工程实现借鉴。
- 自定义对称 GEMM 核(CuteDSL)利用对称性只计算下三角并复制到上三角,在 Hopper/Blackwell 上获得近 2× 速度,适合集成到训练框架中以加速大量对称矩阵乘法。
MALT:通过对角线预处理实现轻量级曲率感知的Muon优化器
🧭 Muon优化器的曲率感知对角线预处理 ★ 深度解读
The Pennsylvania State University · City University of Hong Kong
在Muon正交化前加入行/列对角预处理,以极小开销缓解曲率各向异性,配合噪声自适应步长显著提升LLM预训练性能
📌 工作参考价值(4 点)
- 在推荐模型的矩阵参数(如Embedding、MLP权重)上可直接替换Muon为MALT,在不增加显存和时间的前提下获得更优收敛,尤其适合需要降低训练成本的工业场景。
- 噪声自适应步长MALTER可稳定分布式推荐训练中常见的梯度噪声,避免手动调参,提升训练鲁棒性。
- 对角线预处理与正交化结合的思想可迁移到其他矩阵感知优化器(如Shampoo变体),为结构化参数提供轻量曲率适应。
- 对于曲率各向异性明显的Transformer层(如FFN或Attention投影),建议优先使用MALT,在保持Muon原有优势的同时进一步加速收敛。
利用外推梯度增强锐度感知最小化训练
🧭 训练优化 · 锐度感知最小化改进
将外推梯度引入SAM形成EISAM,两步更新提升泛化且降低超参敏感度
📌 工作参考价值(4 点)
- EISAM 可替换现有 SGD/Adam 训练推荐模型,尤其在数据稀疏或长尾场景下,平坦极小值带来的泛化提升可能直接改善冷启动物品的 AUC 和线上指标。
- 对扰动半径鲁棒的特性意味着超参调优成本更低,适合大规模工业模型快速迭代;可在召回、粗排等不同阶段复用同一组超参。
- 两步更新(预测步探索 loss 地形 + 扰动步微调)提供了一种可工程化的模式:在现有框架中可按 batch 交替执行,额外计算开销主要通过梯度累积补偿,易于分布式实现。
- 当业务模型出现明显过拟合(训练、线上指标背离)时,优先尝试 EISAM,因其不依赖特定架构,可直接融入 CTR 预估、多任务学习等现有训练流程。
为什么 Muon 优于 Adam:曲率视角
🧭 优化器比较 · 曲率与 NDS
National University of Singapore · Yale University · University of Minnesota
Muon 因更低的标准化方向锐度 (NDS) 实现更小的二阶曲率惩罚,从而获得更大的一步损失下降。
📌 工作参考价值(4 点)
- 在大规模推荐或 Agent 模型训练中,可尝试用 Muon 替代 Adam,尤其在数据长尾分布下,Muon 可能带来更快的收敛和更好的泛化。
- 关注 NDS 作为优化器选择的度量:优先选用在训练后段 NDS 更低的优化器,这可能意味着更平滑的收敛和更小的曲率惩罚。
- Muon 的频谱归一化思想可迁移到其他自适应优化器中:通过显式约束更新矩阵的奇异值能量分布,避免梯度对齐高曲率方向导致的优化震荡。
- 数据不平衡(如电商用户行为幂率分布)会放大 Muon 的优势,可据此设计自动优化器选取策略:当数据偏度超过阈值时切换至 Muon 类优化器。
DynMuon:动态频谱塑形优化器
🧭 动态频谱塑形优化 · Muon改进 ★ 深度解读
University of Wisconsin–Madison · MIT · Elorian AI
动态调度Muon类矩阵更新的频谱指数p从正到负,早期聚焦高曲率方向,后期转向平坦方向,实现更优收敛
📌 工作参考价值(4 点)
- 动态频谱调度可迁移:在训练推荐模型(如CTR预估、Embedding矩阵)的矩阵参数时,可引入类似DynMuon的阶段式频谱调整,前期用高曲率方向加速信号衰减,后期转向平坦方向捕捉剩余信号,避免固定谱变换的次优性。
- 高效近似实现:利用Newton–Schulz迭代+二阶泰勒展开近似任意p的UΣ^pV^T更新,额外计算成本仅比Muon增加0.3%~2.5%,可即插即用地嵌入现有AdamW/Muon框架,适合大规模电商模型训练。
- 噪声感知的调度设计思路:通过曲率方向上的信号-噪声分解(如本文的mode-wise分析),可诊断训练中各子空间的优化状态,指导自适应调度,例如在Agent多智体协作训练中动态分配参数更新强度。
- 简单的logistic调度策略:p从1平滑降至-0.25,仅需两个超参数(τ, w),鲁棒性强,无需在线估计曲率,可直接作为默认配置,为其他动态优化器设计提供参考。