Calibration Data Trade-offs Across Capability Dimensions: Why Multi-Source Mixing Matters for High-Sparsity LLM Pruning
About
Post-training pruning compresses large language models to high sparsity using a small unlabelled calibration set, and recent work has concluded that the choice of calibration source has only modest impact on averaged post-pruning accuracy. We ask whether this conclusion survives once calibration impact is evaluated separately across distinct capability dimensions rather than aggregated. Decomposing post-pruning capability into General, Commonsense, Code, and Math, and analysing $n{=}15$ calibration sources via Spearman correlations between OIT information metrics and per-dimension retention, we uncover an opposite-sign trade-off: calibration perplexity correlates positively with General retention ($\rho{=}{+}0.71$) but negatively with Math and Code retention ($\rho{=}{-}0.53,\,{-}0.59$; $p{<}0.05$), so no single source can preserve all capabilities. We respond with multi-source calibration mixing, and propose IGSP, an information-guided self-calibration protocol that automates multi-source construction without capability-aligned corpora by minimising 4-gram aggregation and balancing perplexity across dimensions. On LLaMA-3.1-8B at SparseGPT 60% sparsity, a uniform multi-source mix reaches 58.8% total retention, outperforming the best single source (MetaMath, 50.0%) by $+8.8$ and the C4 default (40.0%) by $+18.8$; IGSP improves over Self-Cal by $+2.4$ and SGS by $+4.8$.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Total Retention | LLaMA-3.1-8B Capability Evaluation Suite | Total Retention (S_m^total)97.9 | 52 | |
| Large Language Model Pruning Calibration | OIT-PC Evaluation Suite | Generation Score (SGen)98.85 | 31 | |
| LLM Pruning Evaluation | General Dimension LAMBADA-Std, TriviaQA | Relative Retention (Generation)79.3 | 12 | |
| Commonsense Reasoning | Capability Evaluation Suite Commonsense Pool | S_Com Score97.03 | 4 | |
| Mathematical Reasoning | Capability Evaluation Suite Math Pool | S_Math Score69.12 | 4 | |
| Overall Capability | Capability Evaluation Suite Aggregate | Overall Capability Score82.69 | 4 | |
| Coding Capability | Capability Evaluation Suite Code Pool | S_Code Score74.63 | 4 | |
| General Capability | Capability Evaluation Suite General Pool | S_Gen Score89.99 | 4 | |
| LLM Pruning Evaluation | Commonsense Dimension PIQA, ARC-Easy | Relative Retention (S_Com)94.2 | 3 | |
| LLM Pruning Evaluation | Code Dimension HumanEval MBPP | S_Code Relative Retention72.3 | 2 |