The Super Weight in Large Language Models

Mengxia Yu, De Wang, Qi Shan, Colorado J Reed, Alvin Wan
11/11/2024
cs.CLcs.AI

No official code found. Best implementation: mengxiayu/LLMSuperWeight (124 stars, last commit 2024-12-03). 2 more below.

Abstract

Recent works have shown a surprising result: a small fraction of Large Language Model (LLM) parameter outliers are disproportionately important to the quality of the model. LLMs contain billions of parameters, so these small fractions, such as 0.01%, translate to hundreds of thousands of parameters. In this work, we present an even more surprising finding: Pruning as few as a single parameter can destroy an LLM's ability to generate text -- increasing perplexity by 3 orders of magnitude and reducing zero-shot accuracy to guessing. We propose a data-free method for identifying such parameters, termed super weights, using a single forward pass through the model. We additionally find that these super weights induce correspondingly rare and large activation outliers, termed super activations. When preserved with high precision, super activations can improve simple round-to-nearest quantization to become competitive with state-of-the-art methods. For weight quantization, we similarly find that by preserving the super weight and clipping other weight outliers, round-to-nearest quantization can scale to much larger block sizes than previously considered. To facilitate further research into super weights, we provide an index of super weight coordinates for common, openly available LLMs.

View on arXivView PDF

Code Implementations(3)

Code for studying the super weight in LLM

12416Shell, PythonNov 11, 20241 years ago

Implementation of The Super Weight in Large Language Models paper (unofficial)

20Jupyter NotebookMar 5, 20251 years ago

Implementation of super-weight aware quantization from "The Super Weight in Large Language Models" (Yu et al., 2024, Apple).

30PythonNov 15, 20258 months ago

Similar Papers

Cite this paper

@article{yu2024super,
  title  = {The Super Weight in Large Language Models},
  author = {Mengxia Yu and De Wang and Qi Shan and Colorado J Reed and Alvin Wan},
  year   = {2024},
  eprint = {2411.07191},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.07191v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.07191.svg)](https://codeofpaper.com/paper/2411.07191)

Discussion