Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Adversarial Dual On-Policy Distillation from Expressive Teacher

About

Learning from demonstrations in embodied control is often cast as behavioral cloning, and recent diffusion or flow-matching policies improve this paradigm by modeling multi-modal expert actions. Yet these methods remain offline supervised learners: the policy is trained only on expert states and receives no corrective signal on the states it actually visits. On-policy distillation (OPD) offers a natural remedy, but standard OPD assumes a strong fixed teacher, which is unavailable in demonstration-only control. We propose \textbf{FA-OPD}, an \emph{adversarial dual on-policy distillation} method in which a Flow Matching (FM) teacher is learned from demonstrations and co-trained with a lightweight MLP student. The teacher provides two complementary signals on student rollouts. The reward channel learns an expert-likeness objective over state-action pairs and drives online exploration through long-horizon policy optimization. The action channel supplies dense local targets at student-visited states, stabilizing exploitation. FA-OPD couples them so that reward distillation enables generalization beyond point-wise demonstrations, while action distillation keeps exploration anchored near expert-like behavior. Across six robot navigation, manipulation, and locomotion benchmarks, FA-OPD beats strong baselines and shows much stronger robustness under noisy or limited demonstrations. Source code: https://github.com/vanzll/FA-OPD.

Zhenglin Wan, Jingxuan Wu, Xingrui Yu, Chubin Zhang, Mingcong Lei, Bo An, Ivor W. Tsang, Yang You• 2026

Related benchmarks

TaskDatasetResultRank
LocomotionHopper (test)
Average Return3.36e+3
8
LocomotionWalker2d (test)
Average Return4.16e+3
8
ManipulationHand-rotate (test)
Average Success Rate97.94
8
ManipulationFetch-pick (test)
Average Success Rate99.84
8
NavigationMaze2D (test)
Average Success Rate87.31
8
NavigationAnt-goal (test)
Average Success Rate82.25
8
Showing 6 of 6 rows

Other info

Follow for update