中文

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

机器人学 2026-06-28 v1 人工智能

摘要

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.

引用

@article{arxiv.2606.29201,
  title  = {Behavior Uncloning: Distilling Mode Redirection into Policy Weights without Inference-Time Steering},
  author = {Hao Wang and Jiuzhou Lei and Dayou Li and Bangya Liu and Minghui Zheng and Manling Li and Ruohan Zhang and Zhiwen Fan},
  journal= {arXiv preprint arXiv:2606.29201},
  year   = {2026}
}