AdaMem 论文:按相关性分配记忆 token,RAG 软压缩再提 3.2 点

_reachsumit · x · 2026-09-22

论文(arXiv:2609.22100)提出 AdaMem,解决 RAG 软压缩中的一个低效问题:现有方法给每条保留的检索段落分配相同数量的连续记忆嵌入,无论其与查询的相关度如何。

方法:相关性引导的软压缩框架——一个共享的查询条件压缩器单次前向同时产出段落记忆与相关性分数;确定性分配规则把更多记忆 token 给高分段落,低分段落甚至可以完全省略。

结果:在六个开放域 QA 基准上,AdaMem 在相同记忆预算下一致超越均匀分配基线 OSCAR 及其他软压缩方法:标准 16× 压缩下子串匹配最多提升 3.2 点(相对提升 5.5%),平均相对提升 3.4%;在激进的 64× 压缩下优势更大。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →