Can Astra Solve RoboMME without Breaking the Bank?

A Three-Tier System for Memory-Augmented Manipulation

Bingao Chen   Haoquan Fang   C. Karen Liu

Stanford University

Astra on RoboMME: results and memory-augmented manipulation

Abstract

Memory-augmented robotic manipulation uses past observations to guide present actions, requiring visual histories to be understood and translated into executable subtasks. We present a three-tier system: System 1, a low-level vision-language-action model (VLA), executes actions; System 2, GPT-6 Astra, performs high-level planning; and System 1.5, a small learned vision-language model, monitors subtask completion. Astra interprets the task and its visual context to generate a grounded subtask. The VLA executes this instruction over successive action chunks, while a small vision-language model checks whether the subtask has finished. Completion events trigger new planning calls, separating frequent progress checks from expensive high-level reasoning. This design brings Astra’s planning capabilities into a closed-loop controller while avoiding a new Astra request at every action chunk. On 800 official RoboMME test episodes across 16 tasks, the controller achieves 79.13% success with an average of 3.63 Astra calls per episode and an average wall-clock episode duration of 80.84 seconds. This nearly matches the 84.08% success rate of GroundSG+Oracle, an oracle-guided upper-bound reference for grounded-subgoal policies.

1. Introduction

Many robot tasks depend on information that is no longer visible. A demonstration may specify the order of several actions; a target may disappear under a container; a moving object may need to pass a location a certain number of times. RoboMME [1] brings these challenges together: the robot must understand what happened earlier to choose the right action now.

Hierarchical systems such as MEM [2] and τ₀-VLA [3] offer a natural starting point: a high-level VLM decides the next subtask, and a low-level VLA executes it. In our development experiments, small high-level VLMs struggled with complex RoboMME video inputs and produced incorrect subtasks. This motivated us to put Astra in charge of high-level understanding and planning.

But a robot does not need to reconsider the entire task every time it takes another action. Once a subtask has been chosen, execution may continue for many action chunks. Repeatedly asking Astra to process the visual history during this period adds inference work even when the next instruction should remain unchanged.

Our solution is System 1.5: a small visual completion monitor that connects planning and action. System 2 (Astra) decides what to do; System 1 (the low-level VLA) carries it out; System 1.5 decides when it is time to ask the planner again. This gives the small model a focused job while keeping complex video understanding with the stronger planner.

2. Methods

Three-tier system control loop: System 2, GPT-6 Astra, generates a grounded subtask for System 1, fine-tuned π0.5. After 16 execution steps, System 1.5, fine-tuned Qwen3-VL-4B, checks completion. An incomplete subtask returns control to the VLA; completion triggers the next Astra planning call.
Figure 1. Three-tier system. System 1 (fine-tuned π0.5) executes low-level actions, System 2 (GPT-6 Astra) plans grounded subtasks, and System 1.5 (fine-tuned Qwen3-VL-4B) monitors completion. The planner uses task instructions and visual history, while the monitor checks completion from recent observations. An incomplete subtask continues under the same instruction; completion triggers the next planning call.

The controller separates three decisions: what to do next, how to execute it, and when to move on. Astra plans a grounded subtask, the VLA carries it out, and a small visual monitor checks completion.

Let g denote the task instruction, o_t the current observation, and l_k the active grounded subtask. Here, t indexes environment steps and k indexes subtasks.

2.1 System 2: High-Level Planning with Astra

Astra chooses one grounded subtask at a time—an executable instruction with a target location where needed. We express the high-level policy as

\pi_{\mathrm{HL}}(l_k \mid o_t, m_t, g).

The memory m_t collects relevant demonstration and execution history, together with subtask progress. This context helps Astra locate objects that have become hidden or follow a demonstrated action order. Each planning call combines that history with the current scene to select the next subtask.

2.2 System 1: Low-Level Action Execution

A π0.5 VLA fine-tuned for grounded-subgoal execution carries out the active subtask through a low-level policy,

\pi_{\mathrm{LL}}(a_{t:t+H} \mid o_t, l_k, g).

Here, a_{t:t+H} is an action chunk and o_t includes camera observations and robot state. The same l_k can guide multiple chunks: the instruction stays fixed while fresh observations guide the next actions.

2.3 System 1.5: Visual Completion Monitoring

A Qwen3-VL-4B model fine-tuned for completion detection predicts whether the active subtask has finished,

p_{\mathrm{MON}}(c_t \mid v_t, l_k, g), \qquad c_t \in \{0,1\}.

The visual context v_t combines recent observations with a reference from the start of the subtask. The binary output indicates whether execution should continue or the planner should be consulted. This focused question lets the small model monitor execution while Astra handles task interpretation and planning.

2.4 Event-Triggered Coordination

After each action chunk, the monitor determines whether to continue or request a new plan. A false result keeps the VLA working on the same instruction; a true result normally records completion and triggers Astra to select the next subtask. This loop ties replanning to detected progress instead of requesting a new plan after every action chunk.

2.5 Inference Efficiency

Frequent progress checks run through the local monitor, reserving Astra calls for initial planning and subtask transitions.

We initially tried running Astra and the VLA asynchronously, allowing the VLA to continue executing while Astra planned the next subtask. In these preliminary tests, most episodes timed out: the VLA exhausted the 1,300-step budget before Astra returned the next subgoal. For the reported experiments, we therefore follow the synchronous execution used in RoboMME’s official two-system evaluation implementation, pausing the simulation during planner inference. These waits add wall-clock time but do not consume environment steps.

3. Results

3.1 Experimental Setup

We evaluated our pipeline on 800 official RoboMME test episodes across 16 tasks, with 50 episodes per task, following RoboMME’s official evaluation protocol.

3.2 Task Performance

Our controller achieves 79.13% success, approaching GroundSG+Oracle’s 84.08%. Baselines include the current-observation policy π0.5 [4], the visual-history method FrameSamp+Modul, SAM2Act+ [6] (visual memory bank), and three VLM–VLA dual-system methods: MemER [5] (keyframe memory), GroundSG+QwenVL (fine-tuned planner), and GroundSG+Gemini (prompted planner) [1]. GroundSG+Oracle supplies ground-truth subgoals to the low-level VLA, providing an empirical upper-bound reference for our approach. Baseline scores come from the official RoboMME leaderboard.

Overall success: 3-tier Astra (Ours) 79.13%, GroundSG+Oracle 84.08%, FrameSamp+Modul 44.51%, MemER 42.38%, GroundSG+QwenVL 32.70%, SAM2Act+ 21.37%, pi0.5 17.93%, GroundSG+Gemini 11.56%. A second panel compares Astra with Oracle across four task suites.
Figure 2. Overall and suite-level success. Teal: our controller; gold: GroundSG+Oracle with ground-truth subgoals.
Success rates (%) on the official test set
MethodCountingPermanenceReferenceImitationOverall
π0.5Current-observation policy28.7817.0017.168.7817.93
GroundSG+GeminiGrounded subgoals · prompted VLM12.7515.7511.256.5011.56
GroundSG+QwenVLGrounded subgoals · fine-tuned VLM38.0039.3431.5621.8932.70
SAM2Act+Visual memory bank35.3326.0016.837.3321.37
MemERVisual keyframes + symbolic subgoals48.8353.1738.0029.5042.38
FrameSamp+ModulPerceptual memory65.2225.1136.3351.3944.51
3-tier Astra (Ours)Astra planner + learned monitor69.5094.0092.0061.0079.13
GroundSG+OracleGround-truth subgoals · reference83.8693.3195.1763.9884.08

Our Astra-based controller outperforms all three dual-system baselines across all four task suites. Overall success reaches 79.13%, compared with 42.38% for MemER, 32.70% for GroundSG+QwenVL, and 11.56% for GroundSG+Gemini. The largest gains over MemER, the strongest of these dual-system baselines, appear in Permanence and Reference: our method achieves 94.00% and 92.00%, versus 53.17% and 38.00%. On Imitation, which requires interpreting demonstrated behavior, success rises from 29.50% to 61.00%.

These results support pairing Astra’s visual and temporal reasoning with completion-triggered planning for memory-dependent manipulation. Astra uses demonstrations and execution history to choose the next grounded subtask, while the monitor checks whether that subtask has finished. This division keeps Astra focused on decisions that require broader context and lets the VLA continue executing between planning events. The resulting controller combines strong task performance with infrequent Astra calls, as quantified below.

All 16 tasks: success, failure, and timeout counts +
Official test outcomes · 50 episodes per task
TaskSuccessFailTimeoutSuccess rate
BinFill416382.00%
StopCube941018.00%
PickXtimes481196.00%
SwingXtimes419082.00%
ButtonUnmask5000100.00%
VideoUnmask461392.00%
VideoUnmaskSwap5000100.00%
ButtonUnmaskSwap426284.00%
PickHighlight420884.00%
VideoRepick433486.00%
VideoPlaceButton5000100.00%
VideoPlaceOrder490198.00%
MoveCube404680.00%
InsertPeg953618.00%
PatternLock446088.00%
RouteStick2921058.00%
Overall6331036479.13%

3.3 Inference Usage

Our pipeline averages 3.63 Astra calls per episode. A two-system controller calling Astra before every VLA action chunk would require an estimated 24.62 calls per episode over the same recorded execution sequences. This corresponds to 85.3% fewer Astra calls: the small local monitor handles frequent completion checks, allowing the VLA to reuse a subtask across multiple action chunks.

3.4 Hard-Task Successes

Successful hard-difficulty test episodes, with one featured task per suite. Expand each suite for the remaining examples. The central robot view plays continuously; System 1.5’s completion checks (red: false; green: true) and System 2’s subgoals update alongside it.

Nine video-conditioned tasks begin with their archived input demonstration, followed by execution. Input frames are displayed at 30 fps; their original timing was not recorded. Execution plays at 30 environment steps/s; highlights mark logged updates and inference waits are omitted.

Counting

BinFill

Hard · test ep023 · Success · 663 steps

put two blue cubes and one green cube into the bin, then press the button to stop

More counting tasks (3)+

PickXtimes

Hard · test ep035 · Success · 638 steps

pick up the blue cube and place it on the target, repeating this action four times, then press the button to stop

SwingXtimes

Hard · test ep047 · Success · 462 steps

pick up the red cube, move it to the top of the right-side target, then move it to the top of the left-side target, repeating this back-and-forth motion three times, finally press the button to stop

StopCube

Hard · test ep011 · Success · 214 steps

press the button to stop the cube just as it reaches the target for the fourth time

Permanence

ButtonUnmaskSwap

Hard · test ep047 · Success · 531 steps

first press both buttons on the table, then pick up the container hiding the green cube, finally pick up another container hiding the red cube

More permanence tasks (3)+

ButtonUnmask

Hard · test ep007 · Success · 372 steps

first press the button, then pick up the container hiding the red cube, finally pick up another container hiding the green cube

VideoUnmask

Hard · test ep019 · Success · 246 steps

watch the video carefully, then pick up the container hiding the green cube, finally pick up another container hiding the blue cube

VideoUnmaskSwap

Hard · test ep035 · Success · 245 steps

watch the video carefully, then pick up the container hiding the blue cube, finally pick up another container hiding the green cube

Reference

PickHighlight

Hard · test ep039 · Success · 503 steps

first press the button, then pick up all cubes that have been highlighteted with white areas on the table

More reference tasks (3)+

VideoRepick

Hard · test ep015 · Success · 215 steps

watch the video carefully, then pick up the same block that was previously picked up again, finally put it down and press the button to stop

VideoPlaceButton

Hard · test ep023 · Success · 162 steps

watch the video carefully, then place the red cube on the target right after the button was pressed

VideoPlaceOrder

Hard · test ep007 · Success · 171 steps

watch the video carefully, then place the green cube on the third target it was previously placed on

Imitation

RouteStick

Hard · test ep027 · Success · 263 steps

watch the video carefully, then use the stick attached to the robot to navigate around the sticks on the table, following the same path

More imitation tasks (3)+

MoveCube

Hard · test ep039 · Success · 100 steps

watch the video carefully, then move the cube to the target in the same manner as before

InsertPeg

Hard · test ep003 · Success · 182 steps

watch the video carefully, then grasp the same end of the same peg you've picked before and insert it into the same side of the box

PatternLock

Hard · test ep007 · Success · 88 steps

watch the video carefully, then use the stick attached to the robot to retrace the same pattern

4. Conclusion

A more capable planner does not need to run at the same frequency as the action policy. Our three-tier system assigns low-level action execution to System 1, high-level reasoning to System 2, and frequent progress checks to System 1.5. The completion signal connects them into a closed loop.

The system can still make planning, monitoring, or execution errors. But the design offers a practical way to use a strong reasoning model in long-horizon manipulation: ask it for the next subtask when progress calls for a new decision.

References

  1. Yinpei Dai, Hongze Fu, Jayjun Lee, Yuejiang Liu, Haoran Zhang, Jianing Yang, Chelsea Finn, Nima Fazeli, and Joyce Chai. RoboMME: Benchmarking and Understanding Memory for Robotic Generalist Policies. In International Conference on Machine Learning (ICML), 2026.
  2. Marcel Torne, Karl Pertsch, Homer Walke, Kyle Vedder, Suraj Nair, et al. MEM: Multi-Scale Embodied Memory for Vision Language Action Models. arXiv preprint arXiv:2603.03596, 2026.
  3. Xiaowei Cai, Yunuo Cai, Bingao Chen, Jingxiao Chen, Zhi Chen, et al. τ₀-VLA: a Hierarchical Robot Foundation Model with World-Model-Guided Test-Time Computation. arXiv preprint arXiv:2608.16885, 2026.
  4. Kevin Black, Noah Brown, James Darpinian, Karan Dhabalia, Danny Driess, et al. π0.5: a Vision-Language-Action Model with Open-World Generalization. In Proceedings of The 9th Conference on Robot Learning (CoRL), PMLR 305:17–40, 2025.
  5. Ajay Sridhar, Jennifer Pan, Satvik Sharma, and Chelsea Finn. Scaling up Memory for Robotic Control via Experience Retrieval. In International Conference on Learning Representations (ICLR), 2026.
  6. Haoquan Fang, Markus Grotz, Wilbert Pumacay, Yi Ru Wang, Dieter Fox, Ranjay Krishna, and Jiafei Duan. SAM2Act: Integrating Visual Foundation Model with A Memory Architecture for Robotic Manipulation. In Proceedings of the 42nd International Conference on Machine Learning (ICML), PMLR 267:15925–15942, 2025.

Citation

If you find this work useful, please consider citing it:

@misc{chen2026astraonrobomme,
  title  = {Can {Astra} Solve {RoboMME} without Breaking the Bank?},
  author = {Chen, Bingao and Fang, Haoquan and Liu, C. Karen},
  year   = {2026},
  url    = {https://github.com/bingaochen/Astra-on-RoboMME}
}
Download BibTeX