浙大港大开源LayerRecall:165万参数路由模块修复长视频遗忘

青稞AI · wechat · 2026-09-11

自回归视频模型生成一段段画面、用KV缓存保留上下文时,早期身份与场景线索常被挤出缓存,导致角色离场再返回时换脸换装。浙江大学杨易团队与香港大学提出并开源 LayerRecall:只训练约1.65M参数的记忆路由模块(约为5B骨干的0.033%),生成骨干保持冻结,同时回答「该想起什么」与「该在哪些层使用」两个问题。

核心做法

结果:100个三镜头评测prompt上,MemoBench Overall 0.548、MovieBench 0.578 均领先最佳基线;VBench-Long 0.978与骨干持平。路由器零再训练迁移到LongLive(0.4985→0.5430)与Self-Forcing(0.3270→0.4965)。额外路由开销仅约3.5秒。

训练推理链路、CHPM训练代码与评测prompt bank均已开源,但用户仍需自备Wan2.2-TI2V-5B与LongLive-2.0基础权重,默认配置需两张GPU。

原文链接 →

「多模态」频道最新

更多「多模态」频道 AI 资讯 →