RoboCoach: Using World Models to Locate Weak Subtasks and Improve Long-Horizon Task Success Through Targeted Teaching
Zhang Changshui and Cui Sen's team at Tsinghua University propose RoboCoach, an embodied intelligence system that uses the world model CoachWorld to run closed-loop trials of long-horizon tasks in imagination, identify the first weak subtask that times out, and request targeted human teaching for it, updating only the corresponding skill expert. Under a fixed demonstration budget, the success rate rises from 13.3% to 75.0% on Franka and from 40.0% to 83.8% on AgileX, while the uniform collection baseline reaches only 30.0% and 47.5%, respectively. On four unseen new task routes, the average success rate is 35.0%, compared with 0% for the shared-policy baseline.
Why it matters: The paper presents a complete experiment on using a world model to locate weak subtasks and improve success rates through targeted teaching; its approach to data budget allocation may be a useful reference.