提出新方法揭示大模型中关键参数与失效模式。
Weight-Adjusted Gradients Reveal Parameter Importance and Failure Modes in LLMs

- 通过权重与梯度交互计算参数重要性
- 发现极少数参数修改即导致性能崩溃
- 适用于模型调试、量化与知识编辑
理解大语言模型中哪些参数具有影响力,是提升其效率、可靠性与可解释性的核心。我们提出权重调整梯度(WAG),一种简单而有效的方法,能显式捕捉模型权重与一阶梯度信息之间的相互作用,识别出对模型行为有不成比例影响的参数,如导致模型坍塌的参数。在多种模型与设置下,我们发现WAG能揭示一个微小但至关重要的参数子集,其修改会导致性能剧烈下降,而现有重要性度量方法对此类失效模式视而不见。这些发现揭示了权重与梯度之间此前未被充分探索的相互作用,表明仅靠单一信号无法全面理解参数重要性。WAG的惊人有效性暗示了训练网络的基本结构特性,并引发关于零阶与一阶信息在深度学习中作用的新问题。我们在混合专家架构中的专家分配、参数级遗忘、混合精度量化及知识编辑的层选择等多个应用中展示了WAG的实际效用。结果表明,WAG可作为分析、调试与控制大语言模型的统一框架,并为原理性模型解读开辟新方向。
原文摘要 · Abstract (English)
Understanding which parameters are influential in Large Language Models (LLMs) is central to improving their efficiency, reliability, and interpretability. We introduce Weight-Adjusted Gradients (WAG), a simple yet effective approach for estimating parameter importance that explicitly captures the interaction between model weights and first-order gradient information and identifies parameters that disproportionately influence model behavior, such as those responsible for collapse phenomena in LLMs. Across a range of models and settings, we show that WAG surfaces a tiny but critical subset of parameters whose modification leads to dramatic degradation in performance, a failure mode that existing importance metrics overlook. These findings reveal a previously underexplored interplay between weights and gradients, suggesting that parameter importance cannot be fully understood through either signal alone. The surprising effectiveness of WAG points to fundamental structural properties of trained networks and motivates new open questions about the role of zeroth-order and first-order information in deep learning. We demonstrate the practical utility of WAG across multiple applications, including expert allocation in mixture-of-expert architectures, parameter-specific unlearning, mixed-precision quantization, and layer selection for knowledge editing. Our results position WAG as a unified approach for analyzing, debugging, and controlling LLMs, and opens new directions for principled model-level interpretation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。