Abstract:Large Language Models (LLMs) such as ChatGPT and LlaMA are advancing rapidly in generative Artificial Intelligence (AI), but their immense size poses significant challenges, such as huge training and inference costs, substantial energy demands, and limitations for on-site deployment. Traditional compression methods such as pruning, distillation, and low-rank approximation focus on reducing the effective number of neurons in the network, while quantization focuses on reducing the numerical precision of individual weights to reduce the model size while keeping the number of neurons fixed. While these compression methods have been relatively successful in practice, there is no compelling reason to believe that truncating the number of neurons is an optimal strategy. In this context, this paper introduces CompactifAI, an innovative LLM compression approach using quantum-inspired Tensor Networks that focuses on the model's correlation space instead, allowing for a more controlled, refined and interpretable model compression. Our method is versatile and can be implemented with - or on top of - other compression techniques. As a benchmark, we demonstrate that a combination of CompactifAI with quantization allows to reduce a 93% the memory size of LlaMA 7B, reducing also 70% the number of parameters, accelerating 50% the training and 25% the inference times of the model, and just with a small accuracy drop of 2% - 3%, going much beyond of what is achievable today by other compression techniques. Our methods also allow to perform a refined layer sensitivity profiling, showing that deeper layers tend to be more suitable for tensor network compression, which is compatible with recent observations on the ineffectiveness of those layers for LLM performance. Our results imply that standard LLMs are, in fact, heavily overparametrized, and do not need to be large at all.

TQCompressor: improving tensor decomposition methods in neural networks via permutations

Joint Structured Pruning and Dense Knowledge Distillation for Efficient Transformer Model Compression

CompactifAI: Extreme Compression of Large Language Models using Quantum-Inspired Tensor Networks

Blockwise Compression of Transformer-based Models without Retraining

Quantization-Aware and Tensor-Compressed Training of Transformers for Natural Language Understanding

Kronecker Decomposition for GPT Compression

Hybrid Tensor Decomposition in Neural Network Compression

Exploring Extreme Parameter Compression for Pre-trained Language Models

TensorGPT: Efficient Compression of Large Language Models based on Tensor-Train Decomposition

Modular Transformers: Compressing Transformers into Modularized Layers for Flexible Efficient Inference

MoDeGPT: Modular Decomposition for Large Language Model Compression

Deep neural network compression by Tucker decomposition with nonlinear response

Combining Compressions for Multiplicative Size Scaling on Natural Language Tasks

A Survey on Transformer Compression

Semi-tensor Product-based TensorDecomposition for Neural Network Compression

Towards Optimal Compression: Joint Pruning and Quantization

Extreme Compression for Pre-trained Transformers Made Simple and Efficient

PET: Parameter-efficient Knowledge Distillation on Transformer

Kronecker CP Decomposition with Fast Multiplication for Compressing RNNs

Compression of Deep Learning Models for Text: A Survey

Quantization Aware Factorization for Deep Neural Network Compression