English

CALM: A CKA-Guided Adaptive Layer-Wise Modularization Framework for LLM Quantization

Machine Learning 2026-01-09 v2 Artificial Intelligence

Abstract

Current mainstream post-training quantization methods for large language models typically apply a uniform quantization strategy across all network layers, overlooking the substantial differences in algorithmic suitability among layers. To address this limitation, we propose CALM (A CKA-guided Adaptive Layer-wise Modularization)a fine-tuning-free, plug-and-play framework for algorithmic heterogeneous quantization. CALM independently evaluates multiple PTQ algorithms on each layer and employs Linear Centered Kernel Alignment (CKA) as a metric to automatically select the optimal quantization strategy per layer. The individually optimized strategies are then integrated to construct a hybrid quantized model. Experiments demonstrate that our approach consistently outperforms both uniform quantization baselines and state-of-the-art mixed-precision methods across mainstream LLMsincluding LLaMA and Qwenin terms of perplexity (PPL) and downstream task performance.

Keywords

Cite

@article{arxiv.2512.16282,
  title  = {CALM: A CKA-Guided Adaptive Layer-Wise Modularization Framework for LLM Quantization},
  author = {Jinhao Zhang and Yunquan Zhang and Daning Chen and JunSun and Zicheng Yan},
  journal= {arXiv preprint arXiv:2512.16282},
  year   = {2026}
}
R2 v1 2026-07-01T08:30:52.383Z