MEKAWIRE
중급arXiv

가상 시점을 상상한 로봇, 가려진 물체 집기 실험 90회 중 85회 성공

Jiawei Geng, Zhenyang Liu, Yuzheng Wu, Yanwei FuFudan University, Shanghai Innovation Institute
30초 요약

더 좋은 카메라 각도에서 보일 장면을 먼저 상상한 로봇 정책이 가려지거나 어지러운 실제 집기 실험 90회 중 85회(94.4%)를 성공해, 피지컬 인텔리전스의 pi0(66회)와 Fast-WAM(72회)을 앞섰다. 푸단대와 상하이혁신연구원(SII)의 사전 공개 논문으로, 월드 모델이 카메라를 직접 움직이는 일을 대신할 수 있음을 보였지만 시뮬레이션에서는 격차가 작고 긴 작업에서는 우위가 없었다.

연구 질문

로봇 조작 정책이 카메라를 실제로 움직이거나 현재 화면에만 의존하지 않고, 계획한 가상 카메라 경로의 시점을 상상해 물체 가림과 시점 변화를 극복할 수 있는가.

문제

정돈된 벤치마크에서 잘하는 정책도 목표물이나 접촉 부위가 가려지면 실패한다. 행동 모델을 키워도 입력에 없는 시각 정보는 되살릴 수 없고, 카메라나 로봇을 움직여 다시 보는 방식은 시간과 추가 동작이 든다.

기존 접근

물리적 능동 인식은 카메라나 로봇을 움직여 시점을 얻는다. 가상 방식은 복원한 3차원 형상에서 추가 시점 한두 개를 렌더링하지만 처음 스캔에 없던 면은 보여줄 수 없고, 예측형 월드 모델은 대개 선택한 시점이 아니라 행동이 낳을 미래를 상상한다.

새 접근

AWAM(능동 월드 액션 모델)은 처음 RGB-D 입력에서 작업에 맞는 가상 카메라 궤적을 예측하고, 이를 기하 조건 월드 모델에 질의로 넣어 경로를 따라 보일 장면을 상상한 뒤, 에피소드마다 한 번 고정 메모리로 압축한다. 확산 기반 행동 헤드가 이 메모리와 실시간 카메라 영상을 교차 어텐션으로 합친다. 카메라는 실제로 움직이지 않는다.

결과

실제 로봇(RealSense D435 카메라, 가림과 어수선한 환경 과제 3종, 각 30회): 85/90, 94.4%로 pi0 66/90(73.3%), Motus 73/90(81.1%), Fast-WAM 72/90(80.0%)을 앞섰다. pi0보다 21.1%포인트, Fast-WAM보다 14.4%포인트 높다(MEKAWIRE 계산). 어수선한 탁자에서 병 집기는 30/30이었다. LIBERO-Occ(묶음당 500회)는 평균 75.9%로 VIM 74.0%, Fast-WAM 73.0%, pi0 70.0%보다 높았지만 Long 묶음에서는 Fast-WAM이 44.2% 대 41.0%로 앞섰다. LIBERO-Plus(1만30개 사례)는 62.4%로 pi0 53.6%, Fast-WAM 48.2%를 넘었고, 카메라 시점 변화에서는 53.4%로 pi0의 13.8%보다 3.87배 높았다. 다만 조명과 배경, 노이즈 변화에서는 pi0가, 언어 변화에서는 Motus가 앞섰다. RLBench-OG Occlusion1은 76.5%로 TVVE 75.0%보다 높았다. 세 구성 요소를 모두 빼고 실시간 화면만 쓰면 LIBERO-Occ가 64.4%로 11.5%포인트 떨어졌다.

한계

사전 공개 논문이다. 보정된 카메라와 깊이 정보가 필요해 영상만 쓰는 정책보다 조건이 까다롭고, 연구진은 메모리를 첫 장면에서 한 번만 만들어 물체가 크게 움직이면 낡은 정보가 될 수 있다고 밝혔으며 이를 긴 작업의 약세와 연결했다. 실로봇 시험은 과제 3종, 방법당 90회이고 본문에 로봇 팔 기종이 나오지 않는다. 시뮬레이션에서 가장 강한 비교 모델과의 차이는 1.5~2.9%포인트로 작고, 시점 변화 견고성은 보정값을 아는 조건에서 측정됐다.

업계 영향

물류 피킹과 매장 진열, 가정용 로봇은 가려진 물체를 늘 만난다. 한 번 상상해 둔 메모리로 둘러보는 추가 동작을 줄이면 작업 주기를 단축할 수 있다. 피지컬 인텔리전스 같은 VLA 개발사와 월드 모델 개발사, 그리고 물류 로봇을 실증하는 국내 업체가 눈여겨볼 기법이다. 1년 안에 연구용 시스템에 쓰일 수 있지만 실제 현장 적용은 깊이 센서와 보정 요구 조건에 달려 있다.

논문 전문

AWAM: Active Trajectory-Conditioned View Imagination for Robust Robot Manipulation

Jiawei Geng, Zhenyang Liu, Yuzheng Wu, Yanwei Fu

CC BY 4.0 라이선스로 공개된 논문입니다. 출처를 밝혀 전재하며, 원문은 arXiv:2610.23396(PDF)에서 볼 수 있습니다.

Abstract

Rapid progress in embodied intelligence has enabled increasingly capable robots to follow language instructions and manipulate diverse objects. However, strong results in structured environments do not automatically transfer to open-world conditions, where clutter, occlusion, and changing viewpoints can obscure the spatial relationships required for action. To address this perceptual gap, a virtual active-perception paradigm is introduced. AWAM, an Active World-Action Model actively plans where to observe and imagines what will be seen along the planned trajectory before acting. Task-related virtual camera motion is predicted and used as a structured query into a geometry-conditioned world model, producing a reusable spatial prior for manipulation, which is later compressed into a fixed memory and fused with live observations through cross-attention for diffusion-based action prediction. AWAM achieves competitive success rates with 75.9% on LIBERO-Occ, 62.4% on LIBERO-Plus, 76.5% on RLBench-OG Occlusion1, and 94.4% in real-robot trials. These results support improved robustness to the evaluated occlusions and distribution shifts, addressing part of the gap between structured manipulation and less controlled environments.

Figure 1: Overview of AWAM. From partially observed initial RGB-D inputs, AWAM imagines task-conditioned virtual views and compresses them into a predictive memory once per episode. The policy then reuses this memory together with live visual observations for closed-loop manipulation, without physically executing the predicted camera trajectory.

1 Introduction

Embodied intelligence has advanced rapidly through large-scale robot learning, vision–language pretraining, and generative action modeling. Increasingly capable policies can translate language goals and visual observations into manipulation behaviors across diverse tasks (Brohan and others, 2022; Brohan and others, 2023; Ghosh et al., 2024; Kim et al., 2024; Black et al., 2025; Chi et al., 2025).

However, much of this progress is evaluated in relatively structured settings in which camera configurations, scene layouts, and task-relevant objects remain sufficiently observable. Occlusion-focused and distribution-shift benchmarks expose a complementary limitation: once deployment becomes less structured, an unfavorable camera pose, clutter, or a shifted layout can hide the target or interaction region and remove the evidence required for control (Li et al., 2026; Fei et al., 2026; Bai et al., 2026; Liu et al., 2025c; Liu et al., 2025d). This creates a perceptual bottleneck rather than only an action-modeling bottleneck. More expressive action models alone cannot directly recover scene evidence that is absent or highly ambiguous in the input, so performance gains obtained under well-observed scenes may not transfer to open-world conditions with incomplete visibility.

Several lines of work have begun to address this problem. Physical active perception acquires additional evidence by moving the camera or robot, but introduces extra execution steps and can alter the robot or scene state. Virtual-view methods avoid this cost by selecting or synthesizing complementary observations. Candidate-view approaches choose from predefined cameras (Liu et al., 2026b; Xu et al., 2026); TVVE predicts task-conditioned virtual poses and re-renders reconstructed geometry (Bai et al., 2026); and world-action models such as Motus and Fast-WAM exploit predictive visual modeling to strengthen action representations (Bi et al., 2026; Yuan et al., 2026). These directions establish the value of additional visual context, yet the underlying limitation is not fully resolved: virtual perception is often restricted to isolated views, reconstruction-based rendering cannot reveal surfaces absent from the initial geometry, and predictive world models are commonly organized around action-induced futures rather than deliberate viewpoint queries.

To address these questions, task-conditioned virtual active perception is formulated as trajectory-conditioned view imagination. Instead of selecting a single view or a small set of independent poses, a virtual camera trajectory specifies how perception should evolve to expose task-relevant evidence. Such a trajectory can reorient toward a target, approach fine interaction geometry, or move around an occluder when no single viewpoint is sufficient. Importantly, the predicted trajectory is never physically executed; it serves only as a sequence of virtual queries, thus retrieving the available evidence and learned spatial priors into a representation that better supports control, separating perceptual exploration from robot motion.

Based on this formulation, AWAM is introduced as an Active World-Action Model that realizes the virtual-perception process before closed-loop control. A task-conditioned trajectory predictor first produces virtual camera poses for better observation from the initial RGB-D image and language instruction. The calibrated observations are then back-projected into coarse 3D geometry and rendered along the predicted path. A geometry-conditioned world model fills unsupported regions and produces multi-view latent features, which are compressed into a persistent world memory. During manipulation, live static- and wrist-camera observations retrieve relevant predictive context from this memory, and a diffusion policy generates short-horizon action chunks in closed loop. The resulting design uses world modeling as an internal perceptual mechanism for asking where to look and what would be seen, rather than as a prediction of the robot’s executed future.

Evaluation is conducted across three simulation benchmarks and real-world manipulation. Success reaches 75.9% on LIBERO-Occ, 62.4% on LIBERO-Plus, 76.5\pm 1.1% on RLBench-OG Occlusion1, and 94.4% in real-robot trials. On LIBERO-Occ, removing the entire predictive-memory branch reduces average success by 11.5 percentage points, and replacing learned task-conditioned virtual trajectory with sampled viewpoints reduces it by 5.7 points. Together, these results demonstrate the effectiveness of learned virtual camera motion and persistent predictive memory for manipulation under incomplete visibility. The contributions are threefold:

  • A trajectory-conditioned virtual active-perception paradigm is introduced for manipulation, replacing isolated or physically executed viewpoint acquisition with a continuous sequence of task-directed virtual queries.
  • AWAM is proposed to transform the predicted observation trajectory into geometry-consistent multi-view representations and compress them into persistent world memory as a reusable spatial prior for manipulation under partial observability.
  • Extensive evaluations across three challenging simulation benchmarks and real-world manipulation demonstrate strong performance under occlusion and distribution shifts and confirm the benefits of learned camera motion and predictive memory through ablations.

2 Related Work

Robot learning and action generation.

Large-scale robot policies map visual observations and language instructions directly to actions. Recent VLA models and diffusion-based policies improve generalization, action representation, and efficient decoding through large-scale pretraining and expressive generative objectives (Brohan and others, 2022; Brohan and others, 2023; Ghosh et al., 2024; Kim et al., 2024; Black et al., 2025; Chi et al., 2025; Pertsch et al., 2025; Kim et al., 2025; Liu et al., 2025a; Liu et al., 2025b). These methods primarily focus on how actions are represented and learned from available observations. In contrast, AWAM targets the perceptual evidence supplied to the policy, constructing additional task-conditioned spatial context when relevant scene information is poorly observed or occluded.

Multiview manipulation under occlusion.

Occlusion can be mitigated by acquiring or synthesizing complementary visual evidence through multiview perception, viewpoint selection, and imagined views (Liu et al., 2026b; Bai et al., 2026; Li et al., 2026; Xu et al., 2026). Existing approaches typically rely on additional sensing, reconstructed views, or a small fixed number of generated observations. AWAM instead predicts a task-conditioned virtual camera trajectory and consolidates the generated multi-view sequence into persistent world memory for closed-loop manipulation. This formulation turns occlusion handling into trajectory-conditioned predictive perception, where the model jointly reasons about where to observe, what would be observed, and how this information should support subsequent control.

World models for robotic manipulation.

Recent video diffusion and world models have shown strong capability in modeling long-horizon visual dynamics, while text-, action-, and camera-conditioned variants further enable controllable prediction of future or novel observations (Zhang et al., 2025a; Zhang et al., 2026; Liu et al., 2026a). Predictive world models provide representations of how visual observations may evolve and have increasingly been integrated with robot control. Video Prediction Policy (Hu et al., 2025) conditions control on features learned from predictive video modeling. Motus (Bi et al., 2026) unifies visual understanding, video generation, inverse dynamics, and action prediction within a latent world-action model. Fast-WAM (Yuan et al., 2026) separates the benefits of video co-training from explicit test-time future generation and shows that strong action prediction can be retained without iterative future-video rollout. Unlike approaches centered on predicting the future scene induced by robot actions, AWAM uses world modeling as a form of active perception. A task-conditioned camera trajectory specifies where the current scene should be queried, and a geometry-conditioned world model predicts visual representations along this trajectory. These representations are consolidated into a compact world memory and reused throughout manipulation rather than interpreted as the robot’s executed future.

3 Method

Figure 2: Overview of AWAM. A task-guided virtual camera trajectory guides a geometry-conditioned world model to construct predictive memory once per episode. During closed-loop control, the cached world memory is fused with live visual observations to predict action chunks.

3.1 Problem Formulation and Overview

We consider language-conditioned manipulation with a static camera s and a wrist-mounted camera w. At control step t, the policy observes RGB images I_{t}^{s} and I_{t}^{w}, proprioceptive state s_{t}, and language instruction \ell. At initialization it also receives metric depth and calibration:

\mathcal{O}_{0}=\left(\{I_{0}^{v},D_{0}^{v},K^{v},T_{0}^{v}\}_{v\in\{s,w\}},s_{0},\ell\right),
(1)

where K^{v} denotes camera intrinsics and T_{0}^{v}\in\mathrm{SE}(3) maps camera coordinates to the world frame. Depth and calibration are used to build the initial memory; the action policy consumes live RGB observations and proprioception. AWAM separates memory construction from repeated action prediction (Fig. 2): given the initial observation, AWAM constructs an active perception memory and subsequently conditions action prediction on both the memory and the current sensory evidence:

\displaystyle\widehat{\mathcal{T}} \\ \displaystyle=G_{\phi}(I_{0}^{s},D_{0}^{s},\ell;T_{0}^{w}), \\ \displaystyle M_{0} \\ \displaystyle=\mathcal{R}_{\eta}\left(\mathcal{F}_{\psi}(\mathcal{O}_{0},\widehat{\mathcal{T}})\right), \\ \displaystyle A_{t} \\ \displaystyle\sim\pi_{\theta}(\cdot\mid I_{t}^{s},I_{t}^{w},s_{t},\ell,M_{0}).
(2) (3) (4)

Here, G_{\phi} predicts task-relevant virtual viewpoints which may offer clearer observations, \mathcal{F}_{\psi} extracts geometry-conditioned predictive representations along the planned perception trajectory, and \mathcal{R}_{\eta} compresses them into a compact world memory. The action policy \pi_{\theta} then combines this memory with live observations during closed-loop control.

3.2 Task-Conditioned Virtual View Planning

An unfavorable initial wrist-camera pose can hide the object needed for manipulation, leaving the policy to act on incomplete visual evidence. Rather than relying on the initial viewpoint alone, a task-conditioned camera trajectory is predicted to specify where additional virtual observations should be queried for better scene perception.

Demonstrated wrist-camera paths provide proxy supervision for task-relevant virtual queries. Each trajectory is parameterized relative to its initial wrist-camera pose, making prediction invariant to the global scene origin. For a wrist pose T_{i}^{w}, the relative transform is defined as T_{i}=(T_{0}^{w})^{-1}T_{i}^{w}.

The relative poses are discretized into trajectory tokens encoding translation, rotation, and trajectory scale. Initial RGB-D observations and the language instruction are encoded into a shared conditioning representation C. A transformer then autoregressively predicts the trajectory sequence:

\mathcal{L}_{\mathrm{traj}}=-\mathbb{E}_{\mathcal{D}}\sum_{j=1}^{J}\log p_{\phi}(y_{j}\mid C,y_{<j}).
(5)

At inference, the decoded relative trajectory is transformed back into metric camera poses using the observed initial wrist pose T_{0}^{w}. A small set of representative viewpoints is retained along the trajectory rather than using every predicted pose. These virtual viewpoints define the camera queries passed to the world model and provide diverse observations of task-relevant regions while avoiding redundant predictions from nearby poses.

3.3 Geometry-Conditioned World Memory

A virtual view contains both geometrically supported pixels and regions that the initial cameras have not observed. We first unproject initial RGB-D observations into a colored point cloud,

\mathcal{P}_{0}=\bigcup_{v\in\{s,w\}}\operatorname{BackProject}(I_{0}^{v},D_{0}^{v},K^{v},T_{0}^{v}).
(6)

Rendering \mathcal{P}_{0} at the predicted camera poses preserves evidence from observed surfaces, but leaves holes where the reconstruction has no support. Such renderings become increasingly incomplete under viewpoint changes, since surfaces that are occluded or outside the initial fields of view have no geometric support, often resulting in large holes and unreliable visual features.

WorldStereo (Zhang et al., 2026) is employed as a geometry-conditioned world model and adapted to manipulation scenes. Intermediate geometry-conditioned features are extracted instead of passing generated RGB video to the action model, following the predictive-representation perspective of Hu et al. (2025). The generation target is an alternate view of the initial scene: objects and the robot are intended to remain fixed while the virtual camera changes pose.

The memory encoder compresses the predicted sequence. It aggregates spatial features within each predicted view and subsequently exchanges information across virtual viewpoints through inter-view attention. The resulting fixed-size token set summarizes the trajectory-level visual prediction while remaining efficient to retrieve during control, capturing what the robot is expected to observe from task-relevant viewpoints while remaining grounded in the geometry of the current scene.

3.4 Memory-Conditioned Closed Loop Control

At each control step, the current static- and wrist-camera observations are encoded into live visual tokens L_{t}. Predictive context is retrieved from the cached world memory M_{0} using residual cross-attention,

F_{t}=L_{t}+\operatorname{CA}(L_{t},M_{0}),
(7)

where L_{t} serves as the attention query and M_{0} provides the keys and values. Although the memory remains fixed throughout the episode, its retrieved content changes with the current observation. This allows the policy to access different parts of the predictive context as the manipulation state evolves, while the residual connection preserves a direct pathway from live perception to action prediction.

A conditional diffusion transformer is used to generate short-horizon action chunks. Given the fused visual representation F_{t}, proprioceptive state s_{t}, and language instruction \ell, the policy predicts

\widehat{A}_{t}=\mathcal{D}_{\theta}\left(z_{\sigma},\sigma;F_{t},s_{t},\ell\right),
(8)

where z_{\sigma} denotes the noisy action representation at diffusion level \sigma. The policy is trained on demonstrated action chunks using a standard diffusion denoising objective. During both action-policy training and evaluation, M_{0} is constructed from the trajectory predicted by G_{\phi} rather than from the demonstrated wrist-camera trajectory, avoiding a train–test mismatch in the memory input.

During closed-loop execution, the geometry-conditioned memory is constructed once from the initial observation and retained throughout the episode. At each replanning step, the policy receives fresh visual and proprioceptive observations, retrieves relevant predictive context, and generates a new action chunk from the updated state. The cached memory therefore acts as persistent predictive context about task-relevant virtual viewpoints rather than as a continually updated scene representation. Live observations remain the primary source of information about the evolving manipulation state, while the memory provides complementary spatial evidence that may be unavailable from the current viewpoint.

4 Experiments

4.1 Experimental Setup

Data and Environments.

Evaluation is conducted on three complementary simulation benchmarks. LIBERO-Occ (Li et al., 2026) tests robustness to scene-induced occlusion across the Spatial, Object, Goal, and Long suites; all 2,000 benchmark instances are evaluated, with 500 rollouts per suite. The Occlusion1 split of RLBench-OG (Bai et al., 2026), which contains ten occlusion-heavy manipulation tasks and uses a control interface different from LIBERO, provides a cross-benchmark test of the proposed world-memory mechanism. Both benchmarks directly test whether predicted viewpoints and imagined world memory predict task evidence omitted by the initial observation. Robustness beyond explicit occlusion is additionally evaluated on LIBERO-Plus (Fei et al., 2026), which introduces controlled shifts in camera viewpoint, robot initial state, language instruction, illumination, background texture, sensor noise, and object layout. Environment-defined task success is reported for LIBERO-Occ and LIBERO-Plus, while mean \pm standard deviation over three independent runs is reported for RLBench-OG. Real-world evaluation is conducted on an ARCHER-Y6 arm equipped with GP-100 grippers, scene-level and wrist-mounted Intel RealSense D435 cameras, and 100 original demonstrations.

Baselines.

Baselines are selected according to the focus and evaluation ecosystem of each benchmark. For LIBERO-based evaluations, three categories are included: (i) representative vision-language-action policies such as \pi_{0} (Black et al., 2025); (ii) world-model-augmented manipulation methods including Motus (Bi et al., 2026) and Fast-WAM (Yuan et al., 2026); and (iii) benchmark-specific occlusion methods represented by VIM (Li et al., 2026) on LIBERO-Occ. This set compares AWAM with both general-purpose policies and methods that explicitly introduce predictive or auxiliary visual representations. For RLBench-OG, the benchmark’s evaluation ecosystem is followed using Diffusion Policy (Chi et al., 2025), ARP (Zhang et al., 2025b), RVT2 (Goyal et al., 2024), and TVVE (Bai et al., 2026). When released results do not cover the exact evaluation setting, the corresponding methods are reproduced and tuned using their recommended training recipes and evaluated under the same protocol as AWAM.

4.2 Experimental Results

Results on LIBERO-OCC.

As shown in Table 1, AWAM achieves an average success rate of 75.9%. The improvement is consistent across three of the four suites: 84.6% on Spatial, 93.2% on Object, and 84.8% on Goal. They benefit from obtaining task-relevant evidence that is difficult to infer from the current view: the predicted viewpoint sequence queries the initially hidden region, while the world model converts those queries into a persistent representation that can be reused during manipulation. On Long, however, Fast-WAM obtains the highest success rate (44.2%), while AWAM reaches 41.0%. Long-horizon tasks remain challenging because the cached memory is constructed from the initial scene and is not regenerated after substantial object-state changes, which might be solved by regenerating memory each time when gripper state changes. Nevertheless, AWAM attains the highest overall average, indicating that active observations remain useful across a broad range of occluded manipulation problems.

Table 1: Success rate (%) on LIBERO-Occ. Each suite contains 500 evaluation rollouts.
  • \pi_{0}
    Spatial
    81.0
    Object
    84.0
    Goal
    76.6
    Long
    38.4
    Avg.
    70.0
  • Motus
    Spatial
    82.2
    Object
    78.0
    Goal
    66.0
    Long
    43.4
    Avg.
    67.4
  • Fast-WAM
    Spatial
    82.8
    Object
    86.4
    Goal
    78.8
    Long
    44.2
    Avg.
    73.0
  • VIM
    Spatial
    83.0
    Object
    92.0
    Goal
    81.0
    Long
    40.0
    Avg.
    74.0
  • AWAM
    Spatial
    84.6
    Object
    93.2
    Goal
    84.8
    Long
    41.0
    Avg.
    75.9
Results on RLBench-OG.

Table 2 presents the quantitative results on the Occlusion1 split of RLBench-OG. AWAM achieves an average success rate of 76.5%. The improvement is evident on tasks where the manipulation target or an interaction-relevant region is difficult to infer from the nominal observation. For example, AWAM reaches 94.7% success on water plants, improving over TVVE (88.0%), ARP (66.0%), and RVT2 (72.0%), and obtains 92.0% on scoop with spatula. It further achieves 100% success on five other tasks. Performance nevertheless remains low on block pyramid and solve puzzle. Both require precise multi-stage interaction, for which recovering visibility alone is insufficient. This suggests that predictive observation is most useful when partial observability is a primary bottleneck, but does not eliminate the intrinsic control difficulty of high-precision long-horizon tasks. Nevertheless, the overall result and the favorable average rank in Table 2 show that the benefit of active world memory extends across a diverse set of occlusion-sensitive tasks and transfers beyond the LIBERO environment.

Table 2: RLBench-OG Occlusion1 success (%, mean \pm std. over three runs). Avg. Rank is averaged across 10 tasks; lower is better.
  • Diffusion Policy
    Avg. \uparrow
    27.4{\pm}0.8
    Avg. Rank \downarrow
    4.70
    basketball in hoop
    0.0{\pm}0.0
    scoop with spatula
    0.0{\pm}0.0
    take plate off dish rack
    4.0{\pm}0.0
    water plants
    2.0{\pm}2.0
  • ARP
    Avg. \uparrow
    73.0{\pm}1.6
    Avg. Rank \downarrow
    2.80
    basketball in hoop
    \mathbf{100.0{\pm}0.0}
    scoop with spatula
    88.0{\pm}4.0
    take plate off dish rack
    \mathbf{100.0{\pm}0.0}
    water plants
    66.0{\pm}14.0
  • RVT2
    Avg. \uparrow
    72.8{\pm}2.0
    Avg. Rank \downarrow
    3.05
    basketball in hoop
    \mathbf{100.0{\pm}0.0}
    scoop with spatula
    82.7{\pm}7.5
    take plate off dish rack
    98.7{\pm}1.9
    water plants
    72.0{\pm}8.6
  • TVVE
    Avg. \uparrow
    75.0{\pm}1.2
    Avg. Rank \downarrow
    2.40
    basketball in hoop
    \mathbf{100.0{\pm}0.0}
    scoop with spatula
    88.0{\pm}8.0
    take plate off dish rack
    \mathbf{100.0{\pm}0.0}
    water plants
    88.0{\pm}4.0
  • AWAM
    Avg. \uparrow
    \mathbf{76.5{\pm}1.1}
    Avg. Rank \downarrow
    \mathbf{2.05}
    basketball in hoop
    \mathbf{100.0{\pm}0.0}
    scoop with spatula
    \mathbf{92.0{\pm}3.3}
    take plate off dish rack
    \mathbf{100.0{\pm}0.0}
    water plants
    \mathbf{94.7{\pm}1.9}
Results on LIBERO-PLUS.

As shown in Table 3, AWAM achieves an instance-weighted success rate of 62.4%, outperforming \pi_{0} (53.6%), Fast-WAM (48.2%), and Motus (44.0%). The gains are most pronounced under geometric perturbations that alter the observation configuration or scene structure, including camera viewpoint (53.4%), robot-state (48.9%), and object-layout shifts (73.0%). Robustness to viewpoint changes is under known calibration. This pattern is consistent with the design of AWAM: instead of relying solely on the current observation, the policy can leverage task-conditioned visual evidence predicted from multiple geometrically related viewpoints, reducing its sensitivity to changes in camera pose and scene geometry.

Table 3: Success rate (%) on all 10,030 LIBERO-Plus instances; Total is instance-weighted.
  • \pi_{0}
    Camera
    13.8
    Robot
    6.0
    Language
    58.8
    Light
    85.0
    Background
    81.4
    Noise
    79.0
    Layout
    68.9
    Total
    53.6
  • Motus
    Camera
    17.9
    Robot
    43.3
    Language
    74.1
    Light
    54.7
    Background
    44.9
    Noise
    26.2
    Layout
    52.0
    Total
    44.0
  • Fast-WAM
    Camera
    18.8
    Robot
    44.5
    Language
    68.9
    Light
    78.2
    Background
    45.7
    Noise
    29.8
    Layout
    60.7
    Total
    48.2
  • AWAM
    Camera
    53.4
    Robot
    48.9
    Language
    61.0
    Light
    84.3
    Background
    61.1
    Noise
    60.7
    Layout
    73.0
    Total
    62.4
Real-world evaluation.

AWAM is evaluated on three real-world manipulation tasks in which the initial observations provide incomplete task-relevant information: (i) retrieving a partially occluded apple from a box; (ii) picking a bottle from a cluttered tabletop and placing it into a nearby box; (iii) grasping a block initially occluded by a box and placing it into the box.

Figure 3: Real-world rollouts of AWAM under partial occlusion and clutters

As illustrated in Fig. 3, AWAM successfully handles partially occluded objects, cluttered scenes, and initially hidden targets in real-world grasping and placement. These rollouts highlight the practical value in addressing limited visibility and spatial ambiguity. Successful execution across these tasks indicates that the active-perception memory framework extends beyond simulation to real-world robotic settings.

The robot platform and camera configuration described in Sec. 4.1 are used, with 30 trials per task for each method. As shown in Table 4, AWAM achieves 85/90 successful trials (94.4%), compared with 66/90 (73.3%) for \pi_{0}, 73/90 (81.1%) for Motus, and 72/90 (80.0%) for Fast-WAM. The consistent gains under different occlusion configurations suggest that imagining task-conditioned virtual viewpoints and caching them as predictive memory can provide complementary visual evidence for closed-loop manipulation without requiring additional physical camera motion.

Table 4: Real-world manipulation, successes/trials (success rate, %).
  • \pi_{0}
    apple in box
    22/30(73.3%)
    bottle in cluster
    24/30(80.0%)
    block behind box
    20/30(66.7%)
    Overall
    66/90(73.3%)
  • Motus
    apple in box
    24/30(80.0%)
    bottle in cluster
    26/30(86.7%)
    block behind box
    23/30(76.7%)
    Overall
    73/90(81.1%)
  • Fast-WAM
    apple in box
    24/30(80.0%)
    bottle in cluster
    25/30(83.3%)
    block behind box
    23/30(76.7%)
    Overall
    72/90(80.0%)
  • AWAM
    apple in box
    28/30 (93.3%)
    bottle in cluster
    30/30 (100.0%)
    block behind box
    27/30 (90.0%)
    Overall
    85/90 (94.4%)

4.3 Ablation Study And Qualitative Analysis

Component Analysis.

To disentangle the contributions of each component, four controlled variants are evaluated while keeping the policy backbone, training data, and evaluation protocol unchanged. Results are summarized in Table 5.

Table 5: Component analysis on LIBERO-Occ. \Delta denotes the change from full AWAM in percentage points.
  • (a)Live observations only
    Learned trajectory
    \times
    Generation
    \times
    Memory
    \times
    Success (%) \uparrow
    64.4
    \Delta
    -11.5
  • (b)Trajectory-conditioned live policy
    Learned trajectory
    ✓
    Generation
    \times
    Memory
    \times
    Success (%) \uparrow
    67.3
    \Delta
    -8.6
  • (c)sampled spherical views
    Learned trajectory
    \times
    Generation
    ✓
    Memory
    ✓
    Success (%) \uparrow
    70.2
    \Delta
    -5.7
  • (d)Learned path, warp only
    Learned trajectory
    ✓
    Generation
    \times
    Memory
    ✓
    Success (%) \uparrow
    69.0
    \Delta
    -6.9
  • AWAM
    Learned trajectory
    ✓
    Generation
    ✓
    Memory
    ✓
    Success (%) \uparrow
    75.9
    \Delta
    –

Variant(a) shows that removing the entire predictive branch and relying only on live observations reduces the success rate from 75.9% to 64.4%, corresponding to the largest drop of 11.5 percentage points. This result confirms that the performance gain cannot be explained by the closed-loop policy alone, and that information constructed before control provides important complementary visual context when the current observations are incomplete or occluded. Variant(b), where the predicted camera trajectory is directly provided to the live policy without virtual-view generation or memory construction, performance improves to 67.3%. The 2.9-point gain over the live-only variant suggests that the task-conditioned trajectory itself carries a useful implicit manipulation prior. However, its substantial gap to the full model indicates that trajectory conditioning alone cannot substitute for explicit predictive visual context. Variant(c), which replaces the learned trajectory with spherical viewpoints uniformly sampled around the whole scene region achieves 70.2%. Despite preserving view diversity as well as the same generation and memory modules, the 5.7-point degradation shows that generic coverage of the scene is less effective than allocating virtual observations according to a learned, task-conditioned trajectory. Variant(d) retaining the learned trajectory while removing generative prediction and constructing memory only from warped geometric observations yields 69.0%. This variant preserves the views queried by the model, but can only reproject evidence already visible in the initial RGB-D observations. Its 6.9-point drop indicates that viewpoint selection alone is insufficient under severe occlusion, where previously unobserved surfaces cannot be recovered through geometric warping.

Taken together, these ablations reveal complementary roles of the proposed components. The predicted trajectory provides both task-dependent viewpoint selection and a weak planning prior, while generative prediction converts these queries into predictive visual context beyond the initially observed geometry. Compressing this predicted evidence into memory further allows the closed-loop policy to access such information throughout manipulation. The full combination therefore achieves the highest success rate of 75.9% on LIBERO-Occ.

Sensitivity to Virtual-View Budget.

The trajectory length K determines the amount of virtual visual context constructed before control. K varies in \{7,14,21,28\} while keeping all other settings fixed. Performance improves as the trajectory provides more virtual observations and saturates around K=21, with success rates of 72.4%, 74.8%, 75.9% and 76.0% on LIBERO-Occ respectively. Increasing the length further to 28 brings no meaningful improvement, suggesting limited sensitivity to the exact choice of K once sufficient task-relevant scene coverage is obtained.

Predicted Virtual Views.

For policy conditioning, intermediate predictive features are extracted after the first denoising update rather than completing the full RGB generation process. The RGB decodes in Fig. 4 are used only for qualitative diagnosis. Since AWAM uses a predicted camera trajectory to construct task-conditioned world memory, observations generated along this trajectory are visualized in Fig. 4.

Figure 4: Virtual-view predictions along the planned trajectory. Warped inputs (orange), predictions (blue), and ground truth (green) show progressively improved visibility; RGB decodes are diagnostic only.

The initial agent view is often occluded and does not clearly expose the task-relevant object or interaction region. As the predicted trajectory proceeds, the virtual camera progressively shifts toward more informative regions, revealing the target object, receptacle, or manipulation interface from more favorable perspectives. For instance, the virtual trajectory gradually exposes both the alphabet soup and the basket in the placement task, and moves toward the cabinet to reveal the middle drawer in the drawer-opening task.

The warped inputs provide a coarse geometric projection under the predicted viewpoints, while a single denoising step substantially restores appearance and scene structure necessary for manipulation. The average hole rate of LIBERO-Occ is 39.29%, with peak values reaching 74.47%. Despite these large holes, the denoised predictions still recover the key object layouts and interaction regions, indicating that the trajectory predictor successfully identifies where to observe and that the world model can translate these virtual viewpoints into task-relevant visual evidence for downstream action generation.

5 Conclusion

Rather than relying only on the currently visible camera frames, AWAM predicts a task-conditioned virtual camera trajectory and uses a geometry-conditioned world model to imagine observations along this trajectory. The resulting representations are compressed into a persistent world memory and jointly used with live observations during closed-loop action prediction. Experiments across occlusion-focused and distribution-shift benchmarks show that this additional predictive context improves manipulation robustness when task-relevant objects or interaction regions are poorly observed. These results suggest that world models can serve not only as predictors of future dynamics, but also as internal perceptual mechanisms for querying spatial information that is missing from the current observation. Future work could selectively refresh memory when large state changes are detected, extending virtual active perception to longer-horizon and more dynamic manipulation.

AI use statement

Generative AI tools were used to assist with programming, including code drafting, debugging, and implementation support, and to improve manuscript clarity, grammar, and presentation. They were not used to autonomously generate experimental results or determine the scientific conclusions of this work. All AI-assisted code was reviewed, tested, and validated by the authors before use in experiments, and AI-assisted manuscript edits were manually reviewed for technical accuracy, consistency with the underlying experiments, and appropriate attribution. The authors take full responsibility for the final content, including text, claims, code, and artifacts produced with the aid of generative AI.

Ethics statement

This work does not involve human subjects, sensitive personal data, or applications with foreseeable societal risks. Experiments are conducted on publicly available datasets and widely used pretrained models under their respective licenses.

Reproducibility statement

Multiple steps are taken to support reproducibility. The model architectures, training data and objectives, and evaluation protocols are specified in the paper, with additional implementation details provided in the appendix. The code will be released upon acceptance.

References

  1. Bai et al. (2026) Y. Bai, Z. Wang, Y. Liu, K. Luo, Y. Wen, M. Dai, W. Chen, Z. Chen, L. Liu, G. Li, and L. Lin Learning to see and act: task-aware virtual view exploration for robotic manipulation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 13386–13396.
  2. Bi et al. (2026) H. Bi, H. Tan, S. Xie, Z. Wang, S. Huang, H. Liu, R. Zhao, Y. Feng, C. Xiang, Y. Rong, H. Zhao, H. Liu, Z. Su, L. Ma, H. Su, and J. Zhu Motus: a unified latent action world model. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 35101–35113.
  3. Black et al. (2025) K. Black, N. Brown, D. Driess, A. Esmail, M. R. Equi, C. Finn, N. Fusai, L. Groom, K. Hausman, B. Ichter, S. Jakubczak, T. Jones, L. Ke, S. Levine, A. Li-Bell, M. Mothukuri, S. Nair, K. Pertsch, L. X. Shi, L. Smith, J. Tanner, Q. Vuong, A. Walling, H. Wang, and U. Zhilinsky \pi_{0}: a vision-language-action flow model for general robot control. In Proceedings of Robotics: Science and Systems,
  4. Brohan et al. (2022) A. Brohan et al. RT-1: robotics transformer for real-world control at scale. arXiv preprint arXiv:2212.06817.
  5. Brohan et al. (2023) A. Brohan et al. RT-2: vision-language-action models transfer web knowledge to robotic control. arXiv preprint arXiv:2307.15818.
  6. Chi et al. (2025) C. Chi, Z. Xu, S. Feng, E. Cousineau, Y. Du, B. Burchfiel, R. Tedrake, and S. Song Diffusion policy: visuomotor policy learning via action diffusion. The International Journal of Robotics Research.
  7. Fei et al. (2026) S. Fei, S. Wang, J. Shi, Z. Dai, J. Cai, P. Qian, L. Ji, X. He, S. Zhang, Z. Fei, J. Fu, J. Gong, and X. Qiu LIBERO-plus: a progressive robustness benchmark for visual-language-action models. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 38574–38583.
  8. Ghosh et al. (2024) D. Ghosh, H. R. Walke, K. Pertsch, K. Black, O. Mees, S. Dasari, J. Hejna, T. Kreiman, C. Xu, J. Luo, Y. L. Tan, L. Y. Chen, Q. Vuong, T. Xiao, P. R. Sanketi, D. Sadigh, C. Finn, and S. Levine Octo: an open-source generalist robot policy. In Proceedings of Robotics: Science and Systems, Delft, Netherlands. External Links: Document
  9. Goyal et al. (2024) A. Goyal, V. Blukis, J. Xu, Y. Guo, Y. Chao, and D. Fox RVT-2: learning precise manipulation from few demonstrations. arXiv preprint arXiv:2406.08545.
  10. Hu et al. (2025) Y. Hu, Y. Guo, P. Wang, X. Chen, Y. Wang, J. Zhang, K. Sreenath, C. Lu, and J. Chen Video prediction policy: a generalist robot policy with predictive visual representations. In Proceedings of the 42nd International Conference on Machine Learning, Proceedings of Machine Learning Research, Vol. 267. External Links: Link
  11. Kim et al. (2025) M. J. Kim, C. Finn, and P. Liang Fine-tuning vision-language-action models: optimizing speed and success. arXiv preprint arXiv:2502.19645.
  12. Kim et al. (2024) M. J. Kim, K. Pertsch, S. Karamcheti, T. Xiao, A. Balakrishna, S. Nair, R. Rafailov, E. Foster, G. Lam, P. Sanketi, et al. OpenVLA: an open-source vision-language-action model. arXiv preprint arXiv:2406.09246.
  13. Li et al. (2026) T. Li, J. Zhang, S. Wang, X. Huang, and Z. Wei LIBERO-occ: evaluating and improving vision-language-action models under scene-induced occlusion via viewpoint imagination. arXiv preprint arXiv:2606.10862.
  14. Liu et al. (2026a) Z. Liu, C. Cao, Y. Zhang, X. Zuo, X. Xue, Y. Fu, T. Wang, and C. Guo OmniCam: omni-camera trajectory generation via geometry-grounded pose token learning. arXiv preprint arXiv:2610.09513.
  15. Liu et al. (2026b) Z. Liu, Y. Gu, Y. Wang, X. Xue, and Y. Fu ActiveVLA: injecting active perception into vision-language-action models for precise 3d robotic manipulation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8141–8151.
  16. Liu et al. (2025a) Z. Liu, Y. Gu, S. Zheng, Y. Fu, X. Xue, and Y. Jiang TriVLA: a triple-system-based unified vision-language-action model with episodic world modeling for general robot control. arXiv preprint arXiv:2507.01424.
  17. Liu et al. (2025b) Z. Liu, Y. Wang, K. Wang, L. Liang, X. Xue, and Y. Fu Spatial-temporal aware visuomotor diffusion policy learning. In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), pp. 7122–7131.
  18. Liu et al. (2025c) Z. Liu, Y. Wang, S. Zheng, T. Pan, L. Liang, Y. Fu, and X. Xue ReasonGrounder: LVLM-guided hierarchical feature splatting for open-vocabulary 3D visual grounding and reasoning. arXiv preprint arXiv:2503.23297.
  19. Liu et al. (2025d) Z. Liu, S. Zheng, S. Chen, C. Zhao, L. Liang, X. Xue, and Y. Fu A neural representation framework with LLM-driven spatial reasoning for open-vocabulary 3D visual grounding. In Proceedings of the 33rd ACM International Conference on Multimedia, pp. 1042–1051.
  20. Pertsch et al. (2025) K. Pertsch, K. Stachowicz, B. Ichter, D. Driess, S. Nair, Q. Vuong, O. Mees, C. Finn, and S. Levine FAST: efficient action tokenization for vision-language-action models. arXiv preprint arXiv:2501.09747.
  21. Xu et al. (2026) T. Xu, R. Zhang, Z. Huang, J. Guan, J. Wang, Y. Ding, Y. Li, L. Chen, G. Chen, and J. Lu UniviewVLA: a unified multiview vision-language-action model with world modeling. arXiv preprint arXiv:2606.21501. External Links: Link
  22. Yuan et al. (2026) T. Yuan, Z. Dong, Y. Liu, and H. Zhao Fast-wam: do world action models need test-time future imagination?. arXiv preprint arXiv:2603.16666.
  23. Zhang et al. (2025a) M. Zhang, T. Wu, J. Tan, Z. Liu, G. Wetzstein, and D. Lin GenDoP: auto-regressive camera trajectory generation as a director of photography. In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), pp. 18229–18239.
  24. Zhang et al. (2025b) X. Zhang, Y. Liu, H. Chang, L. Schramm, and A. Boularias Autoregressive action sequence learning for robotic manipulation. IEEE Robotics and Automation Letters 10 (5), pp. 4898–4905. External Links: Document
  25. Zhang et al. (2026) Y. Zhang, C. Cao, T. Wang, X. Zuo, J. Wu, J. Zhu, and C. Guo WorldStereo: bridging camera-guided video generation and scene reconstruction via 3D geometric memories. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 40327–40339.

Appendix A Appendix

A.1 Additional Implementation Details

A.1.1 Camera Representation and Virtual Trajectory
Coordinate convention.

The static camera and wrist camera are denoted by s and w, respectively. A camera pose T_{i}^{w}=[R_{i}\mid p_{i}]\in\mathrm{SE}(3) maps camera coordinates to the world frame. All poses used by the geometry pipeline follow the OpenCV camera convention, with x pointing right, y down, and z forward. Metric depth and camera translation use the same physical unit.

As described in 3.2, the trajectory predictor operates on poses relative to the initial wrist-camera pose:

\overline{T}_{i}=(T_{0}^{w})^{-1}T_{i}^{w},\qquad T_{i}^{w}=T_{0}^{w}\overline{T}_{i}.
(9)

This representation removes dependence on the absolute world-frame origin. The world-model renderer consumes the corresponding world-to-camera extrinsics obtained by rigid inversion.

Trajectory representation.

Demonstrated wrist-camera trajectories are first temporally resampled. Each relative pose is represented by rotation, normalized translation, camera-related auxiliary coordinates, and a trajectory-scale variable. The normalized coordinates are discretized into trajectory tokens and modeled autoregressively.

For a normalized scalar x\in[-1,1], the reference implementation uses

q(x)=\left\lfloor\operatorname{clip}\left(\frac{B(x+1)}{2},0,B\right)\right\rfloor,\qquad B=256,
(10)

which gives integer values from 0 to 256. The decoded trajectory is converted back to metric scale and anchored to the observed initial wrist-camera pose using Eq. 9.

Trajectory predictor.

The trajectory generator is initialized from a text–RGB-D camera-trajectory model. The static-camera RGB image, metric depth map, and language instruction are independently encoded and projected into a shared conditioning sequence. A causal transformer autoregressively predicts the trajectory tokens using the teacher-forced objective of the main paper.

The visual and language backbone encoders remain frozen during trajectory adaptation, while the trajectory-specific projection layers and causal transformer are optimized for the manipulation-domain trajectories.

Selecting virtual viewpoints.

Rather than sending every predicted pose to the world model, K=21 representative poses are retained along the trajectory. To avoid allocating most of the view budget to nearly stationary trajectory segments, viewpoints are sampled according to accumulated camera motion.

For two consecutive poses, the translation and rotation increments are computed as

\displaystyle d_{i} \\ \displaystyle=\|p_{i}-p_{i-1}\|_{2}, \\ \displaystyle\vartheta_{i} \\ \displaystyle=\arccos\left[\operatorname{clip}\left(\frac{\operatorname{tr}(R_{i-1}^{\mathsf{T}}R_{i})-1}{2},-1,1\right)\right].
(11) (12)

Translation and rotation increments are independently normalized by their trajectory totals before being combined. The selected poses are approximately uniformly distributed over accumulated camera motion and include the trajectory endpoints.

Importantly, these poses are virtual queries. The robot never physically executes the predicted camera trajectory before manipulation.

A.1.2 Geometry-Conditioned Predictive Memory
Initial geometric evidence.

At episode initialization, the calibrated static- and wrist-camera RGB-D observations are back-projected into a joint colored point cloud. For camera v\in\{s,w\} and pixel \widetilde{u}=(u_{x},u_{y},1)^{\mathsf{T}} with valid depth D_{0}^{v}(u),

X_{0}^{v}(u)=R_{0}^{v}\left(D_{0}^{v}(u)(K^{v})^{-1}\widetilde{u}\right)+p_{0}^{v}.
(13)

The two point sets are merged into the initial geometry \mathcal{P}_{0}.

Rendering \mathcal{P}_{0} from a predicted viewpoint preserves appearance for surfaces supported by the initial observations, but leaves holes where the initial cameras contain no geometric evidence. This distinction motivates the predictive world model: point-cloud rendering can rearrange observed surfaces, but cannot reconstruct surfaces that were never observed.

Geometry-conditioned world model.

WorldStereo (Zhang et al., 2026) is used as the predictive visual backbone and adapted to manipulation scenes. The world model receives the coarse geometric rendering, missing-region mask, camera information, and reference observations, and predicts view-consistent representations along the virtual camera trajectory. The generation targets are static multi-view observations of the episode-initial scene. Supervision is obtained by holding the episode-initial simulator state fixed and rendering additional calibrated camera poses without advancing the environment dynamics. Thus, differences across target views are caused by camera motion rather than object or robot motion.

The generation target corresponds to alternative observations of the initial scene. The robot and scene objects are treated as stationary while the virtual camera changes pose. Thus, the predicted sequence should be interpreted as viewpoint-conditioned scene imagination rather than an action-induced future rollout.

The policy consumes intermediate world-model features instead of generated RGB frames. RGB decoding is used only for the qualitative visualizations in Fig. 4 of the main paper.

Predictive feature extraction.

For each of the K=21 virtual views, intermediate world-model features are spatially pooled to a 16\times 16 grid. In the reference implementation, this produces

Z\in\mathbb{R}^{21\times 256\times 5120}.
(14)

The features remain separated by viewpoint during extraction.

Memory compression.

The predictive features from the virtual trajectory contain dense spatial information across multiple viewpoints. Rather than exposing all features directly to the action policy, they are compressed with a lightweight query-based memory encoder.

For each virtual view, a small set of learned queries first aggregates its spatial features through cross-attention. Queries occupying the same slot then exchange information across the virtual-view sequence through inter-view attention.

The outputs from all virtual views are flattened into a compact memory M_{0}, which is cached for the duration of the episode. This representation preserves trajectory-level predictive context. Detailed architectural dimensions of the reference implementation are reported in Table 6.

A.1.3 Memory-Conditioned Closed-Loop Policy
Live visual encoding.

At each replanning step, the latest static- and wrist-camera RGB observations are encoded by a shared visual backbone. Camera-specific embeddings preserve the identity of the two visual streams, after which a language-conditioned resampler converts the spatial features into a fixed-size set of live visual tokens L_{t}.

Unlike the predictive memory, which is constructed once from the initial observation, L_{t} is recomputed from fresh observations during execution. It therefore provides the policy with up-to-date information about the evolving manipulation state.

Memory retrieval.

Predictive information is retrieved from the cached memory using residual cross-attention, where the live representation provides the attention queries and the cached memory provides keys and values.

Although M_{0} remains fixed during an episode, the attention weights depend on the current observation. Consequently, different predictive information can be retrieved as the manipulation state evolves. The residual pathway also preserves direct access to the current visual evidence.

A.2 Reference Architecture Settings

Table 6 summarizes the architectural dimensions of the reference LIBERO implementation. These values are reported for reproducibility and should be interpreted as implementation settings rather than separate methodological contributions.

Table 6: Reference architectural settings used in the LIBERO implementation of AWAM. These values specify the implementation used in our experiments and are reported for reproducibility.
  • Virtual-view queries
    Setting
    21 viewpoints
  • World feature resolution
    Setting
    16\times 16 tokens per view
  • World feature width
    Setting
    5120
  • Memory feature width
    Setting
    384
  • Memory encoder
    Setting
    2 layers, 8 attention heads
  • Compressed memory
    Setting
    294 tokens
  • Live visual representation
    Setting
    224 tokens, width 384
  • Memory retrieval
    Setting
    8-head cross-attention
  • Memory lifetime
    Setting
    One episode
  • LIBERO action horizon
    Setting
    8 actions

A.3 Training Procedure

The modules are optimized sequentially rather than through end-to-end backpropagation. The trajectory predictor is trained first, followed by world-model adaptation and then the action-policy stages.

Trajectory model.

The trajectory model is initialized from a pretrained text–RGB-D camera trajectory generator and adapted using demonstrated wrist-camera paths. The visual and language backbone encoders remain frozen, while the trajectory-specific projection and autoregressive prediction layers are optimized with the teacher-forced objective described in the main paper.

World model.

The geometry-conditioned world model (Zhang et al., 2026) is adapted to manipulation scenes while retaining its pretrained geometric and camera-conditioning capabilities. Training targets correspond to alternative observations of the same initial scene under different camera poses. Thus, the model learns to predict viewpoint-conditioned scene representations rather than robot-induced future states.

During action-policy training, world-model features are precomputed and cached. This avoids repeatedly running the large predictive model during policy optimization and provides a fixed representation interface between predictive perception and action learning.

Action policy.

Training is divided into an interface-alignment stage and an action-model adaptation stage. The first stage optimizes the live visual projection, resampler, memory encoder, and memory-fusion modules while keeping the pretrained action denoiser fixed. The second stage freezes these connection modules and adapts the diffusion action model.

A.4 Additional Evaluation Details

A.4.1 LIBERO-Occ

LIBERO-Occ evaluates robustness to scene-induced occlusion across the Spatial, Object, Goal, and Long suites. All official benchmark initial states are evaluated, corresponding to 500 rollouts for each suite and 2,000 rollouts in total. Success is determined by the environment-defined task success criterion.Example rollouts are shown in Fig. 5

Figure 5: Example rollouts on LIBERO-Occ.
A.4.2 RLBench-OG Occlusion1

Evaluation uses the ten tasks of the RLBench-OG Occlusion1 setting: basketball in hoop, scoop with spatula, take plate off colored dish rack, water plants, block pyramid, solve puzzle, take USB out of computer, close drawer, straighten rope, and toilet seat down. Example rollouts are shown in Fig. 6.

Figure 6: Example rollouts on RLBench-OG Occlusion1.

For RLBench-OG Occlusion1, models are trained and evaluated on the occluded task configurations following the benchmark protocol. RLBench-OG uses a control interface different from LIBERO. The same AWAM mechanism—trajectory-conditioned predictive perception, memory construction, and memory-conditioned closed-loop action generation—is retained with the benchmark-specific controller.

A.4.3 Real-World Task Details

Three manipulation tasks are evaluated on an ARCHER-Y6 arm with a GP-100 gripper and scene-level and wrist-mounted Intel RealSense D435 cameras. Each task is evaluated over 30 trials. AWAM constructs its predictive world memory once from the initial observations and retrieves this context alongside live RGB observations and proprioception during closed-loop control. The predicted camera trajectory specifies virtual viewpoints; it is not executed as a separate physical inspection motion. The third-person views in Figure 7 illustrate early and late execution stages. Descriptions of initial target occlusion below refer to the robot-camera observations.

Retrieving an apple from a box.

A red apple is placed inside an open box, whose walls partially obscure the apple in the initial robot observations. The challenge is to infer the grasp region and its relation to the surrounding walls from incomplete visual evidence. Task-conditioned virtual view planning is designed to provide predictive context for the partially observed box interior, while geometry-conditioned world memory retains these spatial features for action generation. Live observations support gripper positioning and extraction as the scene changes. The selected late frame shows the apple held above the rim, making the retrieval outcome visible from the external viewpoint.

Picking a bottle from clutter and placing it in a box.

A plastic bottle stands on a tabletop among a bowl, an apple, and smaller objects, with an open receiving box nearby. The robot should grasp the bottle, lift it from the clutter, transfer it to the box, and release it inside. This task combines target selection in a visually cluttered scene with coordination between a grasp location and a separate placement region. Also, half of the bottle lies outside the initial robot view, thus complicating the interpretation of the bottle’s local geometry and the gripper approach. AWAM’s task-conditioned virtual-view predictions are intended to complement the initial observations with spatial context, which the world memory retains throughout the transfer. Memory retrieval complements current observations, while closed-loop action prediction follows the changing bottle and gripper states. The late frame shows the bottle standing inside the receiving box as the gripper withdraws.

Grasping an occluded block and placing it in a box.

A green block is placed on the tabletop beside a box, which initially occludes the target in the robot-camera observations. The robot approaches the block, closes the gripper around it, lifts it above the box opening, and deposits it inside. The box therefore serves both as an occluder and as the destination, requiring the policy to relate the initially hidden grasp region to the visible container geometry. The cached memory remains available as live observations reveal the target and guide the subsequent grasp and transfer. The selected late frame captures the block held above the opening before release; completion additionally requires depositing the block inside the box.

Figure 7: Third-person views (for record only) of real-robot inference on three real-world tasks, with one early and one late frame per task.

A.5 Limitations

The current system relies on calibrated cameras and depth information to construct geometry-aware virtual observations, which introduces sensing assumptions beyond purely image-based policies. Besides, the current system constructs M_{0} once from the initial scene and keeps it fixed throughout manipulation. When objects are substantially displaced or articulated, parts of the cached memory may become stale. Fresh live observations provide the primary control feedback, but the memory itself is not explicitly updated. Dynamically scheduled memory reconstruction is a promising direction for long-horizon tasks.