WORLD-MODELSCURRENT2026-06-17

Mem-World: Memory-Augmented Action-Conditioned World Models for Persistent Robot Manipulation

Zirui Zheng, Jiaqian Yu, Xiongfeng Peng, jun shi, Mingyi Li, Chao Zhang, Weiming Li, Dong Wang, Huchuan Lu, Xu Jia

Mem-World solves the core problem of world models forgetting scene geometry during complex Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. by using memory-indexed surface geometry to retrieve relevant past observations. This lets you generate realistic long-horizon rollouts for Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. even when the Movement, Mechanics & Robot BodyEnd-effectorThe tool at the end of a robot arm, like a gripper, hand, or suction cup. occludes the scene, improving Simulation & Sim-to-RealSim-to-real (sim2real)Transferring a policy trained in simulation to a real robot. correlation by 14.5% and boosting Data, Distributions & Training IssuesTask successWhether the robot completed the task correctly. from 58% to 72% using Simulation & Sim-to-RealSynthetic dataArtificially generated training data, often from simulation..

THE PROBLEM

This paper focuses on world models. Mem-World solves the core problem of world models forgetting scene geometry during complex Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. by using memory-indexed surface geometry to retrieve relevant past observations. This lets you generate realistic long-horizon rollouts for Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. even when the Movement, Mechanics & Robot BodyEnd-effectorThe tool at the end of a robot arm, like a gripper, hand, or suction cup. occludes the scene, improving Simulation & Sim-to-RealSim-to-real (sim2real)Transferring a policy trained in simulation to a real robot. correlation by 14.5% and boosting Data, Distributions & Training IssuesTask successWhether the robot completed the task correctly. from 58% to 72% using Simulation & Sim-to-RealSynthetic dataArtificially generated training data, often from simulation.. Read the paper by tracking the Core ConceptsTaskThe job the robot is supposed to complete, such as pick-and-place, navigation, or drawer opening. definition, the Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. or data assumptions, and the evidence that supports the claimed improvement.

HOW IT WORKS

1

Task framing

The paper frames the work as world models. Start here because it defines what success means and which assumptions the rest of the method inherits.

2

Core method

Mem-World solves the core problem of world models forgetting scene geometry during complex Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. by using memory-indexed surface geometry to retrieve relevant past observations. This lets you generate realistic long-horizon rollouts for Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. even when the Movement, Mechanics & Robot BodyEnd-effectorThe tool at the end of a robot arm, like a gripper, hand, or suction cup. occludes the scene, improving Simulation & Sim-to-RealSim-to-real (sim2real)Transferring a policy trained in simulation to a real robot. correlation by 14.5% and boosting Data, Distributions & Training IssuesTask successWhether the robot completed the task correctly. from 58% to 72% using Simulation & Sim-to-RealSynthetic dataArtificially generated training data, often from simulation.. When reading the method section, identify the inputs, the learned or engineered representation, and the Core ConceptsActionA command the robot sends to its motors, controller, or low-level system. or prediction produced by the system.

3

Data and supervision

For robotics work, the data story is part of the method: check whether the system depends on Imitation & Reinforcement LearningTeleoperation (teleop)A human remotely controlling the robot, often to collect demonstrations., Simulation & Sim-to-RealSimulationA virtual environment where robots can be trained or tested., internet video, human labels, or Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. rollouts.

4

Evaluation evidence

The paper should be judged through its Simulation & Sim-to-RealEvaluationMeasuring how well a robot system performs. protocol: what data is used, what Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. or simulator is tested, and which Evaluation & ResearchBaselineA reference method used for comparison. comparisons support the claim. Look for the gap between the headline result and the Simulation & Sim-to-RealDeploymentPutting the trained system on a real robot. setting you would actually care about.

KEY RESULTS

Main contributionConceptual contribution

Mem-World solves the core problem of world models forgetting scene geometry during complex Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. by using memory-indexed surface geometry to retrieve relevant past observations. This lets you generate realistic long-horizon rollouts for Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. even when the Movement, Mechanics & Robot BodyEnd-effectorThe tool at the end of a robot arm, like a gripper, hand, or suction cup. occludes the scene, improving Simulation & Sim-to-RealSim-to-real (sim2real)Transferring a policy trained in simulation to a real robot. correlation by 14.5% and boosting Data, Distributions & Training IssuesTask successWhether the robot completed the task correctly. from 58% to 72% using Simulation & Sim-to-RealSynthetic dataArtificially generated training data, often from simulation..

WHY DEVELOPERS SHOULD CARE

Mem-World solves the core problem of world models forgetting scene geometry during complex Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. by using memory-indexed surface geometry to retrieve relevant past observations. This lets you generate realistic long-horizon rollouts for Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. even when the Movement, Mechanics & Robot BodyEnd-effectorThe tool at the end of a robot arm, like a gripper, hand, or suction cup. occludes the scene, improving Simulation & Sim-to-RealSim-to-real (sim2real)Transferring a policy trained in simulation to a real robot. correlation by 14.5% and boosting Data, Distributions & Training IssuesTask successWhether the robot completed the task correctly. from 58% to 72% using Simulation & Sim-to-RealSynthetic dataArtificially generated training data, often from simulation..

LIMITATIONS

The main limitation to check is whether the claimed behavior holds outside the paper's reported setup. That means testing across different Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. embodiments, scenes, objects, and data distributions.

WHAT COMES NEXT

The practical next step is independent reproduction with clear baselines, ablations, and stress tests. For a developer, the useful follow-up is to map the paper's world models assumptions onto a concrete Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. stack, then test the smallest version of the method that could run end to end.

RELATED PAPERS