Skip to content
简体中文

切换主题

// ▲ 预训练、缩放与分布式训练

μP 超参迁移

调好小模型的超参数,直接迁移到大模型。

μP (Maximal Update Parametrization) 2022 前沿研究 扩展

通俗解释

按特定规则随宽度缩放初始化和学习率,使不同宽度的模型最优超参一致,于是只需在小模型上搜参,省下大量算力。

用在哪里:大模型超参搜索(如 Cerebras-GPT)

时间轴 · 前置 → 当前 → 后继

2017 AdamW2022 μP 超参迁移

01原始论文

Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfer

Greg Yang, Edward J. Hu, Igor Babuschkin, et al. · 2022

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

μP 超参迁移

学完可以学

  • 图谱里暂无后继

相关:Chinchilla 计算最优

AdamWChinchilla 计算最优 μP 超参迁移

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)