IMITATION-LEARNINGCURRENT2026-04-16

R3D: Revisiting 3D Policy Learning

Zhengdong Hong, Shenrui Wu, Haozhe Cui, Boyi Zhao, Ran Ji, Yiyang He, Hangxing Zhang, Zundong Ke, Jun Wang, Guofeng Zhang, Jiayuan Gu

This paper solves the Robot LearningTrainingThe process of fitting a model using data or experience. instability and Data, Distributions & Training IssuesOverfittingWhen a model performs well on training data but poorly on new data. problems that have blocked 3D Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. from scaling, enabling robots to learn Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. skills from 3D observations with better Modern Robot LearningGeneralizationThe robot’s ability to work in new situations it has not seen before. and Modern Robot LearningCross-embodiment transferTransferring knowledge across different robot bodies.. By identifying that missing 3D Data, Distributions & Training IssuesData augmentationArtificially varying training data to improve generalization. and Batch Data, Distributions & Training IssuesNormalizationRescaling inputs or features to stabilize learning. were the culprits, they built a transformer-based 3D encoder with diffusion decoder that now achieves Evaluation & ResearchState of the art (SOTA)The best published result on a benchmark at that time. on Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. benchmarks and can leverage large-scale pre-training.

ARCHITECTURE

THE PROBLEM

This paper focuses on Imitation & Reinforcement LearningImitation Learning (IL)Teaching a robot by showing it examples of how to do a task.. This paper solves the Robot LearningTrainingThe process of fitting a model using data or experience. instability and Data, Distributions & Training IssuesOverfittingWhen a model performs well on training data but poorly on new data. problems that have blocked 3D Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. from scaling, enabling robots to learn Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. skills from 3D observations with better Modern Robot LearningGeneralizationThe robot’s ability to work in new situations it has not seen before. and Modern Robot LearningCross-embodiment transferTransferring knowledge across different robot bodies.. By identifying that missing 3D Data, Distributions & Training IssuesData augmentationArtificially varying training data to improve generalization. and Batch Data, Distributions & Training IssuesNormalizationRescaling inputs or features to stabilize learning. were the culprits, they built a transformer-based 3D encoder with diffusion decoder that now achieves Evaluation & ResearchState of the art (SOTA)The best published result on a benchmark at that time. on Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. benchmarks and can leverage large-scale pre-training. 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 Imitation & Reinforcement LearningImitation Learning (IL)Teaching a robot by showing it examples of how to do a task.. Start here because it defines what success means and which assumptions the rest of the method inherits.

2

Core method

This paper solves the Robot LearningTrainingThe process of fitting a model using data or experience. instability and Data, Distributions & Training IssuesOverfittingWhen a model performs well on training data but poorly on new data. problems that have blocked 3D Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. from scaling, enabling robots to learn Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. skills from 3D observations with better Modern Robot LearningGeneralizationThe robot’s ability to work in new situations it has not seen before. and Modern Robot LearningCross-embodiment transferTransferring knowledge across different robot bodies.. By identifying that missing 3D Data, Distributions & Training IssuesData augmentationArtificially varying training data to improve generalization. and Batch Data, Distributions & Training IssuesNormalizationRescaling inputs or features to stabilize learning. were the culprits, they built a transformer-based 3D encoder with diffusion decoder that now achieves Evaluation & ResearchState of the art (SOTA)The best published result on a benchmark at that time. on Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. benchmarks and can leverage large-scale pre-training. 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

This paper solves the Robot LearningTrainingThe process of fitting a model using data or experience. instability and Data, Distributions & Training IssuesOverfittingWhen a model performs well on training data but poorly on new data. problems that have blocked 3D Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. from scaling, enabling robots to learn Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. skills from 3D observations with better Modern Robot LearningGeneralizationThe robot’s ability to work in new situations it has not seen before. and Modern Robot LearningCross-embodiment transferTransferring knowledge across different robot bodies.. By identifying that missing 3D Data, Distributions & Training IssuesData augmentationArtificially varying training data to improve generalization. and Batch Data, Distributions & Training IssuesNormalizationRescaling inputs or features to stabilize learning. were the culprits, they built a transformer-based 3D encoder with diffusion decoder that now achieves Evaluation & ResearchState of the art (SOTA)The best published result on a benchmark at that time. on Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. benchmarks and can leverage large-scale pre-training.

WHY DEVELOPERS SHOULD CARE

This paper solves the Robot LearningTrainingThe process of fitting a model using data or experience. instability and Data, Distributions & Training IssuesOverfittingWhen a model performs well on training data but poorly on new data. problems that have blocked 3D Imitation & Reinforcement LearningPolicy learningTraining a model that maps observations to actions. from scaling, enabling robots to learn Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. skills from 3D observations with better Modern Robot LearningGeneralizationThe robot’s ability to work in new situations it has not seen before. and Modern Robot LearningCross-embodiment transferTransferring knowledge across different robot bodies.. By identifying that missing 3D Data, Distributions & Training IssuesData augmentationArtificially varying training data to improve generalization. and Batch Data, Distributions & Training IssuesNormalizationRescaling inputs or features to stabilize learning. were the culprits, they built a transformer-based 3D encoder with diffusion decoder that now achieves Evaluation & ResearchState of the art (SOTA)The best published result on a benchmark at that time. on Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. benchmarks and can leverage large-scale pre-training.

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 Imitation & Reinforcement LearningImitation Learning (IL)Teaching a robot by showing it examples of how to do a task. 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