Draft-KV: Learning Useful Latent Communication Between Language Models
Abstract
Latent communication passes internal states between language models instead of decoded text, but higher receiver accuracy does not show that the receiver used the message content. Across five method–dataset pairs, replacing each message with one from an unrelated question changes accuracy by at most points, even when communication adds points over the receiver alone. Thus the interface can supply the gain while making the sharer dispensable. instead sends the key–value states formed while the sharer drafts an answer to the current question. Linear projections place these states in a side memory read through a gated attention branch, and progressive training moves from message reconstruction to answer supervision under a guard on harm from mismatched messages. Both models remain frozen and the interface trains M parameters, fewer than C2C. With a Qwen3-8B sharer, a frozen Qwen2.5-0.5B-Instruct receiver reaches on MMLU-Redux, versus alone and with reassigned messages. At fixed interface size, scaling the sharer from B to B raises accuracy from to ; communication also transfers to held-out tasks and can exceed both models when each holds different evidence.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.