Baidu's Virtual Head Attention tames the KV cache, FLOPs, quality triangle in long-context LLMs

baidu · hf · 2026-10-16

A Baidu and Peking University paper proposes Virtual Head Attention (VHA), addressing the "impossible triangle" of KV cache size, training/prefill FLOPs, and quality in long-context attention. VHA uses two lightweight linear ops — Q Premix, a near-identity query transform that recovers virtual head diversity from halved physical heads, and Linear Postmix, a low-rank I+AB^T residual fusion — improving all three at once with negligible inference overhead since the ops fold into existing projections.

Original post →

More from Infra

Infra channel →