Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Hierarchical Advantage-Weighted Behavior Cloning (HABC), a novel approach for fine-tuning pretrained Variable-Length Action (VLA) policies in online reinforcement learning (RL) settings where only sparse binary outcomes are available. By employing separate critic heads for viability and efficiency and using a state-adaptive gating mechanism, HABC effectively distinguishes between different forms of feedback, leading to improved learning signals that enhance both task success and completion efficiency. In real-robot experiments, HABC significantly outperformed supervised fine-tuning baselines, achieving success rates of 92%, 88%, and 38% across three challenging bimanual tasks.
HABC achieves up to 92% success in complex robotic tasks by intelligently balancing viability and efficiency in sparse outcome scenarios.
When pretrained VLA policies are fine-tuned through online RL, each rollout episode produces only a single binary outcome (success or failure), yet the actor update requires per-transition supervision. Existing approaches commonly reduce this sparse outcome to a single scalar reward or advantage signal, which conflates distinct forms of transition-level feedback and provides limited guidance once basic task success becomes achievable. First, a single scalar signal conflates the two objectives of viability and efficiency; once basic success is achieved, the binary label provides no gradient to distinguish efficient completions from slow ones. Second, real-world rollouts mix autonomous and intervention segments; naively assigning episode outcomes across these boundaries introduces incorrect credit assignment. To address these issues, we propose Hierarchical Advantage-Weighted Behavior Cloning (HABC), which trains separate critic heads for these two objectives on different data subsets and combines their outputs with a state-adaptive balance. A state-adaptive gate g_t merges their one-step advantages, prioritizing viability when success is uncertain and shifting to efficiency only when viability is high, and converts the result into per-transition weights on the actor loss. Intervention-aware credit assignment further restricts outcome labels to segments executed by the current policy, preventing supervision from leaking across intervention boundaries. In real-robot experiments on three contact-rich bimanual tasks, HABC raises success from supervised fine-tuning (SFT) baselines of 36%, 44%, and 12% to 92%, 88%, and 38%.
Tsinghua AI