Daily Tech Briefing
AI 科技速览

每天 5 分钟内学习 AI。获取最新的人工智能新闻,理解其重要性,并学习如何将其应用于您的工作。

AI 快讯
arXiv Machine Learning · 2026/8/4 16:26:47

Cross-Model KV Cache Transfer in LLM Families: A Closed-Form Linear Mapping for Prefill Reuse

AI 中文解读
核心亮点:这项研究让AI大模型之间能“共享记忆”,换模型时不用从头开始重新计算,大幅节省时间和算力。 通俗解读:想象你正和一位专家聊天,中途换了个更资深的专家接手。以前新专家得从头听你复述所有背景,现在有了这个技术,新专家能直接“继承”旧专家的记忆,无缝衔接对话。研究者发现,同一家族的不同尺寸模型,它们的“记忆存储”有很强的相似性,通过一个简单的数学公式就能把旧模型的记忆“翻译”给新模型用,省去了重新阅读全部内容的步骤。 实际影响:以后你使用AI助手时,如果服务商为了平衡成本和效果,在对话中途切换了更小或更大的模型,你完全感觉不到卡顿或“失忆”。这会让AI服务更流畅、更便宜,尤其对长对话或复杂任务来说,响应速度会明显提升,用户体验更佳。
Production deployments often swap between different-sized models in a family for cost-quality cascading, mid-conversation switching, and routing, and each swap forces the receiver to repay the prefill from scratch. We propose cross-model KV cache transfer, where the receiver reuses the source's KV cache, skipping prefill. We find that cross-model KV has substantial linear structure across matched-KV pairs, where source and target share KV head count and per-head dimension. On Qwen3 14B->32B, one source layer explains 56% of variance in the target's keys and 32% in values, rising to 79% and 65% with multiple source layers. Building on this, we design a closed-form ridge mapper that operates per head and proceeds in three steps. First, for each target layer we select the top-k most predictive source layers and concatenate their KV as input. Second, we strip RoPE from the keys before mapping, so the fit is position-free and reusable across context lengths. Third, we fit ridge regression on a small calibration set of 500 FineWeb-Edu sequences of 1,024 tokens each. Surprisingly, across six pairs in three families, this linear mapper retains 73-98% of the receiver's standalone-prefill accuracy on four pairs, while two degrade sharply. A nonlinear MLP recovers up to +37 pp HellaSwag retention on the failures. The mapper runs 2.7-25x faster than re-prefill and remains stable across multi-turn handoff, making cross-model KV cache transfer practical.
分享
阅读原文