Files
myWiki/concepts/mlp-width-tapering.md

1.3 KiB
Raw History

title, created, updated, type, tags, sources, confidence
title created updated type tags sources confidence
MLP Width TaperingMLP 宽度渐缩) 2026-06-29 2026-06-29 concept
language-model
mlp
architecture
efficiency
transformer
tapered-language-models
high

MLP Width Tapering

depth-aware-capacity-allocation 的具体实例化:在固定总参数量下,将 Transformer/LLM 中 MLP 的中间层宽度d_ff沿深度单调递减。

为什么选 MLP

组件 参数占比 跨架构通用性 调节轴
MLP (FFN) 主导(~2/3 所有架构共享 单一标量 d_ff
Attention 次之 架构差异大 多头/维度耦合
LayerNorm 极小 通用 无意义

MLP 是自然的 taper 目标:参数量最大、架构通用、调节维度单一干净。

Cosine Schedule

推荐的衰减方式见 cosine-taper-schedule

w_ = w_max · (cos(πℓ/(2L)))^p

  • ∈ [0, L-1]:层索引
  • p控制陡峭度
  • 最优范围1.50× → 0.50× baseline d_ff

效果

440M Transformeruniform 16.28 → cosine taper 14.44 perplexity零额外参数/计算)。

相关概念