用图神经网络理论解析大模型失效原因,揭示信息传播瓶颈。
Understanding the Failure Modes of Transformers through the Lens of Graph Neural Networks
- 将变压器视为信息混合传播系统,从图神经网络视角分析其失效机制。
- 发现解码器仅依赖的因果结构导致信息传播存在可预测的致命缺陷。
- 为现有改进方法提供理论依据,指导针对性优化。
Transformer,特别是解码器仅有的架构,在现代大语言模型中占据主导地位。尽管表现优异,但其仍存在令人意外的失效模式和不对称的性能退化问题。本文通过图神经网络(GNN)理论,研究了这些失效现象。我们提出,深度学习的核心在于可学习的信息混合与传播,因此模型失效本质上是信息传播中的瓶颈。这自然引向已有丰富理论的GNN领域,其中已有关于信息传播瓶颈和模型失效模式的研究。我们进一步指出,许多GNN面临的挑战同样存在于Transformer中。此外,我们分析了仅解码器架构的因果性带来的独特几何特性,导致可预测且可能灾难性的失效模式。最后,我们观察到当前的Transformer解决方案多为直觉驱动、缺乏理论基础。为此,我们统一这些方法,从更理论的角度解释其有效性,阐明所解决的问题,并提出改进方向。本文旨在弥合大模型失效现象与理论理解之间的鸿沟。
原文摘要 · Abstract (English)
Transformers and more specifically decoder-only transformers dominate modern LLM architectures. While they have shown to work exceptionally well, they are not without issues, resulting in surprising failure modes and predictably asymmetric performance degradation. This article is a study of many of these observed failure modes of transformers through the lens of graph neural network (GNN) theory. We first make the case that much of deep learning, including transformers, is about learnable information mixing and propagation. This makes the study of model failure modes a study of bottlenecks in information propagation. This naturally leads to GNN theory, where there is already a rich literature on information propagation bottlenecks and theoretical failure modes of models. We then make the case that many issues faced by GNNs are also experienced by transformers. In addition, we analyze how the causal nature of decoder-only transformers create interesting geometric properties in information propagation, resulting in predictable and potentially devastating failure modes. Finally, we observe that existing solutions in transformer research tend to be ad-hoc and driven by intuition rather than grounded theoretical motivation. As such, we unify many such solutions under a more theoretical perspective, providing insight into why they work, what problem they are actually solving, and how they can be further improved to target specific failure modes of transformers. Overall, this article is an attempt to bridge the gap between observed failure modes in transformers and a general lack of theoretical understanding of them in this space.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。