Join the conversation

Join the community of Machine Learners and AI enthusiasts.

Sign Up
yunblack 
posted an update 10 days ago
Post
110
Can LLMs from different model families directly share KV caches, without receiver-side prefill?

Our answer is HeteroFold.

I am so excited to share our new paper: Prefill-Free Cross-Family KV Cache Transfer for Heterogeneous Multi-Agent LLMs!

Recent work on prefill-free KV cache transfer has shown that LLMs can reuse previously computed context instead of repeatedly performing receiver-side prefill. However, existing approaches have largely focused on models within the same family or with compatible tokenization.

We introduce HeteroFold, a method for prefill-free cross-family KV cache transfer that enables models such as Llama, Qwen, and Ministral to directly share KV caches despite differences in tokenizers, model depth, KV structure, and representation spaces.

HeteroFold aligns tokens across different tokenizers and layers across different architectures, maps the sender’s K/V states into the receiver’s representation space, and calibrates the transferred cache to preserve the receiver’s attention patterns and outputs. Both the sender and receiver remain frozen, and the learned mappings allow the receiver to directly decode from the transferred cache without processing the original context again.

At 32K context length, HeteroFold achieves about 10.7× faster transfer than native receiver prefill. We evaluate all six transfer directions among Llama, Qwen, and Ministral, with strong results across long-context, short-context, and multi-agent settings.

What excites me most is the possibility of making KV caches reusable across model-family boundaries. Instead of different LLMs repeatedly recomputing the same context, heterogeneous models can directly reuse computation from one another.

Paper: https://arxiv.org/pdf/2609.32259

The 10.7x is the wrong headline. Direction is the story.

I divided HeteroFold by TextMas (native prefill) per direction, averaged over Qasper, HotpotQA and LoCoMo, from Tables 1 and G.

Down, bigger sender into smaller receiver:
Ministral-14B→Qwen-4B 94%, Llama-8B→Qwen-4B 87%, Ministral-14B→Llama-8B 82%.

Up:
Qwen-4B→Llama-8B 63%, Llama-8B→Ministral-14B 55%, Qwen-4B→Ministral-14B 48%.

Every down direction beats every up direction.

Your same-family Qwen3 table pulls size apart from family.
14B→4B keeps 97% in-family, 94% across families.
4B→14B keeps 79% in-family, 48% across families.
Crossing families costs about 3 points going down and 30 going up.

The 10.7x row is Llama-8B→Ministral-14B, an upward pair, at 47% of native Qasper F1.

My guess is Token Alignment. Your own example has Ministral splitting "unbelievable" finer than Qwen, and every unmatched receiver boundary reuses the last shared sender state. A finer receiver gets more copied states.

What fraction of receiver tokens hit that fallback, per direction?