Abstract:Large language models (LLMs) excel in language tasks, especially with supervised fine-tuning after pre-training. However, their substantial memory and computational requirements hinder practical applications. Structural pruning, which reduces less significant weight dimensions, is one solution. Yet, traditional post-hoc pruning often leads to significant performance loss, with limited recovery from further fine-tuning due to reduced capacity. Since the model fine-tuning refines the general and chaotic knowledge in pre-trained models, we aim to incorporate structural pruning with the fine-tuning, and propose the Pruning-Aware Tuning (PAT) paradigm to eliminate model redundancy while preserving the model performance to the maximum extend. Specifically, we insert the innovative Hybrid Sparsification Modules (HSMs) between the Attention and FFN components to accordingly sparsify the upstream and downstream linear modules. The HSM comprises a lightweight operator and a globally shared trainable mask. The lightweight operator maintains a training overhead comparable to that of LoRA, while the trainable mask unifies the channels to be sparsified, ensuring structural pruning. Additionally, we propose the Identity Loss which decouples the transformation and scaling properties of the HSMs to enhance training robustness. Extensive experiments demonstrate that PAT excels in both performance and efficiency. For example, our Llama2-7b model with a 25\% pruning ratio achieves 1.33$\times$ speedup while outperforming the LoRA-finetuned model by up to 1.26\% in accuracy with a similar training cost. Code: <a class="link-external link-https" href="https://github.com/kriskrisliu/PAT_Pruning-Aware-Tuning" rel="external noopener nofollow">this https URL</a>

How to Prune Your Language Model: Recovering Accuracy on the "Sparsity May Cry'' Benchmark

Towards Robust Pruning: An Adaptive Knowledge-Retention Pruning Strategy for Language Models

On Importance of Layer Pruning for Smaller BERT Models and Low Resource Languages

Pruning Foundation Models for High Accuracy without Retraining

From Dense to Sparse: Contrastive Pruning for Better Pre-trained Language Model Compression

Investigating Language-Specific Calibration For Pruning Multilingual Large Language Models

Pruning before Fine-tuning: A Retraining-free Compression Framework for Pre-trained Language Models

DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimization

An Automatic and Efficient BERT Pruning for Edge AI Systems

Self-Data Distillation for Recovering Quality in Pruned Large Language Models

Reconstruct the Pruned Model without Any Retraining

Towards Building Efficient Sentence BERT Models using Layer Pruning

Pruning Large Language Models via Accuracy Predictor

Structured Pruning of Large Language Models

Accurate Neural Network Pruning Requires Rethinking Sparse Optimization

LEAP: Learnable Pruning for Transformer-based Models

Reassessing Layer Pruning in LLMs: New Insights and Methods

Beyond Size: How Gradients Shape Pruning Decisions in Large Language Models

Pruning as a Domain-specific LLM Extractor

Prune Once for All: Sparse Pre-Trained Language Models

PAT: Pruning-Aware Tuning for Large Language Models