一句话定位
Noam Shazeer 提出在 Transformer 前馈层用门控线性单元变体(尤其 GEGLU/SwiGLU)替代 ReLU/GELU,带来稳定质量提升,后被 PaLM、LLaMA 等主流大模型广泛采用。
摘要(3-6 句)
论文系统比较了 Transformer FFN 中用各种 GLU(Gated Linear Unit)变体替换标准 ReLU/GELU 激活的效果,包括 GLU、GEGLU、SwiGLU、ReGLU、Bilinear 等。在 T5 的去噪预训练与多项下游迁移任务上,GLU 变体(尤其 GEGLU、SwiGLU)一致地优于基线 FFN,且额外计算开销很小。这一改进成为后续大模型 FFN 的事实标准之一。
关键技术细节
- GLU 形式:FFN(x) = (xW + b) ⊗ σ(xV + c),即一路线性 × 一路门控激活,逐元素相乘。
- 变体:GLU(sigmoid 门)、GEGLU(GELU 门)、SwiGLU(Swish 门)、ReGLU(ReLU 门)、Bilinear(无激活)。
- 为保持参数量一致,GLU FFN 把隐藏维度按比例缩小(因引入第三个权重矩阵 V)。
- 实验:基于 T5 框架,在 C4 去噪预训练困惑度与 GLUE/SuperGLUE/SQuAD 等下游任务上,GEGLU/SwiGLU 一致领先。
- 影响:SwiGLU 后被 PaLM、LLaMA 等采用为默认 FFN。
原始链接
一手源存档(sources/)
- arxiv-2002.05202.pdf (arXiv 原文 PDF,不入 git)