mlsanityData to 14 Oct 2026

Behavior Uncloning: Distilling Mode Redirection into Policy Weights without Inference-Time Steering

arXiv 2606.29201 ↗First posted 28 Jun 2026cs.RO cs.AIcited by 1

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

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  3. 2022Classifier-Free Diffusion Guidancecited by 5,847
  4. 2017Deep reinforcement learning from human preferencescited by 3,912
  5. 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 3,080
  6. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
  7. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  8. 2022RT-1: Robotics Transformer for Real-World Control at Scalecited by 2,061
  9. 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardwarecited by 1,568
  10. 2024Octo: An Open-Source Generalist Robot Policycited by 1,484
  11. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
  12. 2024Refusal in Language Models Is Mediated by a Single Directioncited by 1,032
38 more
  1. 2022Mass-Editing Memory in a Transformercited by 1,001
  2. 2023Open X-Embodiment: Robotic Learning Datasets and RT-X Modelscited by 862
  3. 2024DROID: A Large-Scale In-The-Wild Robot Manipulation Datasetcited by 852
  4. 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
  5. 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 752
  6. 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 725
  7. 2022On Distillation of Guided Diffusion Modelscited by 725
  8. 2024PaliGemma: A versatile 3B VLM for transfercited by 696
  9. 2023BridgeData V2: A Dataset for Robot Learning at Scalecited by 657
  10. 2023Erasing Concepts from Diffusion Modelscited by 539
  11. 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learningcited by 523
  12. 2017End-to-end Driving via Conditional Imitation Learningcited by 485
  13. 2021Implicit Behavioral Cloningcited by 452
  14. 2021Brax -- A Differentiable Physics Engine for Large Scale Rigid Body Simulationcited by 407
  15. 2019Learning Latent Plans from Playcited by 316
  16. 2019Dynamics-Aware Unsupervised Discovery of Skillscited by 313
  17. 2022Behavior Transformers: Cloning k modes with one stonecited by 304
  18. 2018Residual Reinforcement Learning for Robot Controlcited by 274
  19. 2024KTO: Model Alignment as Prospect Theoretic Optimizationcited by 264
  20. 2024ManiSkill3: GPU Parallelized Robotics Simulation and Rendering for Generalizable Embodied AIcited by 229
  21. 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 204
  22. 2017DART: Noise Injection for Robust Imitation Learningcited by 191
  23. 2023Selective Amnesia: A Continual Learning Approach to Forgetting in Deep Generative Modelscited by 177
  24. 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learningcited by 166
  25. 2018Residual Policy Learningcited by 164
  26. 2024Behavior Generation with Latent Actionscited by 160
  27. 2023Concept Sliders: LoRA Adaptors for Precise Control in Diffusion Modelscited by 153
  28. 2024Steering Your Generalists: Improving Robotic Foundation Models via Value Guidancecited by 106
  29. 2025MuJoCo Playgroundcited by 78
  30. 2025Self-Improving Vision-Language-Action Models with Data Generation via Residual RLcited by 78
  31. 2023Data Quality in Imitation Learningcited by 75
  32. 2020Safe Imitation Learning via Fast Bayesian Reward Inference from Preferencescited by 73
  33. 2018Latent Space Policies for Hierarchical Reinforcement Learningcited by 67
  34. 2019Learning Reward Functions by Integrating Human Demonstrations and Preferencescited by 64
  35. 2024Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrationscited by 53
  36. 2024Re-Mix: Optimizing Data Mixtures for Large Scale Imitation Learningcited by 47
  37. 2025DynaGuide: Steering Diffusion Polices with Active Dynamic Guidancecited by 44
  38. 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

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2022Improving TD3-BC: Relaxed Policy Constraint for Offline Learning and Stable Online Fine-Tuning
  2. 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning
  3. 2025Offline Behavioral Data Selection
  4. 2022Know Your Boundaries: The Necessity of Explicit Behavioral Cloning in Offline RL
  5. 2025Residual Off-Policy RL for Finetuning Behavior Cloning Policies
  6. 2026TOUR: A Trajectory-Level Unlearning Benchmark for Offline Reinforcement Learning
  7. 2026Imitation Learning for Autonomous Driving in CARLA
  8. 2026Replay What Matters: Off-Policy Replay for Efficient LLM Reinforcement Unlearning