DL0119 VLA Imitation vs RL

Explain imitation learning vs RL for VLA training. When would you reach for each one on a real robot, for example when post-training a model like Physical Intelligence’s \pi_0?

Answer

Imitation learning trains a vision-language-action model by maximum likelihood on teleoperated demonstrations: every timestep carries a ground-truth action label, so the objective is the same supervised next-token or flow-matching loss used to pretrain the underlying VLM, and no reward, reset, or simulator is required. Reinforcement learning replaces that per-step label with a scalar return and optimizes the policy on its own state distribution, which is the only way to fix the failure mode imitation cannot fix and to exceed the demonstrator’s performance. The asymmetry is practical rather than philosophical: demonstrations are cheap to collect and trivially parallel across operators, while a single real-robot RL run needs a success detector, an automated reset, and 10^4 to 10^6 environment steps of physically risky exploration. That is why essentially every published generalist VLA (RT-2, OpenVLA, \pi_0) is imitation-pretrained on cross-embodiment data such as Open X-Embodiment, and RL appears only as a narrow post-training stage on a handful of tasks where reward is machine-checkable. The interview answer is therefore not “which is better” but “imitation for coverage and language grounding, RL for the last 10 to 20 points of success rate on tasks you can actually score”.

(1) Different Objectives: imitation maximizes \log \pi_\theta(a \mid o) under a fixed data distribution, while RL maximizes expected discounted return under the policy’s own visitation distribution.
(2) Compounding Error: behavior cloning suffers covariate shift, and its regret against the expert grows as O(\epsilon T^2) in the horizon, whereas interactive or on-policy training reduces this to O(\epsilon T).
(3) Supervision Density: a demonstration gives a full action vector at every control step; a sparse success reward gives one bit per episode, so RL’s credit assignment problem is orders of magnitude harder.
(4) Performance Ceiling: imitation is capped at the demonstrator and inherits their pauses, jerk, and inconsistent strategies; RL can discover non-human solutions but will hack a badly shaped reward.
(5) Multimodality: human demos are multimodal, so a naive MSE head averages incompatible strategies, which is exactly why modern VLAs use discretized action tokens, diffusion, or flow-matching action experts plus action chunking.
(6) Hybrid Is The Default: production recipes pretrain with imitation, then apply human-in-the-loop corrections (HG-DAgger), offline RL with a conservatism penalty, or advantage-weighted fine-tuning with a KL leash to the cloned policy.

Flow diagram: teleoperated demonstrations feed imitation pretraining of a VLA, the resulting policy produces on-robot rollouts, a success detector or human takeover produces a reward signal, and an RL fine-tuning stage with advantage weighting and a KL penalty updates the policy in a loop

Figure 1: The standard VLA recipe is imitation first, RL second: demonstrations supply dense per-step labels with no reward machinery at all, and the RL loop is only viable once the cloned policy already succeeds often enough for a scalar success signal to be informative.

The sample-cost gap dominates every design decision. Collecting 100 demonstrations of a new task takes a single operator under an hour and immediately yields a policy that succeeds sometimes; reaching the same point with online RL from a random initialization means exploring a 7-DoF continuous action space with sparse reward, which is hopeless on hardware and merely expensive in simulation. Once an imitation-pretrained policy exists, RL becomes tractable because exploration starts near the solution manifold: human-in-the-loop RL methods that keep demonstrations in the replay buffer and let an operator intervene on impending failures report near-perfect success on precise, contact-rich assembly tasks after roughly one to three hours of real-world training. The corresponding risk is catastrophic forgetting: unconstrained RL on one task will happily destroy the language grounding and cross-task generalization that the imitation stage paid for, so practitioners freeze most of the VLM backbone, fine-tune the action expert, and add an explicit KL term back to the behavior-cloned reference policy.

Mathematical Formulation:
\theta_{BC} = \arg\max_{\theta} \mathbb{E}_{(o,a) \sim \mathcal{D}} \left[\log \pi_{\theta}(a \mid o)\right]
J(\pi) = \mathbb{E}_{\pi}\left[\sum_{t=0}^{T} \gamma^{t} r(s_t, a_t)\right]
J(\pi^{*}) - J(\pi_{BC}) = O(\epsilon T^{2})
J(\pi^{*}) - J(\pi_{DAgger}) = O(\epsilon T)
\max_{\theta} \mathbb{E}_{\mathcal{D}}\left[\log \pi_{\theta}(a \mid o) \exp\!\left(A(o,a)/\beta\right)\right]

Where:

  • \pi_{\theta}(a \mid o) is the VLA policy mapping an observation o (images, proprioception, language instruction) to an action or action chunk a.
  • \mathcal{D} is the fixed demonstration dataset, and \pi^{*} is the human expert whose behavior it samples.
  • s_t and t \in \{0, \ldots, T\} are the environment state and control step, r the reward, and \gamma \in (0,1] the discount.
  • \epsilon is the per-step supervised action error under the expert’s distribution; the quadratic term comes from errors pushing the policy into states absent from \mathcal{D}.
  • A(o,a) is the estimated advantage and \beta > 0 the temperature; as \beta \to \infty the advantage-weighted objective degenerates to plain behavior cloning, which is why the hybrid is a continuum rather than a binary choice.
  • Required initial condition for the RL stage: \pi_{\theta} is initialized at \theta_{BC} and the task admits an automated reset plus a reward oracle, otherwise J is not estimable on hardware.
Line chart of regret against the expert versus episode horizon for a per-step error of 0.02, comparing a quadratic behavior-cloning bound that reaches 200 at horizon 100 with a linear interactive-training bound that reaches only 2

Figure 2: With a fixed per-step error of \epsilon = 0.02, the quadratic behavior-cloning bound separates from the linear on-policy bound by a factor of T, which is why long-horizon manipulation degrades far faster than single-step prediction accuracy suggests.

PropertyImitation learning (BC)Offline RLOnline RL
Supervision per episodeOne action label at every control stepLogged actions plus reward or success labelsScalar reward, often a single success bit
What it needs to runTeleoperation rig only; no reset, reward, or simulatorA labeled buffer of mixed-quality dataAutomated reset, success detector, safety envelope
Typical budget for one task50 to 2,000 demonstrations, hours of human timeReuses existing logs; compute-bound, not robot-bound10^4 to 10^6 real steps, or a sim-to-real gap to close
Distribution optimized onExpert states only, so covariate shift is unaddressedBuffer states, with pessimism outside their supportThe policy’s own states, which is the point
Performance ceilingThe demonstrator, minus compounding errorBest behavior recoverable from the logged dataCan exceed the human on speed and precision
Main failure modeDrift into unseen states; mode averaging on multimodal demosValue overestimation on out-of-support actionsReward hacking, hardware damage, forgetting language grounding

Login to view more content


Log in to track your progress

Comments

Leave a Reply

Your email address will not be published. Required fields are marked *