Exploring High-Level Goal Prediction for Hierarchical Imitation Learning in Robotic Manipulation

July 2026

Exploring High-Level Goal Prediction for Hierarchical Imitation Learning in Robotic Manipulation

Authors:

Kyutae Sim

Abstract:

Hierarchical imitation learning has become an effective recipe for robotic manipulation: a high-level policy predicts a sub-goal end-effector pose, and a low-level policy produces the actions that realize it. This decomposition improves generalization to unseen objects, and it exposes an interpretable interface at the sub-goal. Yet the design of the high-level goal predictor is far from settled—what it should predict, how the low-level policy should consume it, what capabilities it must encode, and how a user might steer it all remain open. This thesis investigates these questions through three studies that share one object of study:
the high-level goal predictor.

First, on the MimicGen benchmark, we study what the high level should predict. We compare conditioning a 3D diffusion policy on dense predicted goal point clouds, produced by a cross-displacement model (TAX3D), against conditioning it on four predicted endeffector points from a weighted displacement model. The sparse four-point representation is a substantially more reliable conditioning signal across four differenttasks.

Second, on RLBench, we expand the capabilities of the four-point predictor and pair it with a motion planner in place of a learned low-level policy. We add language conditioning and RGB features, and additionally predict discrete gripper open/close actions and
collision-ignore decisions, reaching performance competitive with state-of-the-art keyframeaction methods. We identify a key limitation: because the high level regresses its sub-goal, it predicts the mean of valid targets and cannot represent multi-modal ones, especially when a target admits multiple valid orientations.

Third, in a sim-to-real setting, we make the high level steerable via prompting, letting a user disambiguate among multiple valid targets on the same object—for example, which drawer to open. The prompted high level transfers zero-shot to a real robot, and in simulation it matches the opening success rate of the original single-target policy while adding target selection.

Together, these studies map the design space of high-level goal prediction and offer concrete lessons about representation, capability, and controllability for hierarchical manipulation policies.
@mastersthesis{Sim-2026-88319,
author = {Kyutae Sim},
title = {Exploring High-Level Goal Prediction for Hierarchical Imitation Learning in Robotic Manipulation},
year = {2026},
month = {July},
school = {Carnegie Mellon University},
address = {Pittsburgh, PA},
number = {CMU-RI-TR-26-71},
keywords = {Imitation Learning, Robotics Manipulation, Diffusion Policy},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.