🚀Excited to share our latest #
EMNLP2024# work on benchmarking the long context ability with KV Cache compression across RNN-based architectures, token eviction, prompt compression, and quantization. We also provide an easy-to-use codebase (it also has my favorite WoW quote 😉). Feel free to give it a try and ⭐ it if you find it useful!
📄 Paper:
💻 Code:
Some interesting findings/suggestions include:
1️⃣ Maintaining an uncompressed prefill process is essential for performance, especially with harder tasks.
2️⃣ Combining RNN-based models with attention significantly enhances long-context capabilities.
3️⃣ In "needle-in-a-haystack" evaluation for recent LLMs like Llama-3, we should use longer needles (like 64 digits) since these models tokenize multiple digits into one token.
More results and insights can be found in the paper!
Kudos to all collaborators:
@jiayiy, Hongyi Liu,
@henryzhongsc,
@YuNengChuang, Songchen Li, Guanchu Wang, Duy Le,
@serendip410, Vipin Chaudhary,
@ZhaozhuoX,
@ziruirayliu,
@huxia