Skip to content
Preprint

Weight-Adjusted Gradients Reveal Parameter Importance and Failure Modes in LLMs

Jul 2026 · 2 citations · ⚡ 1 influential · 65 references
Computer Science

Abstract

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.

View source

Similar papers

Preprint Jul 2026

Complexity-Budgeted, Interaction-Aware Interpretable Model for Tabular Data

Results support IAIML as a compact, interaction-aware framework appropriate for settings where bounded explanation size and controlled treatment of feature interactions are design requirements.

S. Krishnamoorthy · 0 citations
Book Open access Jul 2026

Revisiting the Role of Learned Attention Weighting in SASRec

A controlled SASRec variant is studied that replaces learned attention weights with uniform aggregation and is trained under an otherwise identical block structure and training recipe, and this modification often yields performance comparable to the original model, with clear dataset-dependent exceptions.

K.Ozaki K.Ozaki, Keigo Sakurai, Ren Togo et al. · 0 citations
Preprint Jul 2026

Bridging extrinsic and intrinsic variable importance

Simulations show that VIMP and MPLOCO agree most closely when the fitted learner is well aligned with the data-generating mechanism, and clarify when intrinsic and extrinsic importance can be interpreted similarly and when they provide complementary information.

Yucheng Zhao, Brian D. Williamson · 0 citations
Preprint Jul 2026

Individual Parameters in Weight-Sparse Transformers Appear Interpretable

This work introduces an automated LLM pipeline that writes a short, human-readable description of when a weight matters and verifies it on held-out text, crediting a weight only if its description generalizes.

Arnau Marin-Llobet, S. Heimersheim · 0 citations
Preprint Jul 2026

Super Weights in LLMs and the Failure of Selective Training

It is established that parameter importance does not imply parameter trainability in isolation, and that effective fine-tuning relies on structured decompositions over entire layers rather than targeting individually important weights.

S. Subramanian, Adewale Akinfaderin, Akarsha Sehwag · 0 citations
Preprint Aug 2026

What We Observe as LLM Behavior Can Be a Side-effect of Inference Backend

Benchmark scores are reported as properties of a model, yet the inference framework used to produce them, such as HuggingFace, vLLM, or Ollama, are considered non-influential and their names and versions are almost never disclosed. In this work we investigate how much this choice can influence the model output. In a fully-crossed study (three instruction-tuned models x five inference frameworks x six benchmarks x four generation modes) we investigate how different tools (wrappers/backend) influence benchmark scores and how their score changes is influenced by generation hyper-parameters. We find backend to be a non-negligible factor where even under greedy, sampling-noise-free decoding, changing the backend can significantly alter models performance and this effect is structural and strongly model-dependent. Decomposing the variance according to generation mode reveal that considerable portion of the variability (roughly 39\%) a practitioner sees out-of-the-box can stem from the backend, while the remaining stems from sampling noise and each framework's default generation parameters, both of which are avoidable by disclosing and matching the generation configuration. These divergences are more pronounced on factual than on social-bias benchmarks. Overall, benchmark numbers are not backend-agnostic therefore, we recommend disclosing the backend, its version, and the full generation configuration, also using deterministic decoding for cross-backend comparison.

Shahed Masoudian, Passant Shafaei, Monorama Swain et al. · 0 citations