跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

梯度平滑:耦合逐层更新以改进优化

文章摘要

该论文提出“深度梯度增强”范式,并实例化出“梯度平滑”方法,通过对沿深度方向的块更新进行平滑,在不修改模型架构或训练目标的情况下,持续提升语言模型预训练、LLM推理RL后训练、扩散建模和ViT图像分类的优化与泛化性能。

来源arXiv Machine Learning作者: Haoming Meng, Anton Sugolov, Vardan Papyan
梯度平滑:耦合逐层更新以改进优化
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

深度神经网络,尤其是由重复架构模块堆叠而成的 Transformer,在训练过程中往往会自发形成跨层的结构化关系。这种在深度维度上出现的规律性,正是论文《Gradient Smoothing: Coupling Layer-wise Updates for Improved Optimization》的出发点。该论文由 Haoming Meng 等三位作者完成,于 2026 年 6 月 29 日提交至 arXiv(编号 arXiv:2606.30813v1,分类为机器学习 cs.LG,兼属人工智能 cs.AI),并已被第 43 届国际机器学习大会(ICML 2026)收录。

作者提出的核心概念是“深度梯度增强”(Depth-wise Gradient Augmentation)。这是一种通用的优化范式:对每一层施加的更新,不再只由该层自身的优化器更新决定,而是把沿深度方向排列的各模块优化器更新作为一个整体进行变换后再分配回各层。换言之,优化不再逐层独立进行,而是在层的维度上引入耦合。

在这一框架之下,论文重点研究了“梯度平滑”(Gradient Smoothing)这一类深度方向的平滑方法,并用一个简单的局部“窗口平滑”(Window Smoothing)算子加以实例化。该方法直接作用于任意基础优化器(如 SGD、Adam、Muon)所产生的逐模块更新,因此无需改动优化器本身,计算开销极小,也能与现有的优化流水线兼容。

论文在多种架构与训练场景中评估了该方法,包括语言模型预训练、面向推理能力的大语言模型强化学习后训练、扩散模型训练,以及基于 Vision Transformer 的图像分类。在这些设置中,梯度平滑都一致地改善了优化效果与泛化性能,且不需要修改模型架构或训练目标。

作者进一步表明,该方法促进了表示在深度方向上更有结构的演化,这与其作为结构化深度预条件(preconditioning)方法的解释相一致。总体而言,这些结果确立了深度梯度增强作为利用优化中跨深度结构的一个有前景的框架,并说明梯度平滑是该框架下一个简单且适用面广的具体实例。论文全文以 PDF、实验性 HTML 与 TeX 源码形式在 arXiv 上提供,并给出了 DOI 链接(待 DataCite 注册)。

展开要点与分析

文章情报

工程师进阶

要点

  • 提出深度梯度增强范式,将各层更新视为深度维度的集合进行变换。
  • 梯度平滑基于局部窗口平滑算子在任意基础优化器(如SGD、Adam、Muon)之上直接操作块更新,计算开销极小。
  • 在语言模型预训练、LLM推理RL后训练、扩散建模和ViT图像分类等多种设置中一致提升优化与泛化。
  • 该方法可解释为结构化的深度预条件,促进跨深度表示演化更加结构化。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。