Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families

summary

Video file (mp4)

The gist

LLM multi-agent systems split work across agents that communicate through text, which constrains communication to discrete tokens and costs redundant compute, limiting bandwidth and capabilities of

In short

SwaT (Swarm Telepathy) is a method for multi-agent systems to communicate by transferring an agent's internal KV-cache directly to another agent, eliminating the need for prefilling text. This cross-family transfer method achieves higher accuracy and speed than traditional text-to-text communication, even allowing vision models to transfer capabilities.

Key concepts

KV-Cache Transfer
This is a technique where an agent sends its internal memory state, known as the Key-Value (KV) cache, to another agent instead of sending raw text. This allows for direct knowledge sharing between different language models without requiring the receiver to process input text first.
Mapper M_i→j
This is a learned function that translates the sender's KV-cache into a format that approximates the receiver's native prefill cache. The system learns this mapping during training so it can efficiently transfer knowledge across different model families and architectures.
KV-Cache Compaction (SwaT+AM)
This involves combining SwaT with compaction techniques to reduce the size of the transferred cache while preserving its performance. This combination significantly boosts speed and accuracy, making the communication more practical for real-world multi-agent applications.

Terminology used across episodes

This episode discusses

The paper

Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families · Read on arXiv

Abdulrahman Alabdulkareem, Irene Tenison Yang Zhang, Yada Zhu, Boris Katz, Dan Gutfreund

MIT CSAIL · MIT-IBM Computing Research Lab

LLM multi-agent systems split work across agents that communicate through text. An agent summarizes its findings, another agent receives the text. This constrains communication to discrete tokens and costs redundant compute, limiting bandwidth and capabilities of multi-agent systems. Recent works demonstrate KV-cache transfer between LLMs but they either require the LLMs to be within the same family or cannot function across context, thus requiring the receiver to prefill text. We present SwaT (Swarm Telepathy), the first cross-family KV-cache transfer method that allows prefill-free communication in multi-family multi-agent systems. In an agent-to-agent communication setting, combining SwaT with recent advances in KV-cache compaction beats text-to-text communication in both accuracy and speed, with a mean of 40% higher accuracy and a mean of 4 times faster wall-clock time. Experiments show SwaT achieves high performance on 17 benchmarks across 12 LLMs and 6 families. Additionally, given a sender Vision-Language Model (VLM), SwaT is able to transfer the vision capabilities to a text-only receiver LLM exceeding text-to-text communication despite SwaT being trained on text-only data. SwaT keeps both sender and receiver models frozen without fine-tuning and only requires <50M tokens and <1 GPU-day to train.

Transcript

Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.

Tom: I'm Tom.

Jane: And I'm Jane, here with Lu, Meng and our in-house large language model, Lalam.

Tom: Today's paper: "Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families".

Jane: LLM multi-agent systems split work across agents that communicate through text, which constrains communication to discrete tokens and costs redundant compute, limiting bandwidth and capabilities of multi-agent systems The gist.

Tom: First, who's behind it and why it matters.

Title and authors: Tom: Moving on from the mechanics, let’s look at what they actually found in terms of performance and capabilities with this Swarm Telepathy method. The big takeaway here is that SwaT beats text-to-text communication in both accuracy and speed when we look across seventeen benchmarks for twelve LLMs and six families >

Jane: They also demonstrated that SwaT can transfer vision capabilities to a text-only receiver LLM, which is pretty significant because SwaT was trained mostly on text data >

Lu: And they also showed specialized knowledge transfer across languages, where models like Qwen three 4B can transfer knowledge to Llama three 8B >

Meng: That specific example about closing three quarters of the gap on subjects like CMMLU shows that this isn't just a general communication trick; it’s transferring actual specialized skills >

Tom: And they also showed increased context handling beyond the receiver's native context window by chunking and transferring KV-cache across layers >

Jane: And they also showed retention is high, even in the small-to-large pairing where the large model has to use a smaller model's cache compared to using its own native cache >

Lu: The fact that SwaT shows success across many model-pairs and families suggests that what they are transferring is fundamental knowledge that remains useful regardless of how it’s expressed >

Meng: It seems like the core idea here is making the transferred state more robust than just a simple text summary, which I think is a necessary step for real utility >

Tom: So we're looking at a method that boosts accuracy, speed, and capability by using learned mappings to move KV-caches directly between agents without any prefill cost for the receiver >

Jane: It fundamentally changes how agents can talk, moving from discrete tokens to transferring actual latent knowledge states without re-doing all the setup work >

Lu: And they also showed that combining this transfer with something called KV-cache compaction, SwaT plus AM, makes it significantly faster and more accurate than just sending the raw cache state >

Meng: So if a model has an 8k limit, this method lets it process information that exceeds that limit by moving the cache around its layers. That’s a big feature for long contexts >

Tom: It’s about being able to handle much longer documents than the native context window allows for, which is something we all struggle with when dealing with huge inputs >

Tom: So we're looking at a method that boosts accuracy, speed, and capability by using learned mappings instead of just sending raw text or relying on simple prefilling >

Jane: It fundamentally changes how agents can talk, moving from discrete tokens to transferring actual latent knowledge states without re-doing all the setup work.

The paper's summary: Tom: Now let’s talk about what they suggest as improvements to push this research further with Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families. They suggest that the main implication is that we can achieve more accurate and faster agent communication than text-to-text methods by transferring internal states directly instead of sending long text prompts back and forth >

Jane: They suggest that this method proves that cross-family KV-cache transfer is viable without receiver prefill, which opens up a multitude of optimizations in the field >

Lu: This method proves that cross-family KV-cache transfer is viable without receiver prefill, which opens up a multitude of optimizations in the field >

Meng: For deployment at scale, I just need to figure out how to manage those training costs for all those individual mappers effectively across many agents >

Lalam: From my perspective, this is a solid direction for making AI systems more integrated and capable of sharing knowledge internally >

Tom: They suggest having a shared latent space to train those mappers so you don't need one mapper per pair, which would total to only 2n mappers being needed >

Jane: That’s interesting because it addresses the scaling issue mentioned earlier, suggesting a way to make this much more scalable for massive agent swarms >

Lu: If they can solve the problem of training those mappers in a more shared latent space, then this whole cross-family transfer thing becomes much more scalable >

Meng: I’m still focused on the engineering challenge; if we can reduce the number of mappers needed, that helps immensely with managing compute for real-world deployment >

The paper's improvements: Tom: To wrap up this discussion on Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families, the main implication is that we can achieve more accurate and faster agent communication than text-to-text methods >

Jane: It’s a way to enable agents to have deeper, faster communication pathways by transferring internal states directly instead of relying on text exchange >

Tom: And they showed that combining this transfer with something called KV-cache compaction, SwaT plus AM, makes it significantly faster and more accurate than just sending the raw cache state >

Jane: The results on those benchmarks across seventeen models and twelve families show that the performance is actually quite strong compared to standard text-to-text methods >

Lu: It’s also notable how this transfer works even when you try to move capabilities across different model families, like moving vision knowledge from a Vision Language Model to a text-only one >

Meng: But I gotta point out the limitation they mentioned—each specific sender and receiver pair requires a trained mapper, which gets really hard for huge multi-agent systems with thousands of agents >

Lalam: That’s the bottleneck I see; scaling those individual mappings up to a massive swarm is definitely where we need to focus next >

Tom: So the main point is that SwaT achieves better accuracy and speed by using a learned mapping instead of just sending raw text or relying on simple prefilling >

Meng: I'm just thinking about how much compute that learning for each pair adds up when you try to manage a hundred agents all communicating >

Lalam: I’m confident the evidence shows this is a solid step forward for prefill-free communication, even with those scaling hurdles left to tackle.

Tom: Exactly. So we're looking at Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families as a way to build smarter, more interconnected agent systems where agents talk using their internal states instead of just exchanging text strings.

Jane: It shows that the way agents communicate can be fundamentally different than just exchanging text strings.

Conclusion: Tom: So we’ve just talked through Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families, and the main implication is that we can get faster, more accurate agent communication by moving internal states directly instead of sending long text prompts back and forth.

Jane: It’s a way to enable agents to have deeper, faster communication pathways by transferring those internal states directly instead of relying on those discrete tokens that text communication forces you into.

Meng: For deployment at scale, I just need to figure out how to manage those training costs for all those individual mappers effectively across many agents >

Lalam: From my perspective, this is a solid direction for making AI systems more integrated and capable of sharing knowledge internally >

Tom: They showed that combining SwaT with KV-cache compaction, SwaT plus AM, makes it significantly faster and more accurate than just sending the raw cache state >

Jane: The results on those benchmarks across seventeen models and twelve families show that the performance is actually quite strong compared to standard text-to-text methods >

Lu: It’s also notable how this transfer works even when you try to move capabilities across different model families, like moving vision knowledge from a Vision Language Model to a text-only one >

Meng: But I gotta point out the limitation they mentioned—each specific sender and receiver pair requires a trained mapper, which gets really hard for huge multi-agent systems with thousands of agents >

Lalam: That’s the bottleneck I see; scaling those individual mappings up to a massive swarm is definitely where we need to focus next >

Tom: Exactly. So we're looking at Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families as a way to build smarter, more interconnected agent systems >

Jane: It shows that the way agents communicate can be fundamentally different than just exchanging text strings.

More episodes

← Home