MemLearner:学习为视频世界模型查询上下文记忆
摘要
MemLearner 提出了一种基于学习的自适应上下文查询方法,使用查询令牌来改善视频世界模型中的场景一致性和记忆,特别适用于具有遮挡和动态物体的长序列。
查看缓存全文
缓存时间: 2026/07/01 03:40
论文页面 - MemLearner:学习查询上下文记忆以用于视频世界模型
Source: https://huggingface.co/papers/2606.31734
摘要
MemLearner 通过使用基于学习的自适应上下文查询和查询令牌,增强包含遮挡和动态物体的长视频序列中的场景一致性和记忆,从而改进视频世界模型。
视频世界模型(https://huggingface.co/papers?q=Video%20World%20Models)是交互式视频生成模型(https://huggingface.co/papers?q=video%20generation%20model),其根据用户动作和历史视频帧预测未来世界状态。视频世界模型(https://huggingface.co/papers?q=video%20world%20models)的一个关键挑战是缺乏记忆(https://huggingface.co/papers?q=memory),导致长时间生成的场景不一致。先前的方法探索了基于规则的上下文帧检索(https://huggingface.co/papers?q=context%20frame%20retrieval)作为记忆(https://huggingface.co/papers?q=memory),但在场景遮挡和动态物体的场景中泛化能力不足。我们提出 MemLearner,一种基于学习的自适应上下文查询方法,使用查询令牌(https://huggingface.co/papers?q=query%20tokens)来桥接上下文和预测令牌。通过利用视频生成模型(https://huggingface.co/papers?q=video%20generation%20model)自身进行上下文查询,MemLearner 利用了预训练的视觉先验(https://huggingface.co/papers?q=visual%20priors),无需从头训练额外模块,并融入了高效的训练和推理策略。我们收集了一个包含场景遮挡和动态物体的长视频数据集,并配有相机位姿标注(https://huggingface.co/papers?q=camera%20pose%20annotations),提出了一种多数据集训练策略(https://huggingface.co/papers?q=multi-dataset%20training%20strategy),同时利用带标注的渲染视频和无标注的真实世界视频。大量实验表明,MemLearner 在场景一致性(https://huggingface.co/papers?q=scene%20consistency)和记忆(https://huggingface.co/papers?q=memory)方面显著优于先前的视频世界模型(https://huggingface.co/papers?q=video%20world%20models),尤其在具有挑战性的遮挡和动态场景下。
查看 arXiv 页面 (https://arxiv.org/abs/2606.31734)查看 PDF (https://arxiv.org/pdf/2606.31734)项目页面 (https://yujiwen.github.io/memlearner/)添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2606.31734)
在你的代理中获取此论文:
hf papers read 2606\.31734
没有最新 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用此论文的模型0
没有与此论文关联的模型
在模型 README.md 中引用 arxiv.org/abs/2606.31734 以从此页面链接。
引用此论文的数据集0
没有与此论文关联的数据集
在数据集 README.md 中引用 arxiv.org/abs/2606.31734 以从此页面链接。
引用此论文的 Spaces0
没有与此论文关联的 Space
在 Space README.md 中引用 arxiv.org/abs/2606.31734 以从此页面链接。
包含此论文的收藏0
没有包含此论文的收藏
将此论文添加到一个收藏 (https://huggingface.co/new-collection) 以从此页面链接。
相似文章
MemLife:针对长期第一人称视频记忆的构建与推理
MemLife 引入了一种面向长期第一人称视频的多模态记忆系统,它构建以实体为锚点的文本片段(text episodes),并通过基于时间索引的 agentic reader 进行检索。在长期时序基准测试中,其性能较免训练基线提升了 4.6%–12.0%。此外,名为 MemOpt 的强化学习框架进一步优化了记忆写入器的忠实性与可检索性,带来稳定 2.7%–5.0% 的性能提升。
MemTrain:自监督上下文记忆训练
MemTrain 提出了一种自监督训练框架,通过在维基百科语料上使用掩码重建和中间记忆召回代理任务,增强 LLM 智能体的上下文记忆能力,在下游记忆密集型 QA 基准上取得了高达 17.67 个百分点的提升。
MoME:上下文感知稀疏查找的混合记忆嵌入
介绍MoME,一种面向大语言模型的上下文感知记忆机制,它通过使用混合槽位来处理词元多义性,在预训练实验中优于基线方法。
WorldCrafter: 一致的视频世界模型与隐式3D感知记忆
WorldCrafter是一个视频世界模型,它学习一个可相机查询的隐式3D感知记忆,用于从单张图像或文本提示进行一致且相机可控的流式场景探索。
Memorizon:让世界模型在上下文窗口之外完成训练
Memorizon 通过相机共视关系让每个被评分的片段检索一个共享的、有界的 top-K 隐变量库,从而在流式世界模型中将长跨度监督与注意力解耦,以极低的额外开销实现超出上下文窗口的一致性重访。与滑动窗口基线相比,检索将重访一致性提升最高达 30%,且结果表明模型确实利用了检索到的隐变量。