论文精选

论文对比文档存储方式对大模型性能的影响

Where Should a Document Live: Context, Representations, or Parameters?

精选理由

这篇论文用数据说话,帮你搞清楚给大模型加新知识时,是直接塞进去还是改模型参数更好,特别适合做知识问答或需要快速更新信息的应用。

研究显示,将新信息编码到模型参数(如KV缓存)或直接作为上下文输入,在知识密集型任务中表现不同。在理想情况下,Cartridges方法(KV缓存)在存储预算内表现最佳,比参数化方法高10分。但在多文档检索场景下,Cartridges是唯一能匹配上下文学习的方法,领先参数化方法29分。不过,Cartridges也面临灾难性遗忘问题。

原文 · arXiv cs.AI

Where Should a Document Live: Context, Representations, or Parameters?

To answer questions outside of their pre-training data, large language models (LLMs) need access to new information, which can be presented in the context window as documents, encoded into the model's parameters, or injected as latent representations. However, each of these methods comes with different efficiency, cost, and performance trade-offs, with no single winner. We present a controlled comparison of representation-based (KV-cache based) and parametric (fine-tuning-based) adaptation methods on five knowledge-intensive benchmarks. We show that in the oracle setting, Cartridges (KV) are the most accurate injection method at nearly every storage budget, outperforming parametric methods by 10 points. Compaction (KV) matches Cartridges only at low compression rates, lagging behind the parametric methods by 10 points at rates higher than $50\times$. In the more realistic multi-document retrieval scenario, Cartridges are the only method that matches in-context learning (ICL), leading the parametric methods by 29 points and Compaction by 15 points. Nonetheless, Cartridges are also the only method, besides full fine-tuning and large MLP adapters, that suffers from catastrophic forgetting, i.e., a 6% performance degradation on control benchmarks, with 13% in coding.