3BASiL: An Algorithmic Framework for Sparse plus Low-Rank Compression of LLMs
Organizations: Operations Research Center Massachusetts Institute of Technology
Abstract
Sparse plus Low-Rank decomposition of Large Language Models (LLMs) has emerged as a promising direction in model compression, aiming to decompose pre-trained model weights into a sum of sparse and low-rank matrices . Despite recent progress, existing methods often suffer from substantial performance degradation compared to dense models. In this work, we introduce 3BASiL-TM, an efficient one-shot post-training method for decomposition of LLMs that addresses this gap. Our approach first introduces a novel 3-Block Alternating Direction Method of Multipliers (ADMM) method, termed 3BASiL, to minimize the layer-wise reconstruction error with convergence guarantees. We then design an efficient transformer-matching (TM) refinement step that jointly optimizes the sparse and low-rank components across transformer layers. This step minimizes a novel memory-efficient loss that aligns outputs at the transformer level. Notably, the TM procedure is universal as it can enhance any decomposition, including pure sparsity. Our numerical experiments show that 3BASiL-TM reduces the WikiText2 perplexity gap relative to dense LLaMA-8B model by over 30% under a (2:4 Sparse + 64 LR) configuration, compared to prior methods. Moreover, our method achieves over 2.5x faster compression runtime on an A100 GPU compared to SOTA method. Our code is available at https://github.com/mazumder-lab/3BASiL.
Figures & tables
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| OATS | 3:8+64LR | 58.88 | 40.76 | 67.35 | 63.71 | 39.48 | 42.68 | 24.32 | 53.91 | 52.71 | 28.40 | 63.98 | 46.15 |
| Hassle-free-SparseGPT | 29.32 | 21.46 | 32.06 | 68.66 | 51.99 | 50.97 | 30.38 | 63.85 | 53.07 | 32.00 | 71.31 | 52.78 | |
| Hassle-free-ALPS | 23.93 | 18.20 | 26.31 | 70.62 | 56.54 | 54.42 | 30.12 | 64.72 | 55.23 | 32.80 | 71.96 | 54.55 | |
| 3BASiL | 23.07 | 18.03 | 24.84 | 71.06 | 56.96 | 57.70 | 32.59 | 66.69 | 54.51 | 33.00 | 66.70 | 54.90 | |
| 3BASiL-TM | 18.11 | 14.26 | 20.47 | 74.05 | 61.85 | 60.73 | 34.73 | 65.98 | 54.51 | 34.80 | 76.91 | 57.94 | |
| OATS | 4:8+64LR | 16.38 | 10.88 | 17.23 | 75.84 | 67.60 | 67.09 | 41.21 | 70.88 | 60.29 | 38.20 | 73.61 | 61.84 |
| Method | Config | C4 | WT2 | PTB | PIQA | ARC-E | ARC-C |
| Wanda | 38.21 | 26.89 | 47.13 | 67.63 | 49.37 | 29.01 | |
| Wanda-TM | 15.91 | 11.03 | 17.60 | 75.14 | 63.97 | 40.19 | |
| SparseGPT | 22.65 | 16.22 | 25.15 | 71.16 | 56.48 | 32.59 | |
| SparseGPT-TM | 15.30 | 10.83 | 16.77 | 76.28 | 65.28 | 40.36 | |
| ALPS | 19.62 | 14.50 | 21.73 | 73.78 | 60.06 | 35.84 | |
| ALPS-TM | 2:4 | 14.96 | 10.65 | 16.35 | 76.88 | 65.03 | 39.85 |
| Method | C4 | WT2 | PTB | Time (hrs) |
| OATS-10 | 11.75 | 10.48 | 14.65 | 5.81 |
| Hf-SparseGPT | 11.58 | 10.17 | 14.39 | 5.97 |
| Hf-ALPS-10 | 11.56 | 10.05 | 14.33 | 4.33 |
| 3BASiL | 11.53 | 10.04 | 14.26 | 4.20 |
| Dense | 11.44 | 9.56 | 14.04 | – |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| OATS | 17.99 | 12.16 | 21.40 | 71.71 | 57.96 | 57.28 | 33.79 | 59.98 | 52.71 | 33.00 | 63.94 | 53.80 | |
| Hf-SparseGPT | 17.25 | 11.99 | 20.87 | 72.91 | 59.04 | 56.82 | 33.11 | 58.88 | 57.76 | 35.00 | 57.03 | 53.82 | |
| Hf-ALPS | 16.81 | 11.66 | 20.12 | 72.80 | 59.92 | 57.62 | 33.11 | 58.64 | 55.96 | 35.20 | 59.66 | 54.11 | |
| 3BASiL | 16.17 | 11.16 | 20.00 | 73.83 | 60.42 | 58.04 | 34.47 | 60.38 | 53.79 | 36.80 | 58.20 | 54.49 | |
| 3BASiL-TM | 50%+128 | 15.78 | 10.87 | 19.33 | 73.23 | 60.66 | 59.26 | 34.56 | 61.01 | 59.21 | 36.60 | 64.13 | 56.08 |
| Llama-3.2-1B Dense | – | 14.01 | 9.75 | 17.59 | 74.59 | 63.66 | 60.48 | 36.26 | 60.69 | 56.68 | 37.20 | 63.98 | 56.69 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Algorithm | Perplexity ( ) | Zero-shot ( ) | ||||||||||||
| C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg | ||||
| Llama3-8B | Hf-SparseGPT-original | 18.06 | 12.66 | 18.66 | 74.86 | 64.77 | 63.85 | 37.37 | 69.22 | 56.68 | 36.40 | 76.12 | 59.91 | ||
| Hf-SparseGPT-ours | 17.77 | 12.38 | 18.71 | 74.81 | 65.04 | 66.16 | 38.57 | 70.09 | 54.87 | 38.40 | 77.71 | 60.71 | |||
| EoRA-SparseGPT | 21.89 | 15.69 | 23.91 | 72.25 | 58.87 | 57.70 | 34.56 | 66.30 | 54.87 | 33.80 | 73.58 | 56.49 | |||
| Hf-ALPS-original | 16.76 | 11.83 | 17.76 | 75.08 | 66.37 | 63.64 | 37.54 | 69.69 | 64.62 | 37.20 | 77.89 | 61.50 | |||
| Hf-ALPS-ours | 16.15 | 11.38 | 16.71 | 75.19 | 67.10 | 64.44 | 38.91 | 69.53 | 59.93 | 39.40 | 78.38 | 61.61 | |||
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| OATS | 2:8+64LR | 640.86 | 605.20 | 779.86 | 52.01 | 27.67 | 28.66 | 23.12 | 49.96 | 52.71 | 25.00 | 37.83 | 37.12 |
| OATS-TM | 116.29 | 99.92 | 126.06 | 55.55 | 28.78 | 31.44 | 20.56 | 51.38 | 52.71 | 26.80 | 46.79 | 39.25 | |
| Hassle-free-SparseGPT | 162.45 | 134.21 | 170.12 | 54.19 | 28.28 | 31.40 | 22.01 | 49.17 | 52.71 | 26.80 | 57.25 | 40.23 | |
| Hassle-free-SparseGPT-TM | 74.50 | 67.59 | 88.05 | 57.24 | 30.03 | 33.12 | 21.33 | 52.33 | 52.71 | 26.20 | 60.34 | 41.66 | |
| Hassle-free-ALPS | 107.14 | 94.71 | 124.17 | 55.39 | 29.98 | 32.07 | 20.82 | 52.72 | 53.07 | 27.20 | 49.27 | 40.06 | |
| Hassle-free-ALPS-TM | 58.30 | 52.30 | 73.97 | 58.81 | 32.21 | 34.93 | 21.59 | 52.80 | 52.71 | 26.20 | 62.20 | 42.68 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| OATS | 2:8+64LR | 531.47 | 494.31 | 674.71 | 52.50 | 27.33 | 28.16 | 23.29 | 49.57 | 52.71 | 26.60 | 39.60 | 37.47 |
| OATS-TM | 100.87 | 87.20 | 120.98 | 56.64 | 29.01 | 30.13 | 20.65 | 50.99 | 52.71 | 26.00 | 62.11 | 41.03 | |
| Hassle-free-SparseGPT | 106.07 | 106.17 | 151.92 | 54.62 | 29.67 | 29.92 | 21.67 | 50.28 | 52.71 | 26.60 | 61.93 | 40.93 | |
| Hassle-free-SparseGPT-TM | 61.50 | 56.02 | 90.37 | 58.98 | 32.53 | 33.75 | 22.10 | 51.78 | 52.71 | 26.40 | 62.11 | 42.55 | |
| Hassle-free-ALPS | 69.96 | 65.34 | 108.68 | 57.34 | 32.59 | 33.59 | 20.82 | 50.67 | 52.71 | 27.00 | 62.26 | 42.12 | |
| Hassle-free-ALPS-TM | 46.12 | 44.03 | 61.25 | 61.26 | 36.36 | 36.83 | 23.12 | 52.80 | 52.71 | 25.00 | 62.51 | 43.82 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| OATS | 2:8+64LR | 424.55 | 431.81 | 590.88 | 51.63 | 28.11 | 27.61 | 23.72 | 49.72 | 52.71 | 27.00 | 38.26 | 37.34 |
| OATS-TM | 73.42 | 64.21 | 100.59 | 58.43 | 31.23 | 32.20 | 20.22 | 51.38 | 52.71 | 26.00 | 62.29 | 41.81 | |
| Hassle-free-SparseGPT | 88.39 | 96.61 | 109.71 | 54.95 | 30.77 | 31.48 | 20.65 | 50.75 | 52.71 | 26.20 | 61.59 | 41.14 | |
| Hassle-free-SparseGPT-TM | 46.24 | 42.75 | 71.66 | 61.48 | 36.76 | 36.74 | 23.12 | 53.51 | 52.71 | 27.60 | 63.15 | 44.38 | |
| Hassle-free-ALPS | 60.16 | 56.03 | 77.11 | 57.94 | 34.92 | 34.64 | 21.67 | 54.62 | 52.71 | 27.80 | 56.12 | 42.55 | |
| Hassle-free-ALPS-TM | 36.50 | 34.31 | 50.14 | 64.91 | 41.18 | 40.28 | 24.40 | 56.99 | 52.71 | 28.20 | 59.60 | 46.03 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| LFT-OATS | 2:8+64LR | 71.41 | 63.58 | 95.05 | 56.58 | 29.22 | 30.72 | 22.10 | 53.83 | 52.71 | 23.00 | 61.10 | 41.16 |
| LFT-OATS-TM | 53.01 | 47.68 | 68.98 | 59.09 | 31.07 | 36.15 | 21.84 | 52.49 | 52.71 | 24.60 | 62.14 | 42.51 | |
| LFT-Hassle-free-SparseGPT | 51.19 | 47.09 | 63.87 | 59.03 | 31.16 | 35.14 | 22.18 | 50.83 | 52.71 | 26.60 | 60.70 | 42.29 | |
| LFT-Hassle-free-SparseGPT-TM | 42.85 | 39.46 | 55.53 | 60.55 | 33.37 | 35.27 | 22.18 | 50.36 | 52.71 | 28.00 | 62.14 | 43.07 | |
| LFT-Hassle-free-ALPS | 44.15 | 41.02 | 57.45 | 59.36 | 33.22 | 34.68 | 22.18 | 51.93 | 52.71 | 26.60 | 60.95 | 42.70 | |
| LFT-Hassle-free-ALPS-TM | 37.54 | 35.50 | 50.15 | 60.28 | 35.80 | 37.50 | 23.63 | 55.09 | 52.71 | 27.20 | 62.14 | 44.29 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| LFT-OATS | 2:8+64LR | 48.53 | 44.51 | 62.21 | 58.27 | 32.84 | 36.95 | 22.78 | 52.96 | 52.71 | 25.60 | 61.44 | 42.94 |
| LFT-OATS-TM | 100.87 | 87.20 | 120.98 | 56.64 | 29.01 | 30.13 | 20.65 | 50.99 | 52.71 | 26.00 | 62.11 | 41.03 | |
| LFT-Hassle-free-SparseGPT | 35.91 | 33.44 | 46.64 | 61.48 | 37.33 | 36.70 | 23.46 | 51.54 | 52.35 | 26.20 | 62.17 | 43.90 | |
| LFT-Hassle-free-SparseGPT-TM | 31.46 | 30.77 | 42.12 | 62.79 | 40.19 | 38.76 | 23.63 | 54.78 | 53.07 | 29.00 | 58.32 | 45.07 | |
| LFT-Hassle-free-ALPS | 31.43 | 29.50 | 44.39 | 63.38 | 40.65 | 39.73 | 23.46 | 55.72 | 52.71 | 28.40 | 46.06 | 43.76 | |
| LFT-Hassle-free-ALPS-TM | 28.22 | 26.81 | 36.23 | 65.02 | 43.69 | 40.78 | 23.46 | 55.72 | 52.71 | 28.40 | 60.46 | 46.28 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| LFT-OATS | 2:8+64LR | 37.46 | 31.49 | 49.71 | 62.24 | 36.88 | 38.72 | 24.23 | 51.93 | 52.71 | 27.20 | 62.02 | 44.49 |
| LFT-OATS-TM | 28.23 | 23.86 | 36.25 | 65.61 | 43.81 | 42.09 | 25.26 | 52.64 | 52.71 | 30.60 | 64.13 | 47.11 | |
| LFT-Hassle-free-SparseGPT | 28.80 | 24.47 | 33.94 | 62.35 | 43.30 | 40.28 | 24.91 | 54.85 | 53.07 | 29.40 | 64.46 | 46.58 | |
| LFT-Hassle-free-SparseGPT-TM | 24.89 | 22.14 | 31.65 | 66.00 | 48.49 | 43.77 | 26.37 | 58.25 | 52.71 | 29.80 | 66.27 | 48.96 | |
| LFT-Hassle-free-ALPS | 25.31 | 21.97 | 31.99 | 66.38 | 48.42 | 43.73 | 26.62 | 59.43 | 53.43 | 29.60 | 47.16 | 46.85 | |
| LFT-Hassle-free-ALPS-TM | 22.85 | 20.23 | 28.00 | 67.95 | 52.18 | 46.84 | 27.99 | 60.54 | 53.43 | 31.20 | 61.44 | 50.20 |
| Method | Config | C4 | WT2 | PTB | PIQA | HS | ARC-E | ARC-C | WG | RTE | OQA | BoolQ | Avg |
| 3BASiL | 25.31 | 20.59 | 28.33 | 71.27 | 55.56 | 54.25 | 30.80 | 64.56 | 53.43 | 33.20 | 72.91 | 54.50 | |
| 3BASiL+OWL | 23.21 | 19.54 | 27.09 | 71.71 | 57.90 | 56.69 | 33.53 | 67.56 | 54.15 | 33.40 | 77.86 | 56.60 | |
| 3BASiL-TM | 19.55 | 16.44 | 21.64 | 74.16 | 59.76 | 58.04 | 33.28 | 66.38 | 58.12 | 35.20 | 73.15 | 57.26 | |
| 3BASiL-TM+OWL | 70% + 64 | 19.52 | 16.22 | 21.37 | 73.56 | 59.92 | 58.88 | 31.40 | 64.33 | 56.32 | 35.40 | 70.80 | 56.33 |
| 3BASiL | 62.85 | 61.08 | 79.49 | 59.52 | 35.07 | 35.40 | 22.27 | 54.22 | 52.71 | 27.00 | 60.95 | 43.39 | |
| 3BASiL+OWL | 50.51 | 58.16 | 79.09 | 61.70 | 39.42 | 37.63 | 23.98 | 59.35 | 55.60 | 28.00 | 68.23 | 46.74 |
LLM Compression
Large language model compression reduces the memory, compute, and context costs of LLMs while aiming to preserve their capabilities.
See all 273 papers