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

在大规模门控Delta网络中解锁特征学习

文章摘要

该论文将Maximal Update Parametrization(μP)扩展到具有门控机制和循环状态动态的Gated Delta Network,推导了缩放规则,并在语言模型预训练中验证了学习率跨宽度零样本迁移的有效性。

来源arXiv Machine Learning作者: Yifeng Liu, Quanquan Gu
在大规模门控Delta网络中解锁特征学习
报告错误

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

查看更正说明
直接读正文

由Yifeng Liu和Quanquan Gu提交的论文《Unlocking Feature Learning in Gated Delta Networks at Scale》于2026年6月2日发布在arXiv上(编号arXiv:2606.04048v1),属于机器学习(cs.LG)与人工智能(cs.AI)交叉领域。该研究聚焦于大语言模型训练与扩展中面临的计算资源瓶颈,旨在通过高效架构与有原则的超参数调优方法降低训练成本。

近年来,Maximal Update Parametrization(μP)为标准Transformer提供了零样本超参数迁移能力,使得在小规模模型上调优的超参数可直接应用于大规模模型。然而,对于线性模型——尤其是那些具有结构化状态转移和复杂架构的模型——μP的扩展仍鲜有探索。Gated Delta Network作为一类新兴的线性循环架构,其门控机制与循环状态动态使得坐标尺度的传播变得复杂,缺乏相应的缩放理论。

针对这一空白,作者通过严格地在前向传播、门控机制以及循环状态动态中传播坐标大小估计,首次推导出了Gated Delta Network的缩放规则。这些规则明确了参数初始化、学习率等超参数应如何随模型宽度变化而调整,从而保持特征学习的稳定性。

为了验证理论分析的正确性与实用性,研究团队在语言模型预训练任务上进行了实验。结果表明,遵循所推导配置的模型在AdamW和SGD优化器下均能实现学习率跨模型宽度的稳定迁移,即在小宽度上调优的学习率可直接用于大宽度模型而不损失性能。相反,采用标准参数化(如标准初始化)的模型则无法实现这种迁移,学习率需要针对每个宽度单独调优。这一对比有力地支持了作者分析的正确性。

该工作的意义在于,它将μP的理论框架成功扩展至具有门控和循环状态动态的线性架构,为Gated Delta Network及其他类似结构的大规模训练提供了实用的超参数迁移方案。这不仅减少了大规模调参的计算开销,也为未来高效架构的规模化部署奠定了理论基础。论文全文可通过arXiv获取,包含详细的推导与实验设置。

展开要点与分析

文章情报

工程师进阶

要点

  • 首次为Gated Delta Network推导了基于坐标大小传播的缩放规则,支持μP风格的超参数迁移。
  • 实验表明,在AdamW和SGD下,所提配置可实现学习率跨模型宽度的稳定迁移,而标准参数化无法迁移。
  • 该工作填补了μP在具有结构化状态转移的线性模型和复杂架构上的空白。

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