Beyond Activation Alignment:The Alignment-Diversity Tradeoff in Task-Aware LLM Quantization
About
Mixed-precision quantization (MPQ) has become a key technique for deploying large language models under stringent memory and compute constraints. We first identify a phenomenon that we term the Perplexity Illusion: layers ranked as important by perplexity-based sensitivity show little rank correlation with those that are most influential for complex reasoning performance, with Kendall $\tau \approx 0$ in our analysis. We further reveal an Alignment-Diversity Tradeoff: using only target-task calibration data can degrade post-quantization performance, whereas incorporating general-domain data stabilizes sensitivity estimation and improves robustness across tasks. Based on these observations, we propose TASA (Task-Aware Sensitivity Analysis), a two-level framework that jointly optimizes calibration-data composition and mixed-precision bit allocation. Specifically, TASA searches for a calibration-data mixture using a training-free gradient-trace alignment criterion, and then aggregates perplexity and reasoning-oriented sensitivity signals to guide both inter-layer and intra-layer bit allocation. Experiments on LLaMA-3-8B and Qwen2.5-7B reveal a precision inversion: appropriately allocated 3.5-bit models can match or surpass less task-aware 4-bit baselines. At an average precision of 3.5 bits, TASA matches or outperforms several competitive 4-bit uniform baselines in aggregate accuracy, and improves over the strongest W3 baseline on GSM8K by more than 20 absolute points on LLaMA-3-8B. These results show that calibration-data composition substantially affects task-sensitive quantization, a factor underexplored in prior work.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Physical Interaction Question Answering | PIQA | Accuracy78.7 | 462 | |
| Multiple-choice Question Answering | ARC Easy | Accuracy79.2 | 269 | |
| Commonsense Reasoning | HellaSwag | HS Accuracy78.6 | 44 | |
| Commonsense Reasoning | WinoGrande (WG) | Accuracy71.9 | 38 | |
| Language Modeling | Language Modeling Dataset | PPL9.02 | 31 | |
| Mathematical Reasoning | GSM8K TASA TACQ (test) | Accuracy79.2 | 14 | |
| Zero-shot Reasoning and Language Modeling | LLaMA-3-8B Evaluation Suite ARC-C, HellaSwag, WinoGrande, PIQA, ARC-E, BoolQ, GSM8K, WikiText-2 (test) | ARC-C Accuracy52.3 | 13 | |
| Multiple-choice Question Answering | ARC Challenge | ARC-C Accuracy51.5 | 12 | |
| Reading Comprehension | BoolQ | BoolQ Score84.4 | 12 | |
| General Language Understanding | 7-Task Evaluation Suite (test) | Average Accuracy74.1 | 10 |