Super Weights in LLMs and the Failure of Selective Training

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

Recent research challenges the universal importance and trainability of "Super Weights" in Large Language Models (LLMs), parameters previously identified as critical for model performance. A study using OLMo-1B and OLMo-7B models demonstrates that pruning Super Weights does not consistently degrade all LLMs. Crucially, attempts to train Super Weights in isolation, ranging from 100 to 8,192 parameters, or even within local neighborhoods up to 36K parameters, resulted in accuracy levels equivalent to random guessing. This performance collapse is specific to Super Weight coordinates, as training an equal number of randomly chosen parameters in the same down_proj layers actually improved over the baseline. In contrast, vanilla LoRA, which updates 0.16% of parameters through low-rank structured decompositions across entire layers, proved effective, even when applied to down_proj layers. These findings indicate that while certain parameters may be important, they are not necessarily trainable in isolation, suggesting that effective fine-tuning strategies should focus on structured, layer-wide updates rather than targeting individual "Super Weights."

Key takeaway

For Machine Learning Engineers optimizing LLM fine-tuning, reconsider strategies focused on individually "important" parameters. Your efforts to selectively train or prune specific "Super Weights" are unlikely to yield performance gains and may degrade models like OLMo-1B and OLMo-7B. Instead, prioritize structured, low-rank adaptation methods such as LoRA, which effectively update entire layers with minimal parameters (e.g., 0.16%). This approach offers superior results compared to targeting isolated weights, even those deemed critical.

Key insights

Parameter importance does not imply trainability; effective LLM fine-tuning requires structured, layer-wide updates.

Principles

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.