Super Weights in LLMs and the Failure of Selective Training
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
- Pruning Super Weights does not universally degrade all LLMs.
- Training Super Weights in isolation fails, even locally.
- Effective fine-tuning uses structured decompositions over layers.
In practice
- Avoid isolated training of "important" parameters.
- Prioritize low-rank structured updates like LoRA.
- Focus fine-tuning on entire layers, not specific weights.
Topics
- Super Weights
- LLM Fine-tuning
- Parameter Pruning
- LoRA
- OLMo Models
- Low-Rank Adaptation
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.