Transformer中的定向路由机制 / Directional Routing in Transformers
1️⃣ 一句话总结
这篇论文提出了一种名为‘定向路由’的轻量级机制,它能让Transformer模型中的注意力头学会协作,极大地提升了模型理解和记忆事实的能力,但移除这个协作机制会使模型性能崩溃,而单独移除某个注意力头影响却很小。
We introduce directional routing, a lightweight mechanism that gives each transformer attention head learned suppression directions controlled by a shared router, at 3.9% parameter cost. We train a 433M-parameter model alongside an identical baseline in a single run, then trace the resulting circuits through mechanistic interpretability. Routing becomes the model's dominant computational pathway. Disabling it collapses factual recall to near-zero probability across all 8 test prompts and drops induction accuracy from 93.4% to 0.0%. Knocking out individual attention heads has negligible effect: the primary mover head's removal actually increases target probability, and induction heads retain 98.6% accuracy without their strongest member. The coordination mechanism is irreplaceable; the components it coordinates are not. The model also self-organizes, without explicit pressure, into two regimes: domain-adaptive routing in early layers and fixed syntactic pruning in late layers, where the least-varying layer is the most critical (+42.6 PPL when disabled). Routing reduces perplexity 31-56% relative to the baseline, though downstream multiple-choice benchmarks do not yet reflect these gains.
Transformer中的定向路由机制 / Directional Routing in Transformers
这篇论文提出了一种名为‘定向路由’的轻量级机制,它能让Transformer模型中的注意力头学会协作,极大地提升了模型理解和记忆事实的能力,但移除这个协作机制会使模型性能崩溃,而单独移除某个注意力头影响却很小。
源自 arXiv: 2603.14923