GRINQH: Graded Input-based Quantization Hierarchy for Efficient LLM Generation
About
Autoregressive decoding with LLMs is primarily bottlenecked by GPU memory bandwidth, especially in edge-computing settings. While quantization is essential for mitigating this bottleneck, most existing methods treat inference as a uniform process and fail to account for the asymmetry between the compute-bound prefill stage and the memory-bound decoding stage. We propose GRINQH (GRaded INput-based Quantization Hierarchy), a weight-only post-training quantization framework that accelerates decoding by unifying quantization and sparsification. GRINQH leverages activation magnitudes as a proxy for computational importance to dynamically assign weight channels to different precision levels, enabling flexible average bit widths during decoding. Evaluated on Llama3 and Qwen3 models, GRINQH outperforms state-of-the-art fixed- and mixed-precision baselines at comparable 3- and 4-bit settings, even enabling effective 2-bit generation. We experimentally verify theoretical speedups by leveraging a hierarchical nested memory layout for multi-precision storage in a custom GPU kernel. Ultimately, GRINQH establishes a new state-of-the-art Pareto frontier for LLM generation, enabling a dynamic trade-off between generation quality and inference speed.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy79.21 | 897 | |
| Physical Interaction Question Answering | PIQA | Accuracy80.47 | 462 | |
| Language Modeling | Wiki2 | PPL8.95 | 382 | |
| Language Modeling | LAMBADA | Perplexity3.36 | 254 | |
| Reasoning | ARC Easy | -- | 242 | |
| Commonsense Reasoning | WinoGrande | Accuracy74.66 | 61 | |
| General Language Understanding | LAMBADA, ARC-C, HellaSwag, BoolQ, MMLU, GSM8K Average | Accuracy74.34 | 56 | |
| Reasoning | ARC Challenge | Accuracy (ARC)0.5614 | 56 |