arXiv:2603.06922cs.LG2026-03中稿 · ICLR被引 4

揭示大模型前馈层中隐空间信息流动的谱动态规律

NerVE: Nonlinear Eigenspectrum Dynamics in LLM Feed-Forward Networks

  • 通过四个谱指标追踪前馈网络的动态变化
  • 发现非线性函数会重新注入方差,影响隐空间利用率
  • 适用于不同架构与优化器,可指导模型设计

我们提出NerVE,一个统一的特征谱框架,用于理解大语言模型前馈网络(FFNs)在高维隐空间中的信息流动机制。尽管FFNs占据主要参数量,其高维动态仍不清晰。NerVE通过轻量、低内存的四种互补指标——谱熵(离散度)、参与比(有效维度)、特征值早期富集(顶部集中度)和Jensen-Shannon散度(分布偏移)——实现对特征谱动态的跟踪。核心发现是:FFN的非线性函数会将方差重新注入各特征模式,从根本上控制隐空间利用效率;且优化器几何结构显著调节这一方差再注入程度。我们在多种模型规模及架构配置下验证了NerVE的有效性,结果表明:归一化方式调控方差流动,权重几何约束隐空间,位置编码与激活函数调节信息流,优化器选择重新分配有效容量。所有设置下,NerVE均恢复出稳定的谱特征,且与模型泛化能力高度相关,响应设计变化具有可预测性。该框架不仅适用于Transformer,也扩展至MLP-Mixer,为架构与优化器选择提供超越试错的实用洞见。

原文摘要 · Abstract (English)

We introduce NerVE, a unified eigenspectral framework for understanding how feed-forward networks (FFNs) in large language models (LLMs) organize and regulate information flow in high-dimensional latent space. Despite FFNs dominating the parameter budget, their high-dimensional dynamics remain poorly understood. NerVE addresses this gap through lightweight, memory-efficient tracking of eigenspectrum dynamics via four complementary metrics: Spectral Entropy (dispersion), Participation Ratio (effective dimensionality), Eigenvalue Early Enrichment (top-heaviness), and Jensen-Shannon divergence (distributional shifts). Our key insight is that FFN nonlinearities reinject variance across eigenmodes, fundamentally governing latent dimension utilization, and that optimizer geometry strongly modulates the extent of this variance reinjection. We validate NerVE across model scales, and diverse architectural and optimizer configurations, each uniquely shaping FFN dynamics: normalization schemes controlling variance flow; FFN weight geometries constraining latent space; positional encoding and activation functions regulating information flow; and optimizer choices redistributing effective capacity across depth. Across these settings, NerVE consistently recovers stable spectral signatures that correlate with model's generalization ability and respond predictably to design choices, generalizing beyond transformer to MLP-Mixer architectures, providing actionable insights for architectural and optimizer choices beyond trial-and-error.

大模型分析特征谱前馈网络隐空间

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