提出lambda跳接结构,可防止序列模型的秩坍缩问题。
Lambda-Skip Connections: the architectural component that prevents Rank Collapse
- 引入参数化跳接机制,统一对抗秩坍缩
- 理论证明该结构能有效防止秩坍缩
- 适用于Transformer与状态空间模型
秩坍缩指序列模型中嵌入向量快速收敛至统一状态,导致表达能力下降和训练不稳。已有研究发现跳接、LayerNorm和MLP等组件可缓解此问题,但主要聚焦于Transformer。本文首次将秩坍缩理论扩展至近期流行的状態空间模型(SSMs),提出一种参数化跳接结构——lambda跳接,并建立统一框架分析两类架构。通过理论推导,给出防止秩坍缩的充分条件;并通过消融实验与分析案例验证其必要性。结果表明,该机制在Transformer与SSMs中均具有效性。实验进一步证实跳接与门控机制对防止秩坍缩的关键作用。本工作为理论与实践提供了首个通用的秩坍缩防护保障。
原文摘要 · Abstract (English)
Rank collapse, a phenomenon where embedding vectors in sequence models rapidly converge to a uniform token or equilibrium state, has recently gained attention in the deep learning literature. This phenomenon leads to reduced expressivity and potential training instabilities due to vanishing gradients. Empirical evidence suggests that architectural components like skip connections, LayerNorm, and MultiLayer Perceptrons (MLPs) play critical roles in mitigating rank collapse. While this issue is well-documented for transformers, alternative sequence models, such as State Space Models (SSMs), which have recently gained prominence, have not been thoroughly examined for similar vulnerabilities. This paper extends the theory of rank collapse from transformers to SSMs using a unifying framework that captures both architectures. We study how a parametrized version of the classic skip connection component, which we call \emph{lambda-skip connections}, provides guarantees for rank collapse prevention. Through analytical results, we present a sufficient condition to guarantee prevention of rank collapse across all the aforementioned architectures. We also study the necessity of this condition via ablation studies and analytical examples. To our knowledge, this is the first study that provides a general guarantee to prevent rank collapse, and that investigates rank collapse in the context of SSMs, offering valuable understanding for both theoreticians and practitioners. Finally, we validate our findings with experiments demonstrating the crucial role of architectural components such as skip connections and gating mechanisms in preventing rank collapse.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。