IMITATION-LEARNINGCURRENT2026-06-08

Difference-Aware Retrieval Policies for Imitation Learning

Quinn Pfeifer, Ethan Pronovost, Paarth Shah, Khimya Khetarpal, Siddhartha Srinivasa, Abhishek Gupta

DARP fixes Imitation & Reinforcement LearningBehavior Cloning (BC)A simple type of imitation learning where the robot directly copies expert actions.'s Data, Distributions & Training IssuesDistribution shiftWhen the deployment data differs from the training data. problem by retrieving similar expert demonstrations at test time instead of relying on a single learned Core ConceptsPolicyThe rule or model that maps observations or states to actions.. This gives 15-46% better performance on Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. and Control & PlanningControlThe method used to make the robot move the way you want. tasks without needing extra data, online Control & PlanningFeedbackInformation returned from sensors during action to help correct behavior., or retraining—just by comparing your current Core ConceptsStateThe robot’s current condition, such as joint positions, velocity, object positions, or internal variables. to the Robot LearningTrainingThe process of fitting a model using data or experience. set's k-nearest neighbors.

THE PROBLEM

This paper focuses on Imitation & Reinforcement LearningImitation Learning (IL)Teaching a robot by showing it examples of how to do a task.. Semi-parametric retrieval-based approach to Imitation & Reinforcement LearningImitation Learning (IL)Teaching a robot by showing it examples of how to do a task. that predicts actions by finding k-nearest neighbor demonstrations and using their local neighborhood structure rather than learning a global state-to-action Navigation & LocomotionMappingBuilding a representation of the environment.. Addresses compounding errors from Data, Distributions & Training IssuesDistribution shiftWhen the deployment data differs from the training data. in standard Imitation & Reinforcement LearningBehavior Cloning (BC)A simple type of imitation learning where the robot directly copies expert actions.. 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

DARP fixes Imitation & Reinforcement LearningBehavior Cloning (BC)A simple type of imitation learning where the robot directly copies expert actions.'s Data, Distributions & Training IssuesDistribution shiftWhen the deployment data differs from the training data. problem by retrieving similar expert demonstrations at test time instead of relying on a single learned Core ConceptsPolicyThe rule or model that maps observations or states to actions.. This gives 15-46% better performance on Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. and Control & PlanningControlThe method used to make the robot move the way you want. tasks without needing extra data, online Control & PlanningFeedbackInformation returned from sensors during action to help correct behavior., or retraining—just by comparing your current Core ConceptsStateThe robot’s current condition, such as joint positions, velocity, object positions, or internal variables. to the Robot LearningTrainingThe process of fitting a model using data or experience. set's k-nearest neighbors. 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.

FIGURES

KEY RESULTS

Main contributionConceptual contribution

DARP fixes Imitation & Reinforcement LearningBehavior Cloning (BC)A simple type of imitation learning where the robot directly copies expert actions.'s Data, Distributions & Training IssuesDistribution shiftWhen the deployment data differs from the training data. problem by retrieving similar expert demonstrations at test time instead of relying on a single learned Core ConceptsPolicyThe rule or model that maps observations or states to actions.. This gives 15-46% better performance on Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. and Control & PlanningControlThe method used to make the robot move the way you want. tasks without needing extra data, online Control & PlanningFeedbackInformation returned from sensors during action to help correct behavior., or retraining—just by comparing your current Core ConceptsStateThe robot’s current condition, such as joint positions, velocity, object positions, or internal variables. to the Robot LearningTrainingThe process of fitting a model using data or experience. set's k-nearest neighbors.

WHY DEVELOPERS SHOULD CARE

DARP fixes Imitation & Reinforcement LearningBehavior Cloning (BC)A simple type of imitation learning where the robot directly copies expert actions.'s Data, Distributions & Training IssuesDistribution shiftWhen the deployment data differs from the training data. problem by retrieving similar expert demonstrations at test time instead of relying on a single learned Core ConceptsPolicyThe rule or model that maps observations or states to actions.. This gives 15-46% better performance on Core ConceptsRobotA physical system with sensors and actuators that can observe the world and take actions. Manipulation & TasksManipulationUsing a robot arm or hand to move or interact with objects. and Control & PlanningControlThe method used to make the robot move the way you want. tasks without needing extra data, online Control & PlanningFeedbackInformation returned from sensors during action to help correct behavior., or retraining—just by comparing your current Core ConceptsStateThe robot’s current condition, such as joint positions, velocity, object positions, or internal variables. to the Robot LearningTrainingThe process of fitting a model using data or experience. set's k-nearest neighbors.

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