Learning More from Less:
Reinforcement Learning from Hindsight


Abstract

Reinforcement learning (RL) is increasingly used to post-train vision-language-action (VLA) models, but every update consumes robot rollouts that are slow and costly to collect, making sample efficiency a central concern. Manipulation tasks typically provide only sparse rewards, so a weak policy fails almost every rollout early in training and has little to learn from, even when those failures execute coherent behavior. Such a failure, however, is a success at a different task. We present Learning from Hindsight (LfH), which brings hindsight relabeling to RL post-training of VLAs by scoring failed rollouts against the tasks they actually achieved. A single vision-language model relabels both the instruction and the reward, proposing a hindsight instruction for a group of failed rollouts and scoring how well each satisfies it, and the policy trains on the relabeled and original rollouts jointly. Because VLAs generalize across language, relabeling in language lets the policy learn more from the same trajectories. On out-of-distribution LIBERO-PRO tasks, where standard RL improves only slowly, LfH achieves \(5\times\) improvement in sample efficiency, and outperforms a dense progress-reward baseline. The gains hold across VLA backbones and on a physical Franka robot.

1 Introduction↩︎

Reinforcement learning (RL) [1] has become a standard tool for post-training large language models [2], [3], and is increasingly being used to fine-tune vision-language-action (VLA) models for robot control [4]. In robotics, however, post-training is constrained by a much harsher data bottleneck: each RL update depends on rollouts collected from a physical robot, which are slow, expensive, and often difficult to scale. Improving sample efficiency is therefore important for making RL post-training practical for VLAs.

This challenge is especially acute in manipulation, where VLAs are commonly fine-tuned with sparse rewards. A rollout receives reward only if the robot completes the commanded instruction, leaving all other behavior uncredited. Sparse rewards are a long-standing obstacle in RL [5], [6]: before a weak policy can improve, it must first discover a successful trajectory through exploration. Early in training, nearly all rollouts therefore appear useless. For example, when instructed to “close the microwave,” the robot may instead pick up a cup (Figure 1). This behavior is coherent and task-relevant in a broad manipulation sense, but it does not satisfy the commanded instruction. Standard RL treats the entire rollout as a failure, ignoring the fact that the robot successfully executed another meaningful skill.

Our key observation is that many failed rollouts are failures only relative to the original instruction. Hindsight relabeling [7][11] exploits this by evaluating a trajectory against the task it actually achieved rather than the task it was commanded to solve. The cup-picking rollout, for instance, can be relabeled as an instance of “pick up the cup.” This is distinct from dense progress rewards [12], which provide finer feedback for the same commanded instruction. Hindsight changes the task assignment itself, converting an otherwise unrewarded failure into a successful demonstration of a different, related instruction. Since VLAs are conditioned on language and can generalize across instructions, such relabeled rollouts provide useful supervision from data that standard RL would ignore. The remaining challenge is to infer, in language, what the robot actually did; a pretrained vision-language model (VLM) [13] can provide this description directly from the rollout observations.

We propose Learning from Hindsight (LfH), a method for RL post-training of VLAs that turns failed rollouts into additional training signal. LfH uses a single VLM to relabel both instructions and rewards. Given a group of rollouts that fail the commanded instruction, the VLM examines one rollout and proposes a hindsight instruction describing the behavior the robot achieved. It then scores how well each rollout in the group satisfies that hindsight instruction. The VLA is trained jointly on the original rollouts and their hindsight-relabeled counterparts, allowing the policy to learn both from intended successes and from unintended but meaningful behaviors.

We evaluate LfH by fine-tuning VLAs on out-of-distribution LIBERO-PRO tasks [14], where the initial policy has near-zero success and standard RL improves only slowly. LfH recovers usable signal from most failed rollouts, reaches the final success rate of standard RL in roughly one fifth of the training steps, and yields a \(5\times\) improvement in sample efficiency. It also outperforms a dense progress-reward baseline, indicating that in sparse, low-success regimes, changing which task a rollout is credited for can be more valuable than assigning denser feedback to the original command alone. These gains hold across VLA backbones and transfer to a physical Franka robot.

Figure 1: Learning from Hindsight. A trajectory that fails the commanded instruction can still exhibit meaningful behavior. LfH uses a VLM to relabel the achieved behavior in language, turning otherwise unused failures into training signal for VLA fine-tuning.

2 Related Work↩︎

Hindsight Relabeling. Hindsight Experience Replay (HER) [7] established this idea by relabeling goals with the states that were actually achieved by the agent as opposed to the original goal. HER has also been extended to image goals with imagined goal generators [11], [15], goal-conditioned value functions [10], and self-supervised approaches [9]. Concurrent work on open-ended instruction relabeling [16] demonstrates language-based hindsight in game-like environments, while we provide a unique algorithm and implementation of language-based hindsight that we prove is effective in the robotics setting.

Reward Shaping and Foundation Reward Models. Instead of hindsight relabeling, a classic way to address sparse-reward RL is reward shaping, which provides intermediate feedback that guides the agent toward the desired behavior [17]. Since such shaping can be prohibitively expensive to design by hand, recent work has proposed large-scale dense reward models for transferable reward learning. Prior work trains text-conditioned progress models, either as goal-reaching value functions [18], [19], with interpolation heuristics over demonstrations [12], [20], [21], or using preference labels [22]. LfH is complementary to dense progress rewards: rather than densifying feedback for fixed tasks, LfH generates new diverse tasks with relabelling and bridges success to the original tasks via generalization.

Improving Language Steerability. Besides reinforcement learning, another way to raise VLA task completion is to strengthen language steerability. A common strategy is data augmentation under an imitation learning or offline RL paradigm, using VLMs to relabel or synthesize instructions over an offline dataset to expand task diversity [23][26]. A complementary direction steers at inference time: [27] use a VLM to pick the sampled action best matching the VLA’s original text instruction. Our method differs from these methods as it is 1) online and 2) bakes the signal from the relabelled tasks into the weights, instead of doing test-time steering.

3 Preliminaries↩︎

We study RL fine-tuning of vision-language-action (VLA) policies [4], [28]. At the start of each episode the environment samples an instruction \(g \sim P_g\) and an initial RGB observation \(o_0\). A policy \(\pi_\theta\) rolls out actions \(a_t \sim \pi_\theta(\cdot \mid o_t,g)\) for \(T\) steps, yielding a trajectory \(\tau = (o_0,a_0,\ldots,a_{T-1},o_T)\) that receives a binary success reward \(R(\tau,g)\), equal to \(1\) when \(\tau\) completes \(g\) and \(0\) otherwise. The objective is to maximize expected success \(J(\theta) = \mathbb{E}_{g \sim P_g,\, \tau \sim \pi_\theta(\cdot \mid g)}[R(\tau,g)]\).

We instantiate LfH with group relative policy optimization (GRPO) [29]. For each instruction \(g\), GRPO samples a group of \(K\) trajectories \(\{\tau_i\}_{i=1}^K\) from \(\pi_{\theta_{\mathrm{old}}}\), scores them with \(R_i = R(\tau_i,g)\), and normalizes advantages within the group: \[\begin{align} \label{eq:grpo95adv} \hat{A}_i = \frac{R_i-\mu_R}{\sigma_R+\delta}, \qquad \mu_R = \frac{1}{K}\sum_{j=1}^K R_j, \end{align}\tag{1}\] where \(\sigma_R\) is the group standard deviation and \(\delta\) a small constant. With the importance ratio \[\begin{align} \label{eq:grpo95defs} r_{i,t}(\theta) = \frac{\pi_\theta(a_{i,t}\mid o_{i,t},g)}{\pi_{\theta_{\mathrm{old}}}(a_{i,t}\mid o_{i,t},g)}, \end{align}\tag{2}\] and a KL penalty \(D_{i,t}(\theta)\) to a reference policy, GRPO maximizes \[\begin{align} \label{eq:grpo} \mathcal{L}_{\mathrm{GRPO}}(\theta) = \mathbb{E}_{g,\{\tau_i\}}\!\left[ \frac{1}{K}\sum_{i=1}^K \sum_{t=0}^{T-1} \ell_{i,t}(\theta) \right], \qquad \ell_{i,t} = \min(r_{i,t}\hat{A}_i,\, \bar r_{i,t}\hat{A}_i) - \beta D_{i,t}, \end{align}\tag{3}\] with \(\bar r_{i,t} = \mathrm{clip}(r_{i,t},1-\epsilon,1+\epsilon)\). Groups with zero reward variance \(\sigma_R\) provide no learning signal and are discarded.

4 Method: Learning from Hindsight (LfH)↩︎

Problem. A GRPO group provides no learning signal when all its trajectories receive the same reward. All-one groups \(\{\tau_i \mid R_i = 1\}\) already contain successful behavior; the bottleneck is all-zero groups \(\{\tau_i \mid R_i = 0\}\), which have zero normalized advantage in Eq. 1 and contribute no gradient. Since the reward depends on both the trajectory \(\tau\) and the commanded instruction \(g\), a weak initial policy fails most rollouts, leaving many groups all-zero and fine-tuning makes little progress. LfH recovers learning signal from these otherwise discarded trajectories.

Approach. A trajectory that fails under one instruction may succeed under another. A robot commanded to “close the microwave” may instead pick up a mug: this rollout fails for \(g=\text{``close the microwave''}\) but succeeds for the hindsight instruction \(g'=\text{``pick up the mug''}\). LfH relabels failed trajectories with instructions describing what the robot actually accomplished, converting zero-reward samples \((\tau,g)\) into successful examples \((\tau,g')\). The hindsight instruction \(g'\) need not be drawn from \(P_g\): in the low-success regime a failure under the original instruction gives no signal, while a hindsight-labeled success does. LfH lets the policy learn from what it already manages to do before it can reliably solve the commanded tasks. Two questions remain: (1) how to generate hindsight-labeled data (Section 4.1), and (2) how to incorporate it into GRPO (Section 4.2).

4.1 Training Signals from Hindsight↩︎

For each commanded instruction \(g\), GRPO samples a group \(\{\tau_i\}_{i=1}^K\) and scores it with \(R_i = R(\tau_i,g)\). LfH activates only on low-signal groups, i.e. those with mean reward \(\frac{1}{K}\sum_i R_i < \eta\); high-reward groups already reinforce the commanded behavior, so relabeling them would replace a correct training signal with an alternative one. Given a low-reward group, LfH converts it into a hindsight-labeled group in two steps.

Instruction relabeling. We infer an alternative prompt under which a failed rollout still provides supervision. Since robot rollouts may lack privileged object states or task annotations, the VLM relabeler \(\mathcal{M}_\psi\) works from available trajectory information such as RGB observations and actions. LfH samples an anchor from the failed trajectories, \(i^\star \sim \mathrm{Unif}(\{i:R_i=0\})\), since successful trajectories already carry a reliable label. The VLM then generates a hindsight instruction \(g^\prime\) describing what the anchor did: \[\label{eq:lfh95instr} g' \sim \mathcal{M}^{\mathrm{inst}}_\psi(\cdot \mid \tau_{i^\star}).\tag{4}\] Not every failure is useful: some trajectories contain only accidental motion. To avoid noisy prompts, we first ask the VLM to classify whether the anchor contains meaningful behavior and discard those labeled uninteresting.

Reward relabeling. LfH then evaluates every trajectory in the group under the shared hindsight instruction \(g^\prime\): \[\label{eq:lfh95reward} \tilde{R}_i = \mathcal{M}^{\mathrm{rew}}_\psi(\tau_i,g') \in \{0, 0.5, 1\},\tag{5}\] where \(1\) denotes completion of \(g'\), \(0\) failure, and \(0.5\) an ambiguous outcome. A shared instruction is essential because GRPO computes advantages relative to other trajectories in the group; relabeling each with a different prompt would make their rewards incomparable. This yields the hindsight group \(\tilde{\mathcal{G}}=\{(\tau_i,g',\tilde{R}_i)\}_{i=1}^K\), which GRPO consumes like an ordinary rollout group, computing advantages from \(\{\tilde{R}_i\}\) in place of \(\{R_i\}\). Because relabeling fires only on low-reward groups, LfH leaves the standard GRPO update untouched wherever the commanded instruction already provides signal, and acts only on failed groups that would otherwise yield no gradient.

4.2 GRPO with Hindsight Relabeling↩︎

For a hindsight-labeled group \(\tilde{\mathcal{G}}\), we compute group-normalized hindsight advantages \[\label{eq:lfh95adv} \tilde{A}_i = \frac{\tilde{R}_i-\tilde{\mu}_R}{\tilde{\sigma}_R+\delta}, \qquad \tilde{\mu}_R = \frac{1}{K}\sum_{j=1}^K \tilde{R}_j,\tag{6}\] with \(\tilde{\sigma}_R\) the standard deviation of \(\{\tilde{R}_j\}_{j=1}^K\). Since the trajectories were rolled out under the commanded instruction \(g\) but we now optimize the policy at the hindsight instruction \(g'\), we apply an importance correction following hindsight policy gradients [8], replacing the standard ratio with \[\label{eq:lfh95ratio} \tilde{r}_{i,t}(\theta) = \frac{\pi_\theta(a_{i,t}\mid o_{i,t},g')}{\pi_{\theta_{\mathrm{old}}}(a_{i,t}\mid o_{i,t},g)},\tag{7}\] which evaluates the action under the hindsight instruction \(g'\) while correcting for its having been sampled under \(g\). With \(\tilde{D}_{i,t}(\theta)\) the KL penalty between \(\pi_\theta(\cdot\mid o_{i,t},g')\) and the reference policy, the hindsight objective is \[\label{eq:lfh95hgrpo} \mathcal{L}_{\mathrm{H\text{-}GRPO}}(\theta) = \mathbb{E}_{\tilde{\mathcal{G}}}\!\left[ \frac{1}{K}\sum_{i=1}^K \sum_{t=0}^{T-1} \tilde{\ell}_{i,t}(\theta) \right], \quad \tilde{\ell}_{i,t} = \min(\tilde{r}_{i,t}\tilde{A}_i,\, \tilde{\bar r}_{i,t}\tilde{A}_i) - \beta \tilde{D}_{i,t},\tag{8}\] where \(\tilde{\bar r}_{i,t} = \mathrm{clip}(\tilde{r}_{i,t},1-\epsilon,1+\epsilon)\). We train on both commanded and hindsight groups, \[\label{eq:lfh95obj} \mathcal{L}_{\mathrm{LfH}}(\theta) = \mathcal{L}_{\mathrm{GRPO}}(\theta) + \lambda\,\mathcal{L}_{\mathrm{H\text{-}GRPO}}(\theta),\tag{9}\] so the GRPO term keeps training aligned with \(P_g\) while the hindsight term, weighted by \(\lambda\), recovers signal from low-reward groups.

Figure 2: Learning from Hindsight (LfH)

5 Experiments↩︎

We organize our experiments around the following questions: (1) Does LfH improve the sample efficiency of GRPO fine-tuning under sparse rewards? (Sections 5.1 and 5.2) (2) How does LfH improve sample efficiency? (Section 5.3) (3) Is LfH effective on physical robot experiments? (Section 5.4). Additional experimental results are present in Appendix 7.1 and 7.2.

General experiment setup. Unless otherwise specified, all methods are initialized from the same RLinf-Pi05-LIBERO-SFT checkpoint [30], which was trained with few-shot demonstrations on the four LIBERO task suites [31] in simulation. We implement LfH and the baselines in RLinf1. LfH uses Qwen3-VL-235B-A22B-Thinking-FP8 for both instruction relabeling and reward relabeling (Section 4.1). Trajectories are represented as sequences of video frames from the global camera facing the robot.

5.1 Illustrative Example↩︎

We use a task “close the microwave,” from LIBERO-90 to illustrate how LfH improves sample efficiency and helps RL overcome a cold start from a weak initial policy.

LfH makes better use of collected rollouts. The initial policy has zero success on the target task. Consequently, during standard GRPO training, almost all trajectory groups receive all-zero rewards and are discarded, as shown in Figure 3 (a) (a). In contrast, LfH retains nearly 80% of trajectory groups after hindsight relabeling. Since GRPO filters out groups with no reward variation, this result shows that LfH recovers a large fraction of data that would otherwise provide no learning signal. The remaining 20% of groups are still discarded because they correspond to uninformative failures for which the VLM relabeler does not assign a meaningful hindsight instruction and rewards.

LfH improves sample efficiency under sparse rewards. By converting otherwise zero-reward failures into relabeled training examples, LfH enables GRPO to update from useful data even when the original task reward is absent. As shown in Figure 3 (a), GRPO+LfH steadily improves success on the target task over training, whereas standard GRPO remains at zero success throughout. This suggests that hindsight relabeling can provide the initial learning signal needed to escape cold-start regimes where sparse task rewards alone are insufficient.

What does LfH relabel? Figure 3 (c) visualizes representative relabeled trajectories. In this example, the initial policy repeatedly grasps and bumps the mug, behavior that is unrelated to the commanded task “close the microwave.” LfH relabels these failures with instructions describing what the robot actually did. Interestingly, although these hindsight instructions are not semantically related to the target task, the relabeled data still help RL make progress on the target policy. One possible explanation is that relabeling provides a contrastive grounding signal: trajectories that do not complete the target task are explicitly associated with alternative instructions, helping the policy distinguish the target instruction from behaviors induced by spurious or irrelevant actions.

a

b

c

Figure 3: (a) Fraction of groups kept in GRPO versus LfH during training. LfH keeps more trajectory groups for training through relabeling, while GRPO discards most groups. LfH also escapes the weak initial policy with zero success rate and reaches 60% success. (b) Example hindsight relabeled trajectories with their respective VLM-relabled instructions. Although the relabeled instructions are not the target task of closing the microwave, they provide useful learning signal..

5.2 Can LfH improve sample efficiency?↩︎

Setup. We evaluate whether LfH can accelerate RL fine-tuning when the initial VLA policy is weak. To this end, we construct out-of-distribution (OOD) tasks from the LIBERO-PRO dataset with the task perturbation. These tasks preserve the original scenes, object instances, and environment dynamics, but change the task specification, such as the instructed object, target state, or desired action. For example, a policy may have seen the same kitchen scene during SFT, but must now manipulate a new object or reach a different goal state. This setting isolates the central challenge of sample-efficient RL: improving from mostly failed rollouts.

Implementation & Baselines. We train each method for 40 steps, using the same number of collected rollouts for both GRPO and LfH, since GRPO plateaus. For all experiments, we run with four random seeds and report the mean and standard error at each step. We compare LfH with standard GRPO as well as GRPO trained with RoboMETER dense progress rewards [12]. Standard GRPO uses the sparse environment success reward under the original instruction. GRPO+RoboMETER replaces this sparse signal with a dense progress reward toward this instruction. This baseline tests whether providing denser rewards is sufficient to match the sample-efficiency benefit of LfH. All baselines are trained for the same number of steps and with the same number of collected environment rollouts.

Metric. During training, we periodically freeze the policy and evaluate it with 480 rollouts. We report the normalized gain over the initial policy. Let \(\mathrm{SR}_0\) denote the success rate of the initial policy, and let \(\mathrm{SR}_t\) denote the evaluation success rate after collecting \(t\) training rollouts. We define \(\mathrm{Gain}(t)=\frac{\mathrm{SR}_t}{\mathrm{SR}_0}-1.\) This metric measures how quickly each method improves relative to the initial checkpoint as more rollouts are collected. Additional implementation and evaluation details are provided in Appendix 8.

Results. Figure [fig:sample95efficiency] shows that LfH learns much faster than GRPO. LfH matches GRPO’s final performance in about 5 training steps, while GRPO takes nearly 30 steps, yielding roughly a \(5\times\) gain in sample efficiency. LfH also outperforms GRPO with RoboMETER rewards, indicating that dense progress feedback alone is not sufficient when the policy rarely makes progress toward the commanded instruction.

Figure [fig:sample95efficiency95group95kept] helps explain this gain. Standard GRPO and GRPO+RoboMETER drop the groups of trajectories with zero reward variance (Section 3) and keep only about 20–40% of trajectory groups for training, while LfH keeps roughly 70–80%. Thus, LfH does not merely assign denser rewards; it changes how much of the collected experience becomes useful for learning. By relabeling failed rollouts with hindsight instructions, LfH turns many otherwise discarded groups into positive training signal.

Generality. Finally, we test whether this benefit transfers beyond a single VLA backbone. Using the same training protocol, we fine-tune additional VLA models: GR00T [32] for 200 steps and OpenVLA-OFT [28] for 60 steps on the LIBERO PRO Spatial, Goal, and Object suites (again with task perturbation). Figure [fig:generality] shows that LfH also consistently improves sample efficiency across these VLAs.

Figure 4: (a) Gain during training for vanilla GRPO, Robometer, and LfH. LfH improves sample efficiency over GRPO and RoboMETER, achieving higher gain with fewer training steps. (b) The fraction of groups kept during training. LfH keeps a larger fraction of trajectory groups usable for training, making better use of collected data. (c) Final gain for other VLA models. The benefit of LfH transfers beyond \pi_{0.5}, improving the performance on both GR00T and OpenVLA-OFT.

5.3 Analysis: How does LfH improve sample efficiency?↩︎

Can LfH performance gain be attributed to instruction augmentation or reward relabeling alone? We isolate the two steps in LfH: hindsight instruction relabeling and reward relabeling (Section 4). We compare against two ablations. Rephrase-only replaces the hindsight instruction \(g'\) with a paraphrase of the original instruction \(g\) while keeping the original rewards, testing whether language augmentation alone explains the gain. Reward-only keeps the instruction as \(g\) but relabels rewards with the VLM, testing whether reward relabeling alone explains the gain. Figure 5 (a) shows that both ablations marginally improve over GRPO, remain substantially below the full LfH method throughout training. This indicates that LfH is not merely instruction augmentation or reward relabeling; its benefit comes from coupling the two.

Can LfH performance gain be attributed to perturbed rewards? We next test whether LfH works simply because it injects additional reward variance into otherwise discarded groups. The random-reward baseline creates nonzero advantages without using semantically meaningful hindsight labels. The result in Figure 5 (a) shows that the limited gains from using random rewards in the modified batch do not provide enough additional learning signal to account for the gains achieved by LfH. Thus, the benefit of LfH is not merely due to avoiding zero-variance groups. Semantically grounded hindsight rewards provide useful credit assignment, whereas arbitrary reward perturbations produce noisy gradients that do not reliably improve the commanded tasks.

a

b

Figure 5: (a) Comparison of full LfH with ablations that remove/modify parts of the hindsight relabeling mechanism. LfH’s gains come from jointly relabeling instructions and rewards; instruction rephrasing or reward relabeling alone, or additional reward variation does not substantially improve performance. (b) Real-world physical-robot fine-tuning results. LfH improves success rate over GRPO across rollout budgets..

5.4 Real-world results↩︎

Setup. We evaluate LfH on a real Franka FR3 robot with manipulable objects, two drawers, and a two-level rack. We first SFT \(\pi_{0.5}\) on 10 language-conditioned manipulation tasks using 20 SpaceMouse demonstrations per task, spanning short pick-and-place and longer multi-stage tasks. We then fine-tune on a held-out task, put the green container into the bowl, where the initial SFT policy has zero success. Both GRPO and GRPO+LfH use the same binary human reward. See Appendix 8.2 for details.

Results. Figure 5 (b) shows that LfH improves real-world sample efficiency. Both methods start from zero, but LfH improves faster. At 128 rollouts, LfH achieves roughly twice the success rate of GRPO; at 160 rollouts, LfH reaches \(56\%\) success, while GRPO plateaus at \(22\%\).

Analysis. The initial policy often grounds prompts to the wrong object, such as grasping the tape regardless of the commanded language prompt. GRPO treats these rollouts only as failures. LfH instead relabels them with prompts describing the executed behavior, such as “pick up the tape” or “move the tape to the bowl,” turning failed target-task attempts into useful supervision for object grounding and manipulation.

6 Discussion and Conclusion↩︎

More bits per robot sample. LfH is motivated by a simple question: how many useful bits do we extract from each costly robot rollout? Classic RL uses a rollout mainly to evaluate whether the intended outcome happened, often reducing a rich trajectory to one reward. This is inefficient when physical interaction is expensive. Following the broader view that learning systems should extract more information from each sample [33], LfH treats a trajectory as evidence for all language goals consistent with what happened. In this sense, hindsight relabeling increases the bits per sample: one rollout can supervise multiple possible tasks, rather than only the task originally commanded.

Practical implication. This is especially natural for pretrained VLAs. Because these models already generalize across related language instructions, training on what the robot actually did can improve what it was asked to do. LfH therefore changes the role of robot data: failed attempts are no longer merely negative examples, but reusable experience that can be mined for nearby tasks. This offers a practical path for VLA post-training: instead of only collecting more rollouts or taking more optimization steps on the same labels, we can reinterpret each rollout through language and transfer learning across the tasks it supports.

Limitations. LfH is limited by the behaviors present in the collected data. If the policy produces repetitive or uninformative trajectories, there are few meaningful hindsight goals to recover. It also depends on relabeling quality: incorrect or overly generic prompts can introduce noisy supervision. Better filtering, uncertainty estimation, and lightweight human verification are important directions for reliable deployment.

7 Additional experimental results↩︎

7.1 Additional ablation studies↩︎

What is the best way to select groups for hindsight relabeling? We ablate which rollout groups should be selected for hindsight relabeling. Relabeling every trajectory is not necessarily beneficial: VLM relabeling adds compute, and applying hindsight updates to already successful or near-saturated groups can introduce redundant or noisy supervision. Therefore, our default strategy relabels only groups whose mean original-task reward is below a threshold, focusing LfH on groups with enough failed trajectories that hindsight relabeling can meaningfully transform them into successful examples under alternative instructions.

a

b

Figure 6: (a) Selecting groups with mean reward below 0.8 outperforms random selection and relabeling all groups, showing that targeted relabeling is important. (b) Among the tested thresholds, 0.8 performs best, suggesting that LfH benefits from relabeling a broad set of non-saturated groups while avoiding indiscriminate relabeling of all groups..

As shown in Figure 6 (a), threshold-based group selection is important. We compare three group-selection strategies. LfH selects groups whose mean reward is below 0.8. Random selection keeps 70% of groups uniformly at random, matching the fraction of groups retained by our LfH filter. The all-groups variant applies hindsight relabeling to every group. Relabeling groups with mean reward below 0.8 performs substantially better than random group selection and relabeling all groups. Random selection uses the relabeling budget inefficiently because it often spends hindsight queries on groups where relabeling is not useful (e.g. all successful groups). Relabeling all groups is also worse than threshold-based selection, suggesting that indiscriminate relabeling can dilute the learning signal or introduce unnecessary hindsight updates on trajectories that already provide useful original-task supervision.

We also ablate the threshold used for filtering groups. As shown in Figure 6 (b), a very low threshold, such as 0.1, relabels only all-failed groups and therefore misses many trajectories that fail the commanded task but still contain meaningful, reusable object interactions. A more moderate threshold, such as 0.5, improves over this stricter setting but still underperforms the default threshold of 0.8. The threshold of 0.8 provides the best performance, indicating that LfH benefits most when relabeling a broad set of non-saturated groups.

Does increasing updates per rollout match LfH? A natural question is whether LfH’s gains could be matched by simply increasing the update-to-data ratio. To test this, we compare LfH against GRPO and PPO with different update epochs, where a larger update epoch means taking more gradient steps on the same collected rollout batch before collecting new data. Figure 7 shows that increasing update epochs does not consistently improve performance and can even hurt, indicating that there is a narrow sweet spot for data reuse. More importantly, even the best-tuned GRPO and PPO variants remain below LfH. This suggests that the bottleneck is not merely under-optimization, but the usefulness of the supervision contained in each rollout: repeatedly updating on failed trajectories under the original prompt has limited value, whereas LfH relabels those trajectories with prompts they can support, making each interaction more informative.

a
b

Figure 7: (a) Increasing the number of GRPO update epochs does not match LfH, showing that the main bottleneck is not insufficient optimization on the same rollout data. (b) Increasing PPO update epochs also fails to consistently match LfH, further indicating that LfH improves sample efficiency by making each collected trajectory more informative rather than by simply reusing data more aggressively.. a — LfH vs GRPO, b — LfH vs PPO

How does data diversity affect the performance of LfH? We vary the Pi-0.5 rollout sampling noise during data collection to control the diversity of collected trajectories. Figure 8 shows that both GRPO and LfH exhibit an optimal noise level: too little noise limits exploration, while too much noise makes rollouts less coherent and less useful for learning. However, LfH appears to benefit more from this optimal noise than standard GRPO. This suggests that data diversity is especially important for LfH. While GRPO can only use diversity when it leads to higher commanded-task reward, LfH can additionally relabel diverse failures with hindsight instructions and rewards. In this way, exploration produces alternative object interactions that LfH can convert into useful supervision by leveraging the VLA’s pretrained knowledge about language.

Limitations of LfH. LfH depends highly on the quality of the failures it receives. When the policy produces diverse, interpretable failures, hindsight relabeling takes advantage of these useful alternative goals. However, when it collapses to repetitive or uninformative behavior, there may be little meaningful signal to relabel. Also, LfH depends on the VLM relabeler. Incorrect object names, overly generic hindsight instructions, or hallucinated achievements can turn failed trajectories into noisy positive examples, while reward relabeling can introduce false positives or false negatives. We are able to partially mitigate these issues with the two mechanism shown in Figure 9: Nothing filters out rollouts with no confirmed object contact (such as just hovering or erratically moving around) and unsure gives the option assigns reward \(0.5\) instead of assigning a false negative or positive reward. However, this does not fully mitigate these limitations, and for certain tasks where the failures are not meaningful, LfH may be less suitable.

Figure 8: Ablation on the noise level used for rollout sampling. As the rollout sampling noise level increases, trajectory diversity increases and LfH benefits more than GRPO, showing that LfH actively benefits from and encourages a more diverse rollout distribution during training.

Impact of the VLM prompt on performance. We ablate which rollout groups should be selected for hindsight relabeling on the LIBERO-PRO Goal suite with task perturbation using Pi-0.5. The full LfH prompt (8.1.0.2) includes three key components: (1) a list of scene objects provided to the VLM, (2) a “Nothing” option that allows the relabeler to identify uninteresting trajectories, which are then filtered out during training, and (3) an “unsure” option that allows the reward evaluator to abstain when task completion cannot be visually verified, resulting in reward 0.5. We compare this prompt against variants that remove each of these components.

Figure 9: VLM prompt ablation. Removing the scene object list, the Nothing option for filtering uninteresting trajectories, or the unsure option for reward evaluation each reduces the sample-efficiency gains of LfH.

As shown in Figure 9, all three components contribute to the sample-efficiency gains of LfH. Removing the object list makes relabeling more open-ended: the VLM must infer object identities from visual appearance alone, which increases the likelihood of inconsistent, overly specific, or ungrounded labels. Removing the “Nothing” option forces the VLM to assign an instruction even to idle, contact-free, or otherwise uninformative trajectories, turning useless rollouts into noisy positive hindsight examples. Removing the “unsure” option from reward evaluation forces ambiguous videos to be classified as either success or failure, which can introduce incorrect binary rewards when the final goal state is not clearly visible.

These results suggest that effective hindsight relabeling requires not only a capable VLM, but also a prompt that constrains the VLM to produce meaningful relabelings.

7.2 Additional analysis of LfH↩︎

When does LfH help the most? LfH is most effective when failures are diverse and semantically interpretable. Figure 10 shows sample rollouts at training step 40 on the microwave task. Standard GRPO produces highly similar trajectories: the robot repeatedly approaches the same mug with similar gripper poses, leading to an undiverse failure mode. In contrast, LfH rollouts show more varied object interactions and spatial configurations around the microwave and mugs. Since these samples are taken after training has progressed, this difference suggests that LfH may do more than exploit diversity already present in the initial policy; it may also help maintain a more diverse behavioral distribution during RL fine-tuning. One possible explanation is that hindsight relabeling gives credit to multiple meaningful alternative behaviors, whereas GRPO treats them all as failures under the commanded task. As a result, GRPO can collapse toward repeated unsuccessful behaviors, while LfH continues to reinforce a broader set of grounded object interactions.

a
b

Figure 10: Random mid-training rollouts on the microwave task for (a) GRPO and (b) LfH. Standard GRPO collapses toward repeated, low-diversity failure modes, while LfH maintains more diverse and semantically meaningful behavior, indicating that hindsight relabeling not only benefits from, but also helps preserve useful behavioral diversity during training.. a — Repeated GRPO failures, b — Diverse LfH rollouts

Can LfH improve sample efficiency on other OOD perturbations? We next evaluate a less obvious setting for language-based hindsight: position perturbations. Here, the commanded language prompt is unchanged, but the initial object positions are shifted. Since LfH relabels trajectories in language space, it is not immediately clear that it should help when only the environment configuration changes. Surprisingly, the preliminary results in Figure 14 suggest that LfH can still improve over GRPO on position-perturbation tasks, both in sample efficiency and maximum success rate. This points to a possible broader benefit of LfH: relabeled failures may help the policy reuse experience from shifted layouts and learn object interactions that can transfer back to the commanded task.

7.3 Full Training Curves↩︎

The main metric used is Gain as defined in Section 5.2 averaged across suites. However, below we also report the full training curves for each method and model per suite. We report both the raw success rate and the maximum cumulative success.

Figure 11: Full training curves for Pi-0.5
Figure 12: Full training curves for GR00T
Figure 13: Full training curves for OpenVLA-OFT
Figure 14: Full training curves for Pi-0.5 on position perturbation

8 Implementation details↩︎

8.1 LfH experimental details↩︎

8.1.0.1 Hyperparameters.

Throughout the experiments, we use \(64\) environments with \(1\) rollout collection per step. We use group size \(N=8\), microbatch \(32\), global batch \(512\), \(\mathrm{KL\_beta}=0\) (no KL penalty), two gradient update epochs, and VLM relabeling temperature \(0.8\).

8.1.0.2 Prompts.

LfH uses the VLM to both relabel the prompt and also assign rewards. The relabeling prompt is:

Watch this video of a robot arm performing a manipulation task.

The robot was originally instructed to:

"{original_instruction}"

However, the robot may have FAILED or done something DIFFERENT from the original instruction.

Your job is to describe what the robot ACTUALLY DID based solely on visual evidence.

Objects present in the scene:

{scene_objects_text}

First, determine whether the robot did anything INTERESTING.

Interesting behavior means the robot PHYSICALLY CONTACTED an object — the gripper fingers visibly touched or closed around it.

Uninteresting behavior means the robot merely hovered near an object, approached without making contact, or stayed idle.

CONTACT CHECK — before claiming the robot touched anything, ask yourself:

- Did the gripper fingers visibly close around or press against the object? - Did the object move, tilt, or shift in response to the gripper? - Did the object’s shadow change?

If NONE of these are true, the robot did NOT contact the object — even if the gripper moved close to it.

If the behavior is UNINTERESTING (no confirmed physical contact), output exactly:

<thought>brief reasoning about why nothing interesting happened</thought>

<final>Nothing</final>

If the behavior IS interesting (confirmed contact), focus on:

1. Which object(s) did the gripper physically contact? 2. Where did the object(s) move to? 3. What is the final spatial arrangement?

Rules:

- Write ONE imperative instruction (e.g., "pick up the mug and place it in the microwave"). - Use SHORT, GENERIC names for objects: strip any numeric suffixes (_1, _2) and brand/material prefixes (e.g. "akita_black_bowl_1" → "bowl", "flat_stove_1" → "stove"). - Do NOT copy the original instruction — describe what you SEE. - Under 20 words.

Output:

<thought>brief reasoning about what happened</thought>

<final>your instruction</final>

The reward prompt is:

Watch this video of a robot arm performing a manipulation task.

The robot was instructed to:

"{instruction}"

Determine whether the robot FULLY and SUCCESSFULLY completed the instruction.

Follow these steps:

1. FIRST FRAME: Describe the positions of all relevant objects at the start. 2. LAST FRAME: Describe the positions of all relevant objects at the end. 3. CHANGES: What specifically changed between the first and last frames? List each object that moved and where it moved to. 4. VERIFICATION: For each requirement in the instruction, state whether it is satisfied based on what you can clearly see in the last frame. 5. JUDGMENT: Are ALL requirements satisfied?

Rules:

- Only describe what you can clearly see in the frames. Do NOT infer or assume actions that are not visible — narrating a plausible sequence of events is not evidence of completion. - The task is complete ONLY if the last frame shows the EXACT goal state: correct objects in the correct locations as specified in the instruction. - "In" a compartment/container means INSIDE the enclosed space — not on top, not adjacent, not leaning against it. - Moving an object to the WRONG location is a failure (e.g. wrong compartment, wrong side of the table), even if the object was moved. - The robot arm may still be touching the object — what matters is whether the object is at the target location. - If you cannot clearly verify the target location in the last frame, answer UNSURE.

Output format must be exactly:

<thought>your step-by-step analysis following steps 1-5 above</thought>

<answer>yes</answer> or <answer>no</answer> or <answer>unsure</answer>

Use YES if the goal state is clearly achieved.

Use NO if it clearly is not achieved.

Use UNSURE if the evidence is ambiguous (e.g. you cannot tell which compartment an object is in, or whether it is on top vs inside).

8.1.0.3 LfH Group selection.

Trajectories are grouped into sets of \(N{=}8\). Groups whose mean reward falls within \([\theta_\text{low}, \theta_\text{high}]=[0.8, 1.0]\) are skipped; all other groups (mean reward \(< 0.8\)) are selected for hindsight relabeling. This focuses relabeling on groups with substantial failure rates where hindsight instructions provide the most value, while leaving near-saturated groups untouched.

8.1.0.4 Hindsight loss clipping.

LfH can occasionally produce very large hindsight-policy losses, especially early in training when hindsight rewards are noisy. To prevent a small number of hindsight groups from dominating optimization, we clip the hindsight GRPO loss before applying the hindsight weight: \[L_{\mathrm{LfH}}(\theta) = L_{\mathrm{GRPO}}(\theta) + \lambda \cdot \min\left(L_{\mathrm{H\text{-}GRPO}}(\theta), c\right),\] where \(c\) is a clipping threshold and \(\lambda\) controls the overall contribution of hindsight updates. This bounds the maximum contribution of the hindsight term to \(\lambda c\), while leaving the standard GRPO objective unchanged.

We keep this maximum hindsight contribution fixed across all experiments. To set the clipping threshold, we estimate the empirical distribution of hindsight losses during the early stage of training and choose \(c\) as approximately the 85th percentile. We then choose \(\lambda\) such that the weighted LfH batch loss is roughly equal to the original batch policy loss.

8.1.0.5 Same-instruction skip.

Before reward evaluation, any trajectory whose relabeled instruction matches the original task (after case/punctuation normalization) reverts to its original instruction and environment reward, avoiding redundant VLM queries.

8.2 Real-world experimental details↩︎

Setup. Our real-world setup consists of a Franka FR3 arm, a set of manipulable objects, two drawers, and a two-level rack. We first supervised fine-tune (SFT) \(\pi_{0.5}\) on 10 language-conditioned tasks that span different difficulty levels and horizon lengths, from short-horizon pick-and-place to multi-stage manipulation. Table 1 lists all tasks used for collecting SFT demonstrations. For each task, we collect \(20\) demonstration rollouts using SpaceMouse teleoperation. Each demonstration lasts between \(30\) and \(60\) seconds. The SFT hyperparameters are reported in Table 3. Figure 15 shows the setup of the real world experiments.

RL fine-tuning. We then compare GRPO and GRPO+LfH on the task put the green container into the bowl, where the initial SFT policy achieves zero success. Rewards are provided by a human operator as binary labels, with \(1\) indicating success and \(0\) indicating failure. Each rollout has a time limit of \(30\) seconds, corresponding to \(512\) action steps; each step executes an action chunk of size \(16\) predicted by the VLA model. The GRPO hyperparameters are reported in Table 2.

Results. Figure 5 (b) shows that LfH improves sample efficiency over GRPO throughout real-world training. The initial SFT policy often fails by approaching or grasping the wrong object; for example, we observe that it frequently grasps the tape regardless of the language prompt. LfH turns these failures into useful supervision by relabeling them with prompts that describe the behavior actually executed, such as “pick up the tape”, “lift the tape”, or “move the tape to the bowl”. These relabeled rollouts provide additional training signal for object grounding and related manipulation skills, such as identifying the bowl and moving objects toward it. As a result, LfH can extract useful learning signal from trajectories that would otherwise be treated as failures under the original task prompt.

Figure 15: Realworld experiment setup
Table 1: Tasks used for collecting SFT demonstrations.
Task Prompt
1 put the green container in the bowl
2 put the tape on the top rack
3 put the bowl on the top rack
4 put the green container in front of the camera
5 put the bowl in front of the camera
6 open the top drawer and put the bowl inside
7 open the bottom drawer and put the bowl inside
8 open the top drawer and put the tape inside
9 open the top drawer and put the green container inside
10 put the green container on the bottom rack
Table 2: Real-world GRPO hyperparameters.
Group Hyperparameter Value
Rollout episodes per iteration 32
max episode steps (chunks) 64
poll interval (s) 60
heartbeat interval (s) 60
recompute logprobs False
Environment total num envs (train / eval) 32 / 32
group size 4
max episode steps (robot) 1024
Algorithm loss aggregation token-mean
update epochs 2
rollout epoch 1
KL penalty type low_var_kl
KL coefficient \(\beta\) \(1{\times}10^{-3}\)
entropy bonus 0
clip ratio (low / high) 0.2 / 0.2
clip ratio \(c\) 3.0
Model action dim 10
num action chunks 16
flow-matching steps 10
noise level 0.1
Optimizer actor LR \(2{\times}10^{-6}\)
value LR \(1{\times}10^{-4}\)
Adam \((\beta_1, \beta_2)\) \((0.9,\,0.95)\)
Adam \(\varepsilon\) \(1{\times}10^{-8}\)
weight decay 0.01
grad clip 1.0
Batching / FSDP micro batch size 32
global batch size 512
sharding strategy no_shard (DDP-style)
gradient checkpointing False
Table 3: Realworld SFT hyperparameters.
Group Hyperparameter Value
Model precision bf16
action dim 10
num action chunks 50
num flow steps 10
num image inputs 2 (two cameras)
train expert only True
Batching micro batch size 1
global batch size 256
Optimizer actor LR \(1{\times}10^{-4}\)
value LR \(1.55{\times}10^{-4}\)
Adam \((\beta_1, \beta_2)\) \((0.95,\,0.99)\)
Adam \(\varepsilon\) \(1{\times}10^{-8}\)
weight decay \(1{\times}10^{-6}\)
grad clip 1.0
FSDP / mixed prec. strategy FSDP
sharding strategy no_shard
use orig params True
gradient checkpointing False
wrap policy disabled
AMP enabled True
precision bf16
grad scaler False

9 Additional examples of relabeled trajectories↩︎

a

b

c

d

Figure 16: Qualitative examples of LfH relabeling. For each rollout, we show the original instruction and the VLM-generated hindsight instruction describing what the robot actually did. Meaningful failures are relabeled as alternative successful behaviors, uninformative rollouts are filtered with a “nothing” label, and rare VLM hallucinations remain possible..

References↩︎

[1]
L. P. Kaelbling, M. L. Littman, and A. W. Moore, “Reinforcement learning: A survey,” Journal of Artificial Intelligence Research, vol. 4, pp. 237–285, 1996.
[2]
L. Ouyang et al., “Training language models to follow instructions with human feedback,” in Advances in neural information processing systems, 2022, vol. 35, pp. 27730–27744.
[3]
J. Achiam et al., “GPT-4 technical report,” arXiv preprint arXiv:2303.08774, 2023.
[4]
K. Black et al., \(\pi_0\): A vision-language-action flow model for general robot control,” arXiv preprint arXiv:2410.24164, 2024.
[5]
M. Bellemare, S. Srinivasan, G. Ostrovski, T. Schaul, D. Saxton, and R. Munos, “Unifying count-based exploration and intrinsic motivation,” Advances in neural information processing systems, vol. 29, 2016.
[6]
D. Pathak, P. Agrawal, A. A. Efros, and T. Darrell, “Curiosity-driven exploration by self-supervised prediction,” in International conference on machine learning, 2017, pp. 2778–2787.
[7]
M. Andrychowicz et al., “Hindsight experience replay,” in Advances in neural information processing systems, 2017.
[8]
P. Rauber, A. Ummadisingu, F. Mutz, and J. Schmidhuber, “Hindsight policy gradients,” arXiv preprint arXiv:1711.06006, 2017.
[9]
D. Pathak et al., “Zero-shot visual imitation,” in Proceedings of the IEEE conference on computer vision and pattern recognition workshops, 2018, pp. 2050–2053.
[10]
B. Eysenbach, T. Zhang, R. Salakhutdinov, and S. Levine, “Contrastive learning as goal-conditioned reinforcement learning,” in Advances in neural information processing systems, 2022.
[11]
H. Sahni, T. Buckley, P. Abbeel, and I. Kuzovkin, “Visual hindsight experience replay,” Proc. of NeurIPS, 2019.
[12]
A. Liang et al., “Robometer: Scaling general-purpose robotic reward models via trajectory comparisons,” arXiv preprint arXiv:2603.02115, 2026.
[13]
S. Bai et al., “Qwen2.5-VL technical report,” arXiv preprint arXiv:2502.13923, 2025.
[14]
X. Zhou et al., “LIBERO-PRO: Towards robust and fair evaluation of vision-language-action models beyond memorization,” [arXiv preprint arXiv:2510.03827], 2025.
[15]
Z. Zhou, P. Atreya, A. Lee, H. Walke, O. Mees, and S. Levine, “Autonomous improvement of instruction following skills via foundation models.” 2024, [Online]. Available: https://arxiv.org/abs/2407.20635.
[16]
Z. Zhang, Z. Wang, Y. Du, and F. Fang, “Learning instruction-following policies through open-ended instruction relabeling with large language models,” arXiv preprint arXiv:2506.20061, 2025.
[17]
A. Y. Ng, D. Harada, and S. J. Russell, “Policy invariance under reward transformations: Theory and application to reward shaping,” in Proceedings of the sixteenth international conference on machine learning, 1999, pp. 278–287.
[18]
Y. J. Ma, S. Sodhani, D. Jayaraman, O. Bastani, V. Kumar, and A. Zhang, “Vip: Towards universal visual reward and representation via value-implicit pre-training,” arXiv preprint arXiv:2210.00030, 2022.
[19]
Y. J. Ma, V. Kumar, A. Zhang, O. Bastani, and D. Jayaraman, “Liv: Language-image representations and rewards for robotic control,” in International conference on machine learning, 2023, pp. 23301–23320.
[20]
Y. J. Ma et al., “Vision language models are in-context value learners,” in The thirteenth international conference on learning representations, 2025.
[21]
S. Zhai et al., “A vision-language-action-critic model for robotic real-world reinforcement learning,” arXiv preprint arXiv:2509.15937, 2025.
[22]
Y. Wang et al., “Rl-vlm-f: Reinforcement learning from vision language foundation model feedback,” arXiv preprint arXiv:2402.03681, 2024.
[23]
T. Xiao et al., “Robotic skill acquisition via instruction augmentation with vision-language models,” in Robotics: Science and systems, 2023, [Online]. Available: https://arxiv.org/abs/2211.11736.
[24]
J. Zhang, K. Pertsch, J. Zhang, and J. J. Lim, “Sprint: Scalable policy pre-training via language instruction relabeling,” in 2024 IEEE international conference on robotics and automation (ICRA), 2024, pp. 9168–9175.
[25]
C. Glossop, W. Chen, A. Bhorkar, D. Shah, and S. Levine, “Cast: Counterfactual labels improve instruction following in vision-language-action models,” arXiv preprint arXiv:2508.13446, 2025.
[26]
S. Yang et al., “Instructvla: Vision-language-action instruction tuning from understanding to manipulation,” arXiv preprint arXiv:2507.17520, 2025.
[27]
Y. Wu, A. Li, T. Hermans, F. Ramos, A. Bajcsy, and C. PÊrez-D’Arpino, “Do what you say: Steering vision-language-action models via runtime reasoning-action alignment verification,” 2026 IEEE International Conference on Robotics & Automation (ICRA), 2026.
[28]
M. J. Kim et al., “Openvla: An open-source vision-language-action model,” arXiv preprint arXiv:2406.09246, 2024.
[29]
Z. Shao et al., “DeepSeekMath: Pushing the limits of mathematical reasoning in open language models,” arXiv preprint arXiv:2402.03300, 2024.
[30]
K. Chen et al., \(\pi_{\texttt{RL}}\): Online RL fine-tuning for flow-based vision-language-action models.” 2025, doi: 10.48550/arXiv.2510.25889.
[31]
B. Liu et al., “LIBERO: Benchmarking knowledge transfer for lifelong robot learning,” in Advances in neural information processing systems, 2023, vol. 36.
[32]
J. Bjorck et al., “Gr00t n1: An open foundation model for generalist humanoid robots,” arXiv preprint arXiv:2503.14734, 2025.
[33]
Y. LeCun, “A path towards autonomous machine intelligence,” OpenReview, 2022.

  1. https://rlinf.readthedocs.io/↩︎