AtlasVA:面向无教师 VLM 智能体的自进化视觉技能记忆 AtlasVA: Self-Evolving Visual Skill Memory for Teacher-Free VLM Agents
用自进化的视觉热图替代文本记忆与教师 LLM,为 3B VLM 智能体注入密集空间奖励。
前置知识
视觉语言模型(VLM)智能体
VLM 智能体指以预训练视觉语言模型(如 Qwen2.5-VL)作为策略核心,能够读取屏幕截图、自然语言指令并输出离散动作的交互式决策系统。其输入是 RGB 观测 $o_t$ + 文本指令,输出是动作 $a_t$,策略写作 $\pi_\theta(a_t \mid o_{\le t})$。
AtlasVA 直接建立在 VLM 智能体之上,并假设视觉编码器对 RGB 输入已有强先验,因此可以把空间热图作为独立的图像 token 注入。
基于势能的奖励塑形(PBRS)
PBRS 在原始稀疏奖励上叠加一个势能差 $\Phi(s_{t+1}) - \Phi(s_t)$,理论上不改变最优策略,却能为 RL 提供每一步梯度。AtlasVA 将势能函数 $\Phi$ 替换为自进化的 affinity 热图,从而把空间结构转化为稠密奖励。
PBRS 是 AtlasVA 密集奖励塑形一节的理论基础,没有这个视角就难以理解亲和势差为何天然产生逐格正向梯度。
指数滑动平均(EMA)自举
EMA 是一种无需外部监督的离线统计融合方式:新统计量按 $M \leftarrow \alpha M + (1-\alpha) M_{\text{batch}}$ 与历史量融合。AtlasVA 用它把每批失败终点聚合成 danger 图、把成功轨迹的访问频率聚合成 affinity 图。
EMA 是“无教师进化”的核心机制,决定了热图从随机初始化逐步稳定到几何先验的速率。
潜在式 POMDP 与稀疏奖励
具身决策常被建模为部分可观测马尔可夫决策过程,环境只在任务成功时给出 $R \in \{0,1\}$ 的稀疏反馈。对于 SOKOBAN、导航等长视野任务,纯稀疏奖励下的信用分配非常困难。
本文核心挑战——空间感知不足和信用分配困难——都源于稀疏奖励设定,奖励塑形正是为了缓解这一瓶颈。
DINOv2 视觉检索
DINOv2 是自监督视觉 Transformer,能输出具有强空间对应关系的 patch 级特征。AtlasVA 用冻结的 DINOv2 编码器计算当前观测与历史关键帧的余弦相似度,从而检索 top-k 视觉范例。
视觉范例库(Layer 2)正是依赖 DINOv2 完成相似帧检索,这是它在 prompt 中保留“具体布局参考”的关键。
研究动机
现有 VLM 智能体的记忆与反馈机制几乎全部围绕文本展开,导致三个连锁问题。模态失配方面,2D 拓扑结构(死胡同、角落陷阱、子目标区域)被压缩成一维语言规则,几何细节严重损失,零样本 Qwen2.5-VL-3B 在 SOKOBAN 上仅有 0.14 的成功率;教师依赖方面,Reflexion、ExpeL、SkillRL、XSkill 等系统都依赖 GPT-4 类强模型来总结失败、合并技能或重写记忆,部署成本高昂且违背自主进化前提;反馈稀疏方面,文本奖励(如“你不该把箱子推到角落”)抽象且无坐标粒度,叠加在原本只有 $R \in \{0,1\}$ 的稀疏环境奖励之上,无法缓解长视野任务的信用分配困难。这三个问题在空间密集型任务(SOKOBAN、3D 导航、机器人抓取)上同时爆发,单纯扩大模型规模(从 3B 升到 72B)也只把 SOKOBAN 从 0.14 提到 0.18,说明文本先验本身是天花板。
本文的目标是本文希望让一个仅 3B 参数的紧凑 VLM 智能体,在不依赖任何外部教师 LLM 的前提下,于 SOKOBAN、FROZENLAKE、AI2-THOR 3D 导航、ManiSkill PrimitiveSkill 五个 3D 抓取任务上达到乃至超过 GPT-5(0.69 平均成功率)的水平;具体而言是设计一套原生视觉记忆 + 自进化空间热图 + 密集势能塑形的闭环方案,把平均成功率推到 0.93,并把 PrimitiveSkill 五任务全部做到 1.00。
与已有工作不同的是,已有方法或聚焦记忆但用文本(XSkill、SkillRL),或聚焦奖励塑形但需要手工设计势能(经典 PBRS),或在像素层做探索但难扩展(count-based)。AtlasVA 的独特切入点是同时从三个层面打破文本中心范式:用 RGB 热图替代文本规则、用 EMA 自进化替代教师 LLM、用自学习势能替代手工势能,并把三者闭环为 perception–optimization loop。这一点此前在文献中没有完整组合过。
核心方法
直觉上,AtlasVA 把 VLM 智能体改造成一个“带外挂空间雷达”的玩家:雷达屏上红色区域是历史撞过的死角,绿色区域是历史上通向目标的子路径,玩家在做决策时把这张雷达当作额外画面看进去,与此同时雷达本身又会被玩家的新轨迹不断重绘。具体技术路线由四部分构成:(1) 三层视觉技能记忆 VSM——空间热图(Layer 1)+ 视觉范例(Layer 2)+ 文本规则(Layer 3)共同拼成 prompt;(2) 无教师视觉地图自进化——静态启发式 $M_{\text{heuristic}}$(角落/BFS 距离)与轨迹统计 $M_{\text{stat}}$ 通过 EMA 融合;(3) 地图驱动的密集奖励塑形——把 affinity 地图当势能函数、danger 地图当安全约束;(4) 闭环优化——更好的策略产生更好的轨迹,更好的轨迹反过来细化地图。整个框架中 VLM 推理时只用 RGB 观测加渲染好的记忆图像,不接触任何低层状态。
与已有方法最本质的区别在于“模态对齐”这一点:XSkill/SkillRL 把经验存成 Markdown/JSON 再喂回 VLM,等于把 2D 几何硬编码到 1D 文本中再让模型自己“翻译”回去,必然有损;而 AtlasVA 把记忆保留在 VLM 原生的视觉模态里——把空间统计渲染成 RGB 热图直接作为 $\langle\text{image}\rangle$ token 注入,让预训练好的视觉编码器零样本完成空间模式匹配。配合无教师进化(EMA + BFS 启发式)和势能塑形(affinity 当 $\Phi$,danger 当安全罚),形成“可进化的视觉势能场”这一全新范式。读者若只能记住一个点,就是:AtlasVA 把 RL 的稀疏奖励问题,从“如何写 prompt”转移到“如何画热图”。
方法步骤详情
完整流程按时间顺序分为两阶段:在线塑形阶段(每个 PPO epoch 内)与离线进化阶段(epoch 结束后)。在线阶段:对 batch 内每个转移 $(p_t, a_t, p_{t+1})$,先从 GridState 读取精确坐标 $p_t, p_{t+1}$,计算 affinity 势差 $r_{\text{affinity}} = M^{\text{final}}_{\text{affinity}}(p_{t+1}) - M^{\text{final}}_{\text{affinity}}(p_t)$,再计算 danger 惩罚 $r_{\text{danger}} = -\lambda_{\text{danger}} \cdot M^{\text{final}}_{\text{danger}}(p_{t+1})$,最终奖励 $\tilde{r}_t = r^{\text{env}}_t + \lambda_{\text{affinity}} r_{\text{affinity}} + r_{\text{danger}}$,把 $\tilde{r}_t$ 替换 batch 内稀疏奖励供 PPO 更新策略 $\pi_\theta$。离线阶段:把所有轨迹按成功/失败分组,对失败轨迹聚合终点坐标形成 $M^{\text{batch}}_{\text{danger}}(p) = \frac{1}{|T_{\text{fail}}|} \sum_{\tau \in T_{\text{fail}}} \mathbb{I}(p_T = p)$,对成功轨迹聚合每格访问频率形成 $M^{\text{batch}}_{\text{affinity}}$,再用 $M^{\text{stat}} \leftarrow \alpha M^{\text{stat}} + (1-\alpha) M^{\text{batch}}$ 更新历史统计;视觉范例池 $E_{\text{vis}}$ 用 DINOv2 余弦相似度检索关键帧,容量上限为 6(3 正 3 负),FIFO 淘汰;最终呈现给 VLM 的热图 $M^{\text{final}} = (1-\beta_k) M_{\text{heuristic}} + \beta_k M_{\text{stat}}$,其中调度系数 $\beta_k$ 从 0 渐进到 1,前期靠静态几何冷启动,后期切到经验驱动。
技术新颖性
技术上三点新颖性值得拆开看。第一是“无教师”的语义:传统记忆框架必须调用 GPT-4 生成总结,AtlasVA 的 Stext 直接从环境说明书抽取、热图直接从轨迹统计聚合、DINOv2 完成范例检索,全程不调用任何外部 LLM API。第二是“渲染为 token”而非“叠加到画面”:作者发现把热图 alpha-blend 到原图会干扰 VLM 对前景小物体的识别,于是改用独立 $\langle\text{image}\rangle$ token 让视觉编码器通过 cross-attention 自学对齐,这在工程上是反直觉但有效的选择。第三是“危险 ≠ 势能”的设计选择:danger 惩罚故意违反 PBRS 框架,因为它要主动改变最优策略以避开死角,而 affinity 严格满足势能差公式以保留理论最优性,这种“半势能”设计在文献中较少见。
实验结果
主结果(Table 2)显示 AtlasVA 在四大类基准上平均成功率 0.93,超越 GPT-5(0.69)、o3(0.71)、Claude Sonnet 4.5(0.62)、Gemini 2.5 Pro(0.51),并把最强开源基线 VAGEN 从 0.78 拉至接近天花板。逐任务看:在 SOKOBAN 上 AtlasVA 取得 0.79,比零样本 Qwen2.5-VL-3B(0.14)提升约 5.6 倍、超过 GPT-5(0.70)和 Qwen2.5-VL-72B(0.18)一个数量级以上;在 FROZENLAKE 上达到 0.83,超过 GPT-5(0.77)和 o3(0.78);3D Navigation Base 与 Common 子任务分别取得 0.85 和 0.87,比 VAGEN 的 0.78/0.80 各高约 7 个百分点;在 PrimitiveSkill 五个子任务(Place/Stack/Drawer/Align/Swap)全部取得 1.00 完美成功率,而 Qwen2.5-VL-72B 在 Swap 上仅 0.33、GPT-5 仅 0.55。学习效率(图 4)方面,文本基线(仅 Layer 3)在 SOKOBAN 上卡在 0.25 左右,AtlasVA 在 140 个训练步内爬升至约 0.80;PrimitiveSkill 上 AtlasVA 快速收敛到 1.00,文本基线停在 0.60 附近。消融(图 5)依次验证了四个核心组件:去掉整个 VSM 后 SOKOBAN/FROZENLAKE 严重退化、去掉热图或范例任一层都掉点、关掉自进化(退化为静态规则)造成全面回退、关掉密集奖励则陷入局部最优——这四个 ablation 联合支撑了 RQ2-RQ4 的回答。地图进化可视化(图 6)显示 Step 0 无信息、Step 200 起 Danger 图能准确高亮角落陷阱、Affinity 图描绘出可达子目标路径,纯靠环境交互无任何文本干预。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| SOKOBAN | success rate | 0.79 | Qwen2.5-VL-3B 0.14 / GPT-5 0.70 / VAGEN 0.61 | +0.65 vs 同基座 / +0.09 vs GPT-5 / +0.18 vs VAGEN |
| FROZENLAKE | success rate | 0.83 | Qwen2.5-VL-3B 0.14 / GPT-5 0.77 / VAGEN 0.71 | +0.69 vs 同基座 / +0.06 vs GPT-5 / +0.12 vs VAGEN |
| 3D Navigation (Base) | success rate | 0.85 | VAGEN 0.78 / GPT-5 0.75 | +0.07 vs VAGEN / +0.10 vs GPT-5 |
| 3D Navigation (Common) | success rate | 0.87 | VAGEN 0.80 / GPT-5 0.78 | +0.07 vs VAGEN / +0.09 vs GPT-5 |
| PrimitiveSkill Place | success rate | 1.00 | VAGEN 0.88 / GPT-5 1.00 | +0.12 vs VAGEN / 持平 GPT-5 |
| PrimitiveSkill Stack | success rate | 1.00 | VAGEN 0.88 / Qwen2.5-VL-72B 0.50 | +0.12 vs VAGEN / +0.50 vs 72B |
| PrimitiveSkill Drawer | success rate | 1.00 | VAGEN 0.88 / GPT-5 0.63 | +0.12 vs VAGEN / +0.37 vs GPT-5 |
| PrimitiveSkill Align | success rate | 1.00 | VAGEN 0.88 / Qwen2.5-VL-72B 1.00 | +0.12 vs VAGEN / 持平 72B |
| PrimitiveSkill Swap (新任务) | success rate | 1.00 | VAGEN 0.50 / GPT-5 0.55 / Qwen2.5-VL-72B 0.33 | +0.50 vs VAGEN / +0.45 vs GPT-5 / +0.67 vs 72B |
局限与改进
作者在结论部分坦承一个明确短板:把桌面 3D 抓取压成 2.5D 视觉先验在高度遮挡或自我中心视角下会失效,因为 X-Z 平面投影无法表达 Z 轴连续信息。读者还可以观察到的隐含限制包括:(1) GridState 依赖仿真器内部状态 API(SOKOBAN 的 room_state、ManiSkill 的 _handle_info 等),这些特权状态在真实机器人上并不存在,迁移到 sim-to-real 时需要重新设计感知模块;(2) Prompt 长度从 5000 到 8000 tokens 不等,热图 + 范例 + 文本规则叠加会让推理延迟变高,文中没有给出具体延迟数字;(3) 调度系数 $\beta_k$ 从 0 渐进到 1 的曲线、$\lambda_{\text{danger}}/\lambda_{\text{affinity}}$ 在 SOKOBAN(0.05)到 PrimitiveSkill(0.3)之间差异巨大,但超参数搜索过程没有披露;(4) 所有结果均在 4 个已知任务上得到,方法对全新任务类型(如开放词汇导航、动态障碍物)的零样本迁移能力尚未验证。
独立分析的弱点
独立分析可以指出四个可改进点。第一,GridState 特权状态依赖是方法最大的“阿喀琉斯之踵”——SOKOBAN、FROZENLAKE、Navigation、PrimitiveSkill 都被作者承认需要调用仿真器内部 API,当面对纯视觉输入的真实机器人时这套管线需要全部重写;改进方向是引入无监督视觉模块(如 Depth Anything V2 或 NeRF)估计占据栅格,把特权信息变成可学习预测。第二,视觉范例池容量被硬编码为 6(3 正 3 负),但缺乏自适应容量机制,长视野任务后期历史经验被强制淘汰可能丢失稀有但关键的失败模式;可以考虑引入信息论指标(如惊奇度)动态决定保留数量。第三,调度系数 $\beta_k$ 的退火曲线在文中没有明确给出,可能依赖任务手工调参;改进方向是用课程学习自动调节,或者直接以 BFS 距离场作为亲和势的下界从而摆脱 $\beta_k$。第四,方法对静态网格世界非常有效,但论文没有展示动态障碍或部分可观测场景下 danger 地图是否会误标——失败终态聚合策略在“对手智能体”存在的环境中可能把对方策略当作危险,造成过度保守。
未来方向
作者在结论与附录 D 中已经为未来工作埋下三条线索:(1) 将 2.5D 投影扩展到全 3D 占据场或神经辐射场,处理自我中心视角与高度遮挡;(2) 把视觉记忆范式推广到动态多智能体场景。基于成果还可以延伸的方向包括:用 AtlasVA 的热图作为可解释性工具直接可视化策略失败原因;将 danger/affinity 双通道拓展为多通道(如 affordance、traversability、goal-proximity)支持复合任务;以及把 EMA 进化替换为对比学习或世界模型预测,从而在样本更稀缺的环境(如真实机器人)下保持热图质量。
复现评估
复现难度中等偏上。作者给出的配置相当详尽:基础模型 Qwen2.5-VL-3B-Instruct、PPO actor 学习率 $1\times10^{-6}$、critic $1\times10^{-5}$、batch 128、mini-batch 32、8 块 RTX 6000 Ada、EMA 衰减 $\alpha=0.85$、调度 $\beta_k \in [0,1]$、热图渲染单元 40 像素、DINOv2 余弦相似度检索,全部列在 Table 5。代码主页 https://wangpan-ustc.github.io/AtlasvaWeb/ 提供了入口。开源情况方面,作者承诺开源但论文未注明是否已发布完整训练脚本,附录列出了 SOKOBAN、FROZENLAKE、AI2-THOR、ManiSkill 各自的特权 API(Table 4),意味着复现者需要把四套仿真器各自接通;最大隐性门槛是 PrimitiveSkill 的 Swap 任务是作者新引入的,需要自行实现或确认 VAGEN 仓库是否合并;总体算力需求约 8 张 RTX 6000 Ada 训练一天量级,单卡 RTX 4090 也可跑小规模实验但需要相应缩减 batch。
论文图表
上半图展示文本中心范式:文本技能库存储 “avoid corners/keep center” 等规则,再由教师 LLM 通过文本 prompt 反馈;下半图展示 AtlasVA:轨迹统计与提示共同驱动 VSM 自进化,热图作为视觉提示与密集奖励进入优化循环。
这是论文的“问题-方案”总览图,读者没有它就抓不住“文本 vs 视觉”这条核心对立。
System 段依次是规则说明、Spatial Skill Maps(热图)、Visual Exemplars(正/负范例)、Learned Principles(文本规则);User 段贴上当前观测图像并请求决策。
展示三层 VSM 如何在 prompt 中被锚定,是复现的工程蓝图。
四行表格列出 SOKOBAN/FROZENLAKE/Navigation/PrimitiveSkill 的 Success、Failure、Format Penalty、$\lambda_{\text{danger}}$、$\lambda_{\text{affinity}}$:前两者 $\lambda=0.05$,Navigation $\lambda=0.1$,PrimitiveSkill $\lambda=0.3$。
展示奖励强度需要随任务长视野程度放大,是复现必查的超参表。
PPO & GAE(actor/critic 学习率、batch、mini-batch、KL 系数、GPU 数)、Rollout Engine(vLLM、max tokens、prompt/response 长度)、Layer 1($\alpha=0.85$、像素 40、字段 Danger/Affinity)、Layer 2(3 正 3 负、上限 6、注入 4、DINOv2)、Layer 3(top-k、剪枝频率)全列出。
复现的“圣经表”,任何想做对照实验的人都必须照抄。