Update README
Browse files
README.md
CHANGED
|
@@ -1,3 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
-
---
|
|
|
|
| 1 |
+
**KVQuant** is a methodology for efficient KV cache quantization that incorporates several innovations to acheive accurate low-precision quantization,
|
| 2 |
+
thereby enabling efficient long context length inference.
|
| 3 |
+
|
| 4 |
+
**TLDR:** KVQuant addresses the memory bottleneck with long context length inference by quantizing the KV cache to low precision.
|
| 5 |
+
KVQuant achieves high accuracy with low-precision KV cache quantization by considering several consistent patterns observed in cached KV values across different LLMs,
|
| 6 |
+
and by developing methods to exploit these patterns, including:
|
| 7 |
+
|
| 8 |
+
- **Per-channel, Pre-RoPE** Key quantization to better match the outlier channels in Keys
|
| 9 |
+
- Non-Uniform Quantization (**NUQ**) to better represent the non-uniform activations
|
| 10 |
+
- **Dense-and-Sparse Quantization** to mitigate the impacts of numerical outliers on quantization difficulty
|
| 11 |
+
- **Q-Norm** to mitigate distribution shift at ultra low precisions (eg. 2-bit)
|
| 12 |
+
- **Attention-Sink Aware Quantization** to avoid quantization error with the first token, which is disproportionately sensitive to quantization error
|
| 13 |
+
|
| 14 |
+
For more details please check out our [paper](https://arxiv.org/abs/2401.18079.pdf).
|
| 15 |
+
|
| 16 |
+
## Model description
|
| 17 |
+
|
| 18 |
+
Quantizer file for running DBRX with 2-bit KV cache using KVQuant.
|
| 19 |
+
|
| 20 |
+
* **Base Model:** [DBRX-Instruct](https://www.databricks.com/blog/introducing-dbrx-new-state-art-open-llm)
|
| 21 |
+
* **Bitwidth:** 2-bit
|
| 22 |
+
* **Sparsity Level:** 1%
|
| 23 |
+
|
| 24 |
+
## Links
|
| 25 |
+
|
| 26 |
+
* **Paper**: [https://arxiv.org/abs/2401.18079.pdf](https://arxiv.org/abs/2401.18079.pdf)
|
| 27 |
+
* **Code**: [https://github.com/SqueezeAILab/KVQuant](https://github.com/SqueezeAILab/KVQuant)
|
| 28 |
+
|
| 29 |
---
|
| 30 |
license: mit
|
| 31 |
+
---
|