arXiv:2502.17405stat.MLcs.LG2025-02ICML被引 3

提出函数空间学习率,实现跨模型规模的高效超参迁移。

Function-Space Learning Rates

  • 用输出函数变化衡量学习率,替代传统参数空间更新。
  • 仅需少量反向传播即可计算,训练时可周期性执行。
  • 适用于不同结构和规模模型,支持超参自动适配。

我们研究层间函数空间学习率,衡量神经网络输出函数对参数张量更新的响应幅度,与传统参数空间学习率形成对比。本文提出高效方法,在任意神经网络中测量并设置函数空间学习率,仅需在训练开始或周期性执行少量额外反向传播,计算开销极小。实验展示两大应用:(1) 在函数空间而非参数空间分析标准优化器动态;(2) 提出FLeRM(函数空间学习率匹配)方法,实现模型规模间的超参数迁移。FLeRM在小型廉价基模型上记录函数空间学习率,再在训练大模型时自动调整参数空间各层学习率,以保持一致的函数空间更新。该方法在多种架构(含带残差连接的MLP、不同层归一化方案的Transformer)中成功实现宽度、深度、初始化尺度及LoRA秩的超参数转移。

原文摘要 · Abstract (English)

We consider layerwise function-space learning rates, which measure the magnitude of the change in a neural network's output function in response to an update to a parameter tensor. This contrasts with traditional learning rates, which describe the magnitude of changes in parameter space. We develop efficient methods to measure and set function-space learning rates in arbitrary neural networks, requiring only minimal computational overhead through a few additional backward passes that can be performed at the start of, or periodically during, training. We demonstrate two key applications: (1) analysing the dynamics of standard neural network optimisers in function space, rather than parameter space, and (2) introducing FLeRM (Function-space Learning Rate Matching), a novel approach to hyperparameter transfer across model scales. FLeRM records function-space learning rates while training a small, cheap base model, then automatically adjusts parameter-space layerwise learning rates when training larger models to maintain consistent function-space updates. FLeRM gives hyperparameter transfer across model width, depth, initialisation scale, and LoRA rank in various architectures including MLPs with residual connections and transformers with different layer normalisation schemes.

学习率超参迁移函数空间优化器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。