RoboArchitect: Automating the Design of Robotic Agents with Persistent Memory
Organizations: Stanford University
Abstract
Recent work shows that a frozen robot policy performs far better inside an agent harness, where control code decides when to query the policy, consult vision tools, or command the arm directly. Yet today these harnesses are built by hand and redesigned for each new robot or benchmark. We show that harness design can be automated, offering a strong alternative to finetuning on a novel setup. In RoboArchitect, a coding model observes a robot's rollouts, diagnoses failures, and iteratively revises the harness around the frozen policy. The harness evolves on a set of episodes for its optimization and is evaluated on held-out scenes and tasks. Without updating any model weights, RoboArchitect raises average success from 9.7% to 43.3% on unseen scenes (4.5x) and from 12.7% to 40.2% on unseen tasks (3.2x), across multiple benchmarks and policies. The gains are largest where the benchmark is far from the policy's training data: wrapping a general-purpose policy, the evolved harness yields a specialist that often outperforms even the finetuned version. Intuitive control strategies are automatically designed, such as lifting a jammed arm clear or reaching in horizontally when an obstacle blocks the way from above. The harness also evolves its own persistent memory, carrying learned experience, such as effective grasp depths, into later decisions. These results suggest that substantial headroom in robotic manipulation lies in the agent harness, and its design can be effectively automated and optimized without weight updates.
Figures & tables
| Unseen Scenes (%) | Unseen Tasks (%) | |||||||||
| Bare | Harness (ours) | Bare | Harness (ours) | |||||||
| Benchmark | Policy | Trained On | Policy | One-Shot | Iterative | Policy | One-Shot | Iterative | ||
| LIBERO-Pro-10 | MolmoAct-2 | LIBERO | 10.0 | 12.0 | 32.0 | +22.0 | 2.5 | 2.5 | 15.0 | +12.5 |
| OpenVLA-7B | LIBERO-10 | 0.0 | 2.0 | 26.0 | +26.0 | 0.0 | 0.0 | 22.5 | +22.5 | |
| LIBERO | 8.0 | 4.0 | 24.0 | +16.0 | 0.0 | 0.0 | 2.5 | +2.5 | ||
| RLinf- | LIBERO-130 | 2.0 | 2.0 | 40.0 | +38.0 | 0.0 | 0.0 | 10.0 | +10.0 | |
| Training Pool (%) procedurally generated | Held-Out (%) standard benchmark episodes | |||||||||
| Bare | Harness (ours) | Bare | Harness (ours) | |||||||
| Policy | Trained On | Benchmark | Policy | One-Shot | Iterative | Policy | One-Shot | Iterative | ||
| General X-VLA | general robot data | LIBERO-Pro | 0.0 | 3.5 | 29.4 | +29.4 | 0.1 | 3.5 | 24.1 | +24.0 |
| VLABench | 7.1 | 14.8 | 17.5 | +10.4 | 7.5 | 17.3 | 21.6 | +14.2 | ||
| SimplerEnv WidowX | 0.0 | 65.8 | 68.3 | +68.3 | 0.0 | 45.8 | 24.0 | +24.0 | ||
| SimplerEnv Google | 0.0 | 22.0 | 51.2 | +51.2 | 0.0 | 20.0 | 45.6 | +45.6 | ||
| Setting | Memory Entries | Reads per Episode | Actionable Reads |
|---|---|---|---|
| X-VLA on VLABench | finger depths that have been lifting objects | 3.4 | 98% |
| X-VLA on SimplerEnv WidowX | height at which to close the fingers, per measured object height | 3.5 | 96% |
| MolmoAct-2 on LIBERO-Pro-10 | average depth of successful grasps below an object’s top | 9.9 | 99% |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Benchmark | Trials | Optimization Tasks | Held-Out Tasks |
|---|---|---|---|
| LIBERO-Pro-10 | 10 | turn on the stove and put the moka pot on it ; put the black bowl in the bottom drawer of the cabinet and close it ; put both the alphabet soup and the tomato sauce in the basket ; put both the cream cheese box and the butter in the basket ; put the white mug on the left plate and put the yellow and white mug on the right plate | put the yellow and white mug in the microwave and close it ; put both moka pots on the stove ; put both the alphabet soup and the cream cheese box in the basket ; put the white mug on the plate and put the chocolate pudding to the right of the plate |
| RoboCerebra | 10 | cream cheese, popcorn and butter, each into a white storage box; cream cheese into a white storage box, then the frypan onto the rack; plate, popcorn and cookies rearranged on the dining set, then wine poured onto the plate; chocolate pudding from the cabinet top onto the plate, then wine poured onto the plate; chocolate pudding and butter into a black bowl, the stove turned on, then the frypan onto the stove | cookies into a basket, tomato sauce and bbq sauce poured onto the frypan, wine poured into a mug, then another mug into a wooden tray; cream cheese and popcorn into a white bowl, then the frypan onto the cabinet top; the short cabinet opened, butter into the cabinet, then tomato sauce poured onto the plate; the short cabinet opened, cream cheese into the cabinet, chocolate pudding into a black bowl, then wine poured into a mug; white bowl from the cabinet top, cookies and chocolate pudding into the bowl, the microwave opened, then the bowl into the microwave |
| VLABench | 10 | select the named fruit; select the named mahjong tile; select the named poker card | select the named drink out of a fridge; select the -th largest poker card; select the mahjong tile of the unique type |
| SimplerEnv WidowX | 24 | put the spoon on the towel ; stack the green block on the yellow block | put carrot on plate ; put eggplant into yellow basket |
| RoboCasa-365 | 50 | close the toaster oven door; open the cabinet door; open the drawer; pick an object from the counter and place it in the cabinet; press the start button on the microwave | close the fridge door; pick an object from the plate and place it in the pan on the stove; fully slide the top dishwasher rack out; press down the lever to turn on the electric kettle |
Agent Harness Optimization
Agent harness optimization improves the prompts, tools, memory, workflows, and control code that shape how AI agents act.
See all 197 papers