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>

$\rm SP^3$: Enhancing Structured Pruning via PCA Projection

Structured Pruning for Efficient Convolutional Neural Networks Via Incremental Regularization

Structured Probabilistic Pruning for Convolutional Neural Network Acceleration.

Enhancing One-shot Pruned Pre-trained Language Models through Sparse-Dense-Sparse Mechanism

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

Structured Pruning of Large Language Models

Structured Pattern Pruning Using Regularization

Pruning Foundation Models for High Accuracy without Retraining

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

Probing Structured Pruning on Multilingual Pre-trained Models: Settings, Algorithms, and Efficiency

CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Information

Pruning as a Domain-specific LLM Extractor

PAT: Pruning-Aware Tuning for Large Language Models

Structured Pruning Learns Compact and Accurate Models

DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimization

CRISP: Hybrid Structured Sparsity for Class-aware Model Pruning

Gradient-Free Structured Pruning with Unlabeled Data

DISP-LLM: Dimension-Independent Structural Pruning for Large Language Models

SPDF: Sparse Pre-training and Dense Fine-tuning for Large Language Models

Enhanced Sparsification via Stimulative Training

Structured Pruning of a BERT-based Question Answering Model