Thanos: A Block-wise Pruning Algorithm for Efficient Large Language Model Compression
Preprint
Abstract
This paper presents Thanos, a novel weight-pruning algorithm designed to reduce the memory footprint and enhance the computational efficiency of large language models (LLMs) by removing redundant weights while maintaining accuracy. Thanos introduces a block-wise pruning strategy with adaptive masks that dynamically adjust to weight importance, enabling flexible sparsity patterns and structured formats, such as $n:m$ sparsity, optimized for hardware acceleration. Experimental evaluations demonstrate that Thanos achieves state-of-the-art performance in structured pruning and outperforms existing methods in unstructured pruning. By providing an efficient and adaptable approach to model compression, Thanos offers a practical solution for deploying large models in resource-constrained environments.
Keywords
All Related Versions
This publication has 9 references indexed in Scilit:
- Warming and parasitism impair the performance of Baltic native and invasive macroalgae and their associated faunaLimnology and Oceanography, 2023
- Intriguing Properties of Quantization at Scale2023
- Quantizable Transformers: Removing Outliers by Helping Attention Heads Do Nothing2023
- LLM.int8(): 8-Bit Matrix Multiplication for Transformers at Scale2022
- A framework for few-shot language model evaluation2021
- Array programming with NumPyNature, 2020
- BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questions2019
- HellaSwag: Can a Machine Really Finish Your Sentence?2019
- Iterative Thresholding for Sparse ApproximationsJournal of Fourier Analysis and Applications, 2008