MemFold软记忆:固定256向量实现长上下文翻倍与4倍提速

MemFold用256个潜向量压缩任意长对话,在保持或提升准确率的同时实现约4倍提速。
大语言模型的长上下文记忆历来面临"堆token耗显存"与"RAG增延迟"的两难困境。MemFold提出将对话历史压缩为固定256个潜在向量的软记忆表示,使模型输入占用与历史长度完全解耦。论文实验显示,这一方案在PersonaMem-128K、PrefEval、LongMemEval等基准上达到或超越最强基线准确率,端到端推理成本与全上下文方案相当甚至更低,长历史场景下相对朴素拼接可实现约4倍提速。此外,无需目标域训练即可跨域迁移的特性,暗示软记忆有望成为通用长上下文接口。主要局限在于256向量预算固定,以及置信度门控蒸馏引入的额外训练成本。
长上下文的两难困境
大语言模型要记住更长的用户历史,长期只有两条路可走:要么把越来越长的token流直接喂给模型,这会迅速吃光GPU显存;要么在prompt里穿插检索调用(RAG),这又会增加延迟并常常拖累后续推理质量。
这两种做法的共同问题在于,它们都没有把"记忆"当作一个可学习的接口来对待,本质上只是随历史长度线性扩张的工程补丁。随着对话累积,成本和显存开销都会同步膨胀,这让生产环境中的长上下文助手面临难以回避的扩展瓶颈。
MemFold这篇论文提出的思路,正是要把记忆从"堆token"的粗放模式,转变为一个紧凑、可训练的中间表示。

软记忆:把长对话压成一把潜向量
MemFold的核心做法是把任意长度的对话压缩成少量潜在向量(latent vectors),阅读模型把这些向量当作普通token来消费,因此模型从不需要处理完整的原始转录文本。
关键特性在于这套软记忆是真正"尺寸无关"的——按论文表述,其大小不依赖于历史长度,阅读器可以直接读取它。系统固定使用 K = 256 个连续向量,无论原始历史是32K还是128K token,模型的输入占用都不会增加。
这是与前述两条老路最根本的区别:输入footprint被锁定在一个常数上,而不是随历史线性增长。换句话说,用户历史翻倍,模型的实际处理负担几乎不变。
潜在向量(latent vectors)是神经网络在高维输入空间中学到的低维压缩表示。在MemFold的框架里,一个专门的"写入模型"(writer)负责将原始对话历史编码为这256个向量,这一过程类似于自编码器的编码阶段,但写入模型的参数是端到端可训练的,而非固定的哈希或统计压缩。与传统RAG中离散的文本片段检索不同,这些向量是连续的、可微的,因此梯度可以穿透记忆接口反向传播,使写入模型能够学会"对下游任务更有用的压缩方式",而不仅仅是最大化字面重建精度。这也是论文将其称为"可学习接口"的根本原因——记忆本身参与了训练目标的优化,而不是事后插入的工程模块。
精度不打折,成本还更低
压缩表示最容易让人担心的,就是信息损失导致任务性能下降。但论文给出的数据恰恰相反。
在准确率方面,MemFold"在每一个基准和每一个骨干模型上都取得了最高准确率",仅在PrefEval的一个单元格上与最强基线打平。这说明压缩后的表示在上下文增长时并没有牺牲任务表现。
推理成本同样令人印象深刻。作者报告称,"端到端平均成本接近全上下文推理,并且在三个骨干模型上的总体成本更低",意味着额外的记忆模块几乎不增加运行时开销,反而整体上带来更快的预测。原标题所称的"4倍提速",正是这种固定预算设计在长历史场景下相对朴素拼接方案的加速优势体现。
跨域迁移能力初现
MemFold的主要训练场景是persona驱动(用户画像个性化)的任务,但作者还演示了它在无目标域训练的情况下迁移到PrefEval和LongMemEval的能力。
这一点值得关注,因为它暗示该方法的适用面可能超出个性化对话本身,延伸到开放式生成、代码补全等更广泛的领域。如果这种迁移性在更多场景中成立,软记忆就不只是一个针对特定基准的trick,而是一种通用的长上下文接口。
论文使用的几个压力测试基准——PersonaMem-128K、PrefEval、LongMemEval——如今都有了一个更强的参照线:MemFold在PersonaMem-128K上准确率更高、总体成本更低,在PrefEval上与最强基线持平。
遗留问题与现实意义
MemFold并非没有待解之处。
首先,软记忆预算被固定在256个向量,这意味着若要扩展到更大规模,可能需要重新设计架构,而非简单外推。其次,论文采用的on-policy优化和置信度门控蒸馏(confidence-gated distillation)虽然有效,却引入了一个带冻结教师模型的额外训练阶段——对缺乏大规模算力的团队而言,这可能是不小的负担。
尽管存在这些限制,其工程含义依然清晰:如果这些结果能在更多领域复现,那么朴素拼接用户历史的标准做法就应该被放弃。生产环境中的助手可以采用MemFold这座256向量的"桥梁",在不购买更大GPU的前提下,把可用上下文翻倍。
对于正在为长上下文成本发愁的团队来说,这提供了一个值得认真评估的新基线。
置信度门控蒸馏(confidence-gated distillation)是MemFold训练流程中的关键机制。知识蒸馏通常指用一个大型"教师"模型的软标签(概率分布)来监督小型"学生"模型,使学生能够学到教师隐含的知识。MemFold在此基础上加入了"门控"条件:只有当教师模型对某个样本的预测置信度超过阈值时,该样本的蒸馏信号才会被纳入训练。这样做的好处是避免教师在模糊或噪声样本上传递错误的监督信号,提升训练稳定性。然而代价是需要在推理阶段持续运行一个冻结的教师模型来生成置信度分数,这在训练资源上形成了额外门槛,对中小团队并不友好。on-policy优化则指写入模型在与阅读模型实际交互产生的轨迹上进行更新,而非依赖预先收集的静态数据集,有助于减少训练与推理之间的分布偏移。
参考资料
相关推荐

3分钟看懂DeepSeek:从量化私募到开源AI领跑者的崛起之路
3分钟看懂DeepSeek发展史:从幻方量化的算力积累,到V2的MLA推理降本、V3的557.6万美元训练账单、R1开源引爆全球,再到V4与70亿美元A轮融资,梳理深度求索的关键节点与技术路线。

本地文学创作模型PK:Qwen3.8 27B微调版胜出,DeepSeek v4.1表现平淡
B站UP主横向测评七款本地文学创作模型:海明威、HumanLike Chat基于Qwen3.8 27B微调表现最佳,DeepSeek v4.1叙事平淡,东方朔角色混乱垫底。附三道测试题评分与使用建议。

梁文锋的三次豪赌:DeepSeek为何将82%毛利开源?
DeepSeek为何把82%毛利的推理框架DeepInfer开源,又启动千亿融资?从技术路线、国产芯片到组织纯粹性,解析梁文锋的三次豪赌与开源战略背后的商业与理想逻辑。