A Simple and Effective Pruning Approach for Large Language Models

AI-generated keywords: Large Language Models Pruning Wanda Sparsity Efficient Techniques

AI-generated Key Points

The license of the paper does not allow us to build upon its content and the key points are generated using the paper metadata rather than the full article.

  • Authors: Mingjie Sun, Zhuang Liu, Anna Bair, J. Zico Kolter
  • Introduce novel pruning method called Wanda (Pruning by Weights and activations) for Large Language Models (LLMs)
  • Wanda prunes weights with smallest magnitudes multiplied by corresponding input activations on a per-output basis
  • Does not require retraining or weight updates, making it practical for immediate use on pruned LLMs
  • Motivation behind Wanda is emergent large magnitude features in LLMs
  • Outperforms established baseline methods like magnitude pruning and competes competitively with recent approaches involving intensive weight updates
  • Code available at https://github.com/locuslab/wanda
  • More information on project page at https://eric-mingjie.github.io/wanda/home.html
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Mingjie Sun, Zhuang Liu, Anna Bair, J. Zico Kolter

project page available at https://eric-mingjie.github.io/wanda/home.html

Abstract: As their size increases, Large Languages Models (LLMs) are natural candidates for network pruning methods: approaches that drop a subset of network weights while striving to preserve performance. Existing methods, however, require either retraining, which is rarely affordable for billion-scale LLMs, or solving a weight reconstruction problem reliant on second-order information, which may also be computationally expensive. In this paper, we introduce a novel, straightforward yet effective pruning method, termed Wanda (Pruning by Weights and activations), designed to induce sparsity in pretrained LLMs. Motivated by the recent observation of emergent large magnitude features in LLMs, our approach prunes weights with the smallest magnitudes multiplied by the corresponding input activations, on a per-output basis. Notably, Wanda requires no retraining or weight update, and the pruned LLM can be used as is. We conduct a thorough evaluation of our method Wanda on LLaMA and LLaMA-2 across various language benchmarks. Wanda significantly outperforms the established baseline of magnitude pruning and performs competitively against recent method involving intensive weight update. Code is available at https://github.com/locuslab/wanda.

Submitted to arXiv on 20 Jun. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2306.11695v2

This paper's license doesn't allow us to build upon its content and the summarizing process is here made with the paper's metadata rather than the article.

In their paper titled "A Simple and Effective Pruning Approach for Large Language Models," authors Mingjie Sun, Zhuang Liu, Anna Bair, and J. Zico Kolter introduce a novel pruning method called Wanda (Pruning by Weights and activations) designed to induce sparsity in pretrained Large Language Models (LLMs). As the size of LLMs increases, network pruning methods become essential to drop a subset of network weights while maintaining performance. Existing methods often require retraining or solving weight reconstruction problems, which can be computationally expensive for billion-scale LLMs. Wanda stands out as a straightforward yet effective approach that prunes weights with the smallest magnitudes multiplied by corresponding input activations on a per-output basis. Notably, Wanda does not necessitate retraining or weight updates, making it practical for immediate use on pruned LLMs. The motivation behind Wanda stems from the observation of emergent large magnitude features in LLMs. The authors conducted a comprehensive evaluation of Wanda on LLaMA and LLaMA-2 across various language benchmarks. Their results demonstrate that Wanda significantly outperforms established baseline methods such as magnitude pruning and competes competitively with recent approaches involving intensive weight updates. For those interested in implementing Wanda, the authors have made their code available at https://github.com/locuslab/wanda. Additionally, more information about the project can be found on the project page at https://eric-mingjie.github.io/wanda/home.html. This research provides valuable insights into efficient pruning techniques for optimizing the performance of Large Language Models without the need for extensive retraining or computational overhead.
Created on 28 Feb. 2024

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.