arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.11236cs.CL

XQuant:通过跨层压缩实现超低比特 KV 缓存量化

XQuant: Achieving Ultra-Low Bit KV Cache Quantization with Cross-Layer Compression

School of Artificial Intelligence, Wuhan University(武汉大学人工智能学院);School of Computer Science, Wuhan University(武汉大学计算机学院);School of Computer Science, Shanghai Jiao Tong University(上海交通大学计算机学院);Xiaomi Inc.(小米公司)

Haoqi Yang, Yao Yao, Zuchao Li, Baoyuan Qi, Guoming Liu, Hai Zhao

AI总结:

本文提出无需训练的即插即用框架XQuant,通过免数据校准与跨层压缩实现超低比特KV缓存量化,在保持模型精度的同时显著降低内存消耗。

摘要:

大型语言模型(LLMs)在各类自然语言处理任务中展现出卓越能力。然而,其庞大的内存需求,尤其是长文本理解和生成过程中 KV 缓存的不断增长,给在资源受限环境中的部署带来了巨大挑战。量化已成为一种极具前景的解决方案,能在保留历史信息的同时降低内存消耗。我们提出了 XQuant,这是一个无需训练且即插即用的框架,可实现超低等效比特宽度的 KV 缓存量化。XQuant 引入了两项关键创新:计算开销可忽略的免数据校准方法以及跨层 KV 缓存压缩,使量化能够降至 1.4 比特以下。在 TruthfulQA 和 LongBench 上的大量实验表明,XQuant 通过在保持卓越性能的同时实现更低比特宽度,超越了现有最先进方法(例如 KIVI-2bit 和 AsymKV-1.5bit),在内存效率与模型精度之间建立了更优的平衡。

英文摘要:

Large Language Models (LLMs) have demonstrated remarkable capabilities across diverse natural language processing tasks. However, their extensive memory requirements, particularly due to KV cache growth during long-text understanding and generation, present significant challenges for deployment in resource-constrained environments. Quantization has emerged as a promising solution to reduce memory consumption while preserving historical information. We propose XQuant, a training-free and plug-and-play framework that achieves ultra-low equivalent bit-width KV cache quantization. XQuant introduces two key innovations: a computationally negligible data-free calibration method and cross-layer KV cache compression, enabling quantization to sub-1.4 bits. Extensive experiments on TruthfulQA and LongBench demonstrate that XQuant outperforms state-of-the-art methods (e.g., KIVI-2bit and AsymKV-1.5bit) by achieving lower bit-width while maintaining superior performance, establishing a better trade-off between memory efficiency and model accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑