Entropy-Based Block Pruning for Efficient Large Language Models

Abstract

As large language models continue to scale, their growing computational and storage demands pose significant challenges for real-world deployment. In this work, we investigate redundancy within Transformer-based models and propose an entropy-based pruning strategy to enhance efficiency while maintaining performance. Empirical analysis reveals that the entropy of hidden representations decreases in the early blocks but progressively increases across most subsequent blocks, suggesting that entropy serves as an effective measure of information richness within computation blocks. Unlike cosine similarity, which primarily captures geometric relationships, entropy directly quantifies uncertainty and information content, making it a more reliable pruning criterion. Extensive experiments demonstrate that our entropy-based pruning approach surpasses cosine-similarity-based methods in reducing model size while preserving accuracy, offering a promising direction for efficient model deployment.

Type
Publication
The Fourteenth International Conference on Learning Representations (ICLR 2026)
Liangwei Yang
Liangwei Yang
Research Scientist at Salesforce Research

My research interests include Agent, Data Mining and Efficient Modeling.