PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
September 29, 20250 citationsOpen Access

NSNQuant: A Double Normalization Approach for Calibration-Free Low-Bit Vector Quantization of KV Cache

View Full Paper
DSDonghyun SonECEuntae ChoiSYSungjoo Yoo

Key Points

  • NSNQuant achieves up to 3x throughput improvement compared to full-precision methods, enhancing KV cache efficiency.
  • The technique employs a novel double normalization approach with Hadamard transform to ensure calibration-free VQ.
  • This study demonstrates strong generalization capabilities for low-bit vector quantization across various settings.
  • Extensive experiments validate NSNQuant's superiority over traditional methods in both 1-bit and 2-bit quantization.

Abstract

Large Language Model (LLM) inference is typically memory-intensive, especially when processing large batch sizes and long sequences, due to the large size of key-value (KV) cache. Vector Quantization (VQ) is recently adopted to alleviate this issue, but we find that the existing approach is susceptible to distribution shift due to its reliance on calibration datasets. To address this limitation, we introduce NSNQuant, a calibration-free Vector Quantization (VQ) technique designed for low-bit compression of the KV cache. By applying a three-step transformation-1) a token-wise normalization (Normalize), 2) a channel-wise centering (Shift), and 3) a second token-wise normalization (Normalize) -with Hadamard transform, NSNQuant effectively aligns the token distribution with the standard normal distribution. This alignment enables robust, calibration-free vector quantization using a single reusable codebook. Extensive experiments show that NSNQuant consistently outperforms prior methods in both 1-bit and 2-bit settings, offering strong generalization and up to 3 throughput gain over full-precision baselines.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Son et al. (2025) studied this question.

synapsesocial.com/papers/68da58d8c1728099cfd10fe3https://doi.org/10.48550/arxiv.2505.18231
Ask AI
Helpful
Bookmark
Share
View Full Paper