arXiv:2607.20484cs.AI2026-07

通过拓扑正则化路径缓解大模型表示崩溃问题

The Devil is in the Spectrum: Mitigating Representation Collapse in LLMs via Topologically Regularized Side-Path

论文配图:The Devil is in the Spectrum: Mitigating Representation Collapse in LLMs via Topologically Regularized Side-Path
图 1 · 摘自论文原文
  • 提出非侵入式TRSP架构,用三角盒机制调节令牌交互拓扑
  • 在8倍长上下文下保持83%准确率,优于现有方法30~50个百分点
  • 适合关注长文本建模与注意力机制优化的研究者

大语言模型受限于表示崩溃,严重损害长上下文表现。我们发现现有方法易陷入两种病理极端:同质化崩溃(如注意力下沉导致秩不足)和孤立化崩溃(如局部注意力造成上下文断连)。通过分析注意力动态的谱特性,我们揭示了混合效率(谱间隙)与信息容量(有效秩)之间的内在权衡,标准机制难以平衡。为此,提出拓扑正则化侧路径(TRSP),一种无需参数的非侵入式架构改进。TRSP采用无参三角盒机制,由轻量级长度感知门控缩放,通过近邻耦合保留有效秩、远距离传播支持非退化混合,实现几何更健康的转移算子,不改变核心注意力结构。实验表明,在通用能力与长上下文基准上均有显著提升。特别地,在训练长度扩大8倍的NoLiMa任务中,TRSP保持83%准确率,分别超越差分变换器与门控注意力约30和50个百分点。代码已开源。

原文摘要 · Abstract (English)

Large Language Models (LLMs) are fundamentally limited by representation collapse, a bottleneck that severely degrades long-context performance. We identify that existing approaches risk drifting into one of two pathological extremes: homogenization collapse (e.g., attention sinks causing rank deficiency) and isolation collapse (e.g., local attention causing context disconnection). Through spectral analysis of attention dynamics, we derive an intrinsic trade-off between mixing efficiency (spectral gap) and information capacity (effective rank) that standard mechanisms struggle to balance. To resolve this dilemma, we propose the Topologically Regularized Side-Path (TRSP), a non-invasive architectural intervention that achieves spectral balance. TRSP employs a parameter-free Triangular Box mechanism, scaled by a lightweight, length-aware gate, to regularize the token interaction topology. By integrating proximal coupling to preserve effective rank and distal propagation to support non-degenerate mixing, TRSP promotes a geometrically healthier transition operator without altering core attention. Experiments show significant improvements across general capabilities and long-context benchmarks. Notably, on NoLiMa at $8\times$ the training length, TRSP retains $83\%$ accuracy and surpasses the Differential Transformer and Gated Attention by approximately 30 and 50 percentage points, respectively. Code available at: https://github.com/Eziotao-tyd/TRSP.

大模型注意力机制长文本拓扑正则

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。