MEGA Hub

WnW: Waxing-and-Waning KV Cache for Long-Form Speech LLMs

Authors

Do you know Yiming Yao?You can claim authorship or link another user.Do you know Chenyang Lyu?You can claim authorship or link another user.Do you know Xuanfan Ni?You can claim authorship or link another user.Do you know Longyue Wang?You can claim authorship or link another user.Do you know Weihua Luo?You can claim authorship or link another user.Do you know Yazheng Yang?You can claim authorship or link another user.Do you know Jinsong Su?You can claim authorship or link another user.

Abstract

Long-form audio inputs make the KV cache the dominant memory cost of speech LLMs. Prefill-only KV compression methods permanently discard audio KV positions once evicted, with no pathway to recover them during decoding. We show this is fragile on long-form audio: prefill attention concentrates near the audio start (an attention-sink effect), while decode-time attention distributes broadly, and the two rankings overlap weakly. We propose WnW (Waxing-and-Waning KV cache), which classifies KV-heads into anchor, tidal, and fixed roles via offline calibration. Anchor heads remain on GPU and serve as a decode-time importance observer; tidal heads keep a CPU-resident complement that is recalled chunk-by-chunk based on aggregated anchor-head scores; fixed heads keep only an on-GPU subset, with the rest permanently discarded. On LibriSpeech-Long with two 3B backbones (Voxtral-mini-3b and Qwen2.5-Omni-3B), WnW preserves near-Full-Cache accuracy while keeping only 20% of audio tokens on GPU, where prefill-only baselines fail to terminate. Results generalize across language, task, and domain shifts, and CPU-GPU recall adds little decode-time overhead in our measurements.

Community

00

Publication notes

Author note
Accepted at EMNLP 2026 Main Conference. 8 pages, 5 figures