跳到正文
/阅读与研究

外观

快速抵达

↑ ↓ 选择 · Enter 打开 · Esc 关闭MyHOT
原文
Hugging Face · 每日论文· 媒体 / 作者来源·精选

DAGFormer:用全局拓扑预测器规划Transformer跨层连接

原文标题

DAGFormer: Global Topology Prediction for Transformer Cross-Layer Connections

AI 导读

研究者提出 DAGFormer,在标准 Transformer 残差连接之外增加从浅层到深层的直接连接,并由与模型联合训练的拓扑预测器根据输入 token 控制每条连接的强度;与现有逐层独立设置连接的跨层路由架构不同,它一次性规划整个模型的路由。在 75M 到 1B 参数、与 OLMo-2 匹配的对比中,WikiText-2 困惑度比原版 Transformer 低 17%–25%,1B 版本用 5B token 训练即优于用 10B token 训练的原版模型;在 13 项常识、科学、阅读和数学评测任务上平均准确率高 2.0–2.5 分,指令微调后 AlpacaEval 胜率 63%–70%,结构化剪枝下六种设置中有五种退化最小。

来源:Hugging Face · 每日论文 · arxiv.org

导读与选稿说明

把跨层连接由逐层独立决策改为全局联合规划,在同等参数与算力下显著降低困惑度并提升下游任务表现,也为剪枝鲁棒性提供了一种可复用的架构设计思路。

导读由模型辅助整理,非独立核验。完整信息见原文。