Behavior Uncloning: Distilling Mode Redirection into Policy Weights without Inference-Time Steering
Not on the map. Why?
Abstract (arXiv)
Behavior-cloned policies often learn multiple behavior modes from demonstration datasets, including modes that are unsafe or otherwise undesired at deployment. For example, a policy trained on diverse handover demonstrations may learn to pass a knife blade-first. Standard remedies such as data curation and inference-time steering either require access to the original demonstrations for full retraining or add substantial inference-time overhead. To address this gap, we propose MoRE(Mode Redirection), which redirects policy rollouts toward desired behavior modes through a short "uncloning" step. Specifically, MoRE distills the redirection signal from a temporary mode classifier into the policy weights to steer behavior. A retain loss balances this edit by preserving desired-mode competence, allowing the standalone policy to suppress unwanted modes with zero inference-time overhead. Across eight simulated and real-world tasks, MoRE improves the average deployment success rate (SR) by 44 percentage points over the original mixed-mode policy. Among all compared adaptation and steering baselines, MoRE achieves the strongest SR and approaches the filtered-data retraining reference, while preserving task competence and inference speed. MoRE also generalizes across robot policy backbones, including Diffusion Policy and the Pi0.5 VLA, diverse task categories, and real-world deployments.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 59 of its 65 references that are in the map. First posted on arXiv 28 Jun 2026.
Builds on
59 of 65 references.
References · 59
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2017Deep reinforcement learning from human preferencescited by 3,912
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 3,080
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2022RT-1: Robotics Transformer for Real-World Control at Scalecited by 2,061
- 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardwarecited by 1,568
- 2024Octo: An Open-Source Generalist Robot Policycited by 1,484
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
- 2024Refusal in Language Models Is Mediated by a Single Directioncited by 1,032
38 more
- 2022Mass-Editing Memory in a Transformercited by 1,001
- 2023Open X-Embodiment: Robotic Learning Datasets and RT-X Modelscited by 862
- 2024DROID: A Large-Scale In-The-Wild Robot Manipulation Datasetcited by 852
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 752
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 725
- 2022On Distillation of Guided Diffusion Modelscited by 725
- 2024PaliGemma: A versatile 3B VLM for transfercited by 696
- 2023BridgeData V2: A Dataset for Robot Learning at Scalecited by 657
- 2023Erasing Concepts from Diffusion Modelscited by 539
- 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learningcited by 523
- 2017End-to-end Driving via Conditional Imitation Learningcited by 485
- 2021Implicit Behavioral Cloningcited by 452
- 2021Brax -- A Differentiable Physics Engine for Large Scale Rigid Body Simulationcited by 407
- 2019Learning Latent Plans from Playcited by 316
- 2019Dynamics-Aware Unsupervised Discovery of Skillscited by 313
- 2022Behavior Transformers: Cloning k modes with one stonecited by 304
- 2018Residual Reinforcement Learning for Robot Controlcited by 274
- 2024KTO: Model Alignment as Prospect Theoretic Optimizationcited by 264
- 2024ManiSkill3: GPU Parallelized Robotics Simulation and Rendering for Generalizable Embodied AIcited by 229
- 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 204
- 2017DART: Noise Injection for Robust Imitation Learningcited by 191
- 2023Selective Amnesia: A Continual Learning Approach to Forgetting in Deep Generative Modelscited by 177
- 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learningcited by 166
- 2018Residual Policy Learningcited by 164
- 2024Behavior Generation with Latent Actionscited by 160
- 2023Concept Sliders: LoRA Adaptors for Precise Control in Diffusion Modelscited by 153
- 2024Steering Your Generalists: Improving Robotic Foundation Models via Value Guidancecited by 106
- 2025MuJoCo Playgroundcited by 78
- 2025Self-Improving Vision-Language-Action Models with Data Generation via Residual RLcited by 78
- 2023Data Quality in Imitation Learningcited by 75
- 2020Safe Imitation Learning via Fast Bayesian Reward Inference from Preferencescited by 73
- 2018Latent Space Policies for Hierarchical Reinforcement Learningcited by 67
- 2019Learning Reward Functions by Integrating Human Demonstrations and Preferencescited by 64
- 2024Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrationscited by 53
- 2024Re-Mix: Optimizing Data Mixtures for Large Scale Imitation Learningcited by 47
- 2025DynaGuide: Steering Diffusion Polices with Active Dynamic Guidancecited by 44
- 2025Streaming Flow Policy: Simplifying diffusion/flow-matching policies by treating action trajectories as flow trajectoriescited by 39
50 most cited of 59 All 59 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2022Improving TD3-BC: Relaxed Policy Constraint for Offline Learning and Stable Online Fine-Tuning
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning
- 2025Offline Behavioral Data Selection
- 2022Know Your Boundaries: The Necessity of Explicit Behavioral Cloning in Offline RL
- 2025Residual Off-Policy RL for Finetuning Behavior Cloning Policies
- 2026TOUR: A Trajectory-Level Unlearning Benchmark for Offline Reinforcement Learning
- 2026Imitation Learning for Autonomous Driving in CARLA
- 2026Replay What Matters: Off-Policy Replay for Efficient LLM Reinforcement Unlearning