DAGFormer:用拓扑预测器全局规划 Transformer 跨层连接,降低困惑度

AI-Interpretability-illinois · hf · 2026-10-19

伊利诺伊团队的论文提出 DAGFormer,改进 Transformer 效率:不再让每层固定读取上一层,而是用一个联合训练的 2 层因果 Transformer「拓扑预测器」,根据输入逐 token 规划 Q/K/V/residual 的跨层连接(DAG 结构),全局规划读取来源。在 75M、300M、1B 三个规模上,规划拓扑加 refinement 的变体均降低语言建模困惑度并提升下游任务表现,还提供交互式可视化展示训练后的连接图。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →