Tapered Language Models
锥形语言模型:深度感知的 MLP 宽度分配
🧭 语言模型架构优化 · 深度感知容量分配
Mila · Cornell University · Université de Montréal · CIFAR AI Chair
在总参数固定下,让 MLP 宽度随层深递减(余弦锥形)可免费提升 Transformer 等模型的困惑度与下游性能
📌 工作参考价值(4 点)
- 免费性能提升:在训练推荐领域的 Transformer 模型(如 CTR 预估、行为序列建模)时,可将 MLP 中间宽度按余弦锥形从底层向顶层递减,在零额外参数和计算下获得困惑度或下游指标提升。
- 超参搜索技巧:锥形范围(起始宽度倍数->结束宽度倍数)和衰减形式(余弦 vs 线性)可视为新的架构超参,直接用固定总预算的缩放实验快速寻优,无需改动训练流程。
- 与现有架构解耦:方法不依赖特定注意力机制,可直接叠加到 Gated Attention、Hope-attention、Titans 等变体上,方便复用到自研或改进的推荐模型。
- 强化学习 Agent 内部模型:若 Agent 内部使用 Transformer 作为策略或值函数网络,锥形 MLP 同样可作为低成本的结构性改进,在不增加推理成本的前提下提升模型容量利用效率。