alphaFeedback

WorksComputer Science

VaDar: Implicit Goal Conditioning via Value Disaggregation

AI contributions
Not recorded for this paper
View details
Ancillary data
1 code link
View

Abstract

Long-horizon tasks are important for modern AI and robotics, yet pose a challenge for reinforcement learning. Historically, a popular approach has been to decompose a task into subgoals, and learn a subgoal-conditioned policy to execute each subtask in sequence. We introduce an alternative called Value Disaggregation (VaDar), which trains a single subgoal-independent policy to maximize a linear combination of per-subgoal critic values. Because subgoal information is needed only at training-time, VaDar removes the need for possibly costly or cumbersome subgoal-selection during policy deployment. Moreover, through careful experiments, we show that VaDar (1) is always competitive with, and frequently outperforms, natural baselines like goal-conditioning and (2) succeeds on tasks with non-sequential goal structure and multiple optimization objectives. In particular, VaDar is the first algorithm to solve challenging, long-horizon tasks in the RoboCasa suite, even when starting from a policy pretrained from behavior cloning with near-zero initial task success. Taken together, VaDar's successes suggest that subgoal information benefits actor-critic reinforcement learning by enhancing the efficacy of critic learning, whereas policy goal-conditioning is often unnecessary.

Introduction

Learning to complete long-horizon, multi-step tasks is a fundamental challenge across a number of core AI domains, from robotics [26], to language modeling [67], to chemical reaction design [39]. A popular approach, often referred to as hierarchical goal conditioning (HGC) in reinforcement learning literature, decomposes these complex, long-horizon tasks into a set of subgoals, each of which can be executed in sequence by a dedicated learned, goal-conditioned policy ([26, 45, 34, 42]). However, attempting to learn per-subtask policies introduces a new set of challenges. Myopic completion of a given sub-task might compromise performance on future sub-tasks. Moreover, explicit decompositions often require sub-tasks to be sequential in nature, and necessitate a dedicated sub-task planner to decide the current subtask to be attempted during policy deployment.

In this paper, we propose an alternative to HGC that we call Value-Disaggregation (VADAR), a framework for sample-efficient reinforcement learning on long-horizon multi-step tasks. We focus on continuous control robotic manipulation tasks that can be decomposed into sub-tasks g1,…,g∣G∣g^1,\ldots,g^{|G|}. These subgoal evaluations are rapidly becoming highly reliable with today’s VLMs ([20]). While HGC learns policies π(s,g)\pi(s,g) that take sub-goals gg as explicit input, VADAR instead learns a sub-goal agnostic policy π(s)\pi(s), with a critic that estimates the values of all sub-tasks simultaneously. The policy is then trained to maximize a linear combination of critic values, optimizing for an aggregate performance on current-sub-goals as well as future tasks. We stress that VADAR is a generic algorithmic template, being compatible with any preferred choice of critic learning algorithm or policy extraction method; we depict VADAR in Figure 1.

VADAR overview showing hierarchical goal conditioning, single-goal supervision, and implicit Bellman backup through multi-goal supervision

Figure 1. VADAR uses subgoals during training but requires none at deployment. (a) HGC conditions both policy and critic on the current subgoal; iHGC removes policy conditioning. VADAR adds all-subgoal critic supervision and future-aware policy extraction. (b) With single-goal supervision, each transition supplies a TD target for only the current subgoal. (c) VADAR supervises all subgoals on each transition, updating a shared critic representation. We hypothesize that this enables an implicit Bellman backup that improves final-goal value estimates.

VADAR removes the need for subgoal-evaluation at inference by design. Thus, compute-intensive VLM calls used to delineate subgoals, need only be made at training time, and not during policy rollouts. Moreover, VADAR is flexible across both task structure and reward constraints, enabling activations both to safety-constrained reinforcement learning (Appendix § 19.§ 1), and to non-sequential task orders which could admit a directed acyclic graph (DAG) structure. (Appendix § 15).

Across a variety of benchmarks for continuous control and policy extraction methods, we find that VADAR, (1) Consistently outperforms HGC (§ 4), hindsight experience replay (HER) ([4]), and dense preference-based rewards ([10, 37]); (2) Enables the first non-VLA BC-to-online RL method to converge on the evaluated long-horizon ROBOCASA365 tasks ([46]) under fixed initial-state conditions; (3) Enables multi-objective reinforcement learning by incorporating safety constraints (Appendix § 19.§ 1), and enables test-time steering (§ 6.3) by combining critic learning with best-of-NN sampling ([36]); and (4) Works seamlessly with VLM supervision (Appendix § 17) and reaches convergence when goals are specified by reasoning Vision-Language Models (VLMs). This allows sample efficient robot learning from high-dimensional visual observations.

To understand these gains, our ablations examine dense subgoal supervision and future-aware policy extraction (Section 6.2). We find that future-aware policy extraction reduces myopic behavior, while intermediate sub-goal supervision improves learning even when policy extraction uses only final-subgoal values. This suggests that subgoals can support long-horizon RL through critic learning without explicit policy goal conditioning. Taken together, our findings invert the traditional perspectives on overcoming the challenges of long horizon, valued-based reinforcement learning. In short, we find: Rather than myopically conditioning policies via decomposed long-horizon sub-goals, predicting temporally rich subgoal successes can provide critics with dense supervision to enhance the accuracy of long-reaching measurements of full-task success.

Related Work

Long-horizon reinforcement learning via temporal abstraction. Long-horizon reinforcement learning is commonly addressed by introducing temporal abstraction: instead of reasoning over primitive actions, the agent reasons over extended behaviors, skills, or subgoals. Classical formulations include the options framework, where temporally extended actions are specified by initiation sets, intra-option policies, and termination conditions ([62, 5]); hierarchical abstract machines ([51]); feudal reinforcement learning ([17, 65]); and MAXQ value decomposition ([18]). Subsequent work studies how such skills and hierarchies can be discovered from data ([22, 21, 11, 53]). While these methods differ in how subtasks are specified or learned, they use subgoal structure to organize policy execution: the learned policy must either act through a prescribed hierarchy of subtasks([18, 51]) or utilize temporally extended controllers ([5, 22]). In contrast, VADAR assumes that subgoals are part of the MDP design, but does not learn or execute a hierarchy over them; subgoals are used only to construct training-time value targets for a single flat policy.

Goal-conditioned reinforcement learning. Goal condition RL provide a natural design for conditioning on desired outcomes/sub-goals. Instead of learning a separate controller for each task, goal-conditioned methods learn policies or value functions indexed by a goal, enabling transfer across related goal-reaching problems. Classical dynamic-goal learning first studied transfer across changing goal-reaching tasks ([30]). This idea was later extended through universal value function approximators, which learn goal-indexed value functions with function approximation ([57]), and hindsight relabeling, which reuses failed rollouts as successful examples for alternative achieved goals ([4]). However, conditioning a flat policy on the final task goal does not by itself resolve long-horizon credit assignment. For this reason, hierarchical goal-conditioned RL (HGC) uses goal conditioning as a local control primitive: a high-level policy, planner, or task specification provides intermediate subgoals, and a low-level goal-conditioned policy learns to reach them ([42, 34, 27, 43]). This reduces the effective horizon, but introduce additional dependencies: they often assume a sequential subgoal structure, treat only one subgoal as active at a time, and require the deployed policy to be supplied with the correct current goal. These issues are costly and difficult when subgoal progress and completion must be evaluated by expert rules or large vision-language models during inference. VADAR avoids this test-time dependency by learning a subgoal-agnostic policy; goal information enters only through the critic during training.

Value decomposition and reward-factorized critics. MAXQ decomposes the value of a hierarchical policy into child-task values and completion functions, reflecting the procedural structure of a subroutine hierarchy ([18]). Successor representations and successor features instead factor value prediction into predictive components and reward-dependent weights ([16, 7]). VADAR follows this latter spirit of reward-factorized value estimation, but uses subgoal-completion signals to learn a vector-valued critic in an offline-to-online actor-critic pipeline. Closer to robotics, task-and-motion planning (TAMP) couples high-level symbolic task planners with low-level continuous motion planners over a fixed subtask vocabulary ([31, 25]). VADAR shares the assumption that the task admits a subgoal-level decomposition, but does not plan over symbolic actions or commit to a fixed sequencing at execution time: the decomposition enters only through the critic, and the deployed policy is a single flat, subgoal-agnostic controller.

Representation learning for value-based RL. Recent work has shown that auxiliary prediction objectives can improve value learning by shaping the representations used by the critic. MR.Q ([24]) learns state and state-action representations that predict rewards, termination, and future latent states, using model-based objectives to improve a model-free actor–critic without planning through the learned dynamics. DR.O ([40]) builds on this approach by encouraging the learned representations to retain information about future states through a mutual-information objective, together with a recency-aware replay scheme. VADAR is complementary to these approaches: rather than introducing an auxiliary dynamics or representation-learning objective, it obtains additional supervision directly from the RL problem by fitting subgoal value functions at multiple temporal resolutions. Thus, the shared critic representation is trained simultaneously by intermediate and long-horizon TD targets, which we find can improve final-task value estimation even when the intermediate values are not used for policy extraction.

Preliminaries

MDPs. We formulate our problem as a Markov Decision Process (MDP) M:=(S,A,P0,P,R,γ)M := (S, A, P_0, P, R, \gamma), with states s∈Ss \in S, actions a∈Aa \in A, initial state distribution P0P_0, transition probabilities PP, reward RR, and discount factor γ∈(0,1)\gamma\in(0,1). At each timestep tt, the agent (e.g., robot) observes the state st∈Ss_t \in S, takes an action at∼π(at∣st)∈Aa_t \sim\pi(a_t \mid s_t) \in A, transitions to the next state st+1s_{t+1} according to st+1∼P(st+1∣st,at)s_{t+1} \sim P(s_{t+1} \mid s_t, a_t) while receiving the reward R(st,at)R(s_t, a_t). For the MDP MENVM_{\mathrm{ENV}}, we let Eπ,M\mathbb{E}^{\pi,M} (resp. Pπ,M\mathbb{P}^{\pi,M}) denote the expectation (resp. probability distribution) over trajectories (s0,a0,…,sT,aT)(s_0, a_0, \ldots, s_T, a_T) with length T+1T+1, with initial state distribution s0∼P0s_0 \sim P_0 and transition operator PP. The object of reinforcement learning (RL) is to train a policy to optimize the cumulative discounted return: JM(πθ)=Eπ[∑t≥0γtR(st,at)]J^M(\pi_\theta) = \mathbb{E}^{\pi}\left[\sum_{t \ge0} \gamma^t R(s_t, a_t)\right]. We also recall the Q-function Qπ,M(s,a):=Eπ[∑t≥0γtR(st,at)∣(st,at)=(s,a)]Q^{\pi,M}(s,a) := \mathbb{E}^{\pi}\left[\sum_{t \ge0} \gamma^t R(s_t,a_t) \mid(s_t,a_t) = (s,a)\right] and value function Vπ,M(s):=Ea∼π(⋅∣s)[Qπ(s,a)]V^{\pi,M}(s) := \mathbb{E}_{a \sim\pi(\cdot\mid s)}[Q^{\pi}(s,a)]. When MM is clear from context, we can omit it from the subscript.

Value-based Reinforcement Learning. Value-based RL methods ([60]) simultaneously learn a policy πθ\pi_\theta and associated QϕQ_\phi-function, maintaining (1) a replay buffer B={(s,a,s′,r,done)}\mathcal{B} = \{(s,a,s',r,\text{done})\} of past states ss, actions aa, successor states s′∼P(s,a)s' \sim P(s,a), rewards rr, and done indicators done; (2) a policy πθ\pi_\theta, critic network Qϕ:S×A→RQ_\phi: S \times A \to\mathbb{R} trained from this buffer. The parameters of QϕQ_\phi are updated to approximate QπθQ^{\pi_\theta}, and πθ(s)\pi_\theta(s) is updated to maximize critic values. Specifically, updates are sampled (s,a,s′,r)∼B(s,a,s',r) \sim\mathcal{B}, and new actions a′∼πθ(s′)a' \sim\pi_\theta(s'). Then,

  1. Critic parameters ϕ\phi are updated via gradient descent on a given critic loss

Lcritic(Qϕ∣s,a,ytarget)=(Qϕ(s,a)−ytarget)2+Lreg(s′,a′,πθ)(1)L_{\mathrm{critic}}(Q_\phi\mid s,a,y^{\mathrm{target}}) = (Q_\phi(s,a) - y^{\mathrm{target}})^2 + L_{\mathrm{reg}}(s',a',\pi_\theta) \tag*{(1)}

which measures the squared Bellman residual to the future target ytarget=r+γQϕ(s′,a′)y^{\mathrm{target}} = r + \gamma Q_{\phi}(s', a'), and which may also incorporate some additional regularization Lreg(⋅)L_{\mathrm{reg}}(\cdot) depending on s′s', a′a', πθ\pi_{\theta} (omitted as arguments in the critic loss for simplicity).

  1. Policy parameters θ\theta are updated via gradient descent on a given policy loss Lpolicy(πθ∣s,Q)L_{\mathrm{policy}}(\pi_{\theta} \mid s, Q) mapping from policies πθ:S→Δ(A)\pi_{\theta}: \mathcal{S} \to\Delta(\mathcal{A}), states s∈Ss \in\mathcal{S}, and QQ to R\mathbb{R}.

Representative Algorithms: SAC, FQL and OGPO. We instantiate VADAR with Soft Actor-Critic (SAC) (Haarnoja et al., [29]), Flow Q-Learning (FQL) (Park et al., [50]), and off-policy generative policy optimization (OGPO) (Patil et al., [52]). SAC uses a maximum-entropy policy objective; FQL and OGPO support expressive generative control policies. These backbones allow us to test value disaggregation across different policy-extraction methods. Their policy objectives and algorithm-specific critic updates are detailed in Appendix § 11.

Sparse-Reward RL. In many tasks, such as those encountered in robot manipulation, defining reward that densely measure progress towards goal completion is challenging ([3, 32]). Thus, we consider a sparse-reward defined via the done indicator done∈{0,1}\mathrm{done} \in\{0, 1\} indicating task completion. Standard practice is to set r←rsparse(done)r \leftarrow r_{\mathrm{sparse}}(\mathrm{done}), where

rsparse(done)=−1⋅I{done=0}+rdoneI{done=1},(2)r_{\mathrm{sparse}}(\mathrm{done}) = -1 \cdot\mathbb{I}\{\mathrm{done} = 0\} + r_{\mathrm{done}}\mathbb{I}\{\mathrm{done} = 1\}, \tag*{(2)}

and where rdoner_{\mathrm{done}} is a small positive constant (e.g. rdone=5r_{\mathrm{done}} = 5). In words, rsparser_{\mathrm{sparse}} penalizes each time-step until the task is not completed, and provides a small positive bonus for task success.

Offline-to-Online RL. In many robotics settings, environment interaction is expensive and a fixed dataset of demonstrations Doff\mathcal{D}_{\mathrm{off}} is available. We follow the now-standard offline-to-online recipe (Nair et al., [44]): pretrain a behavior-cloning policy πBC\pi_{\mathrm{BC}} on Doff\mathcal{D}_{\mathrm{off}}, then initialize the online policy from πBC\pi_{\mathrm{BC}} and fine-tune online, optionally mixing Doff\mathcal{D}_{\mathrm{off}} into the off-policy updates at a fixed ratio (Ball et al., [6]). All algorithms in our experiments share the same BC-pretraining and warm-start protocol, so comparisons isolate the online fine-tuning method.

Dense Preference Reward. An alternative approach to long-horizon RL is to learn dense preference-based rewards. We describe the formulation and implementation in Appendix § 13 and compare against this approach in § 6 and Appendix § 14.§ 2.

Hierarchical Goal Conditioning and Its Limitations

Algorithm 1 Hierarchical Goal Conditioning RL (HGC-RL).

1: Parameters: discount factor γ\gamma; learning rate η\eta 2: Initialize critic networks Qϕ:S×A×G→RQ_{\phi}: \mathcal{S} \times\mathcal{A} \times G \to\mathbb{R}, policy πθ:S×G→Δ(A)\pi_{\theta}: \mathcal{S} \times G \to\Delta(\mathcal{A}), buffer B←∅\mathcal{B} \leftarrow\varnothing 3: for episode =1,2,…= 1, 2, \ldots do 4:     while episode not done do 5:         at←πθ(st,gt)a_t \leftarrow\pi_{\theta}(s_t, g_t), transition st+1∼P(st,at)s_{t+1} \sim P(s_t, a_t)         ▹gt=gcurr:=min⁡{g:doneg=0}\triangleright g_t = g^{\mathrm{curr}} := \min\{g : \mathrm{done}_g = 0\} 6:         B←B∪{(st,at,st+1,rt,gt)}\mathcal{B} \leftarrow\mathcal{B} \cup\{(s_t, a_t, s_{t+1}, r_t, g_t)\}, rt←rsparse(donegt)r_t \leftarrow r_{\mathrm{sparse}}(\mathrm{done}_{g_t}) 7:     for number of updates do 8:         Sample (s,a,s′,r,g)∼B(s, a, s', r, g) \sim\mathcal{B}, a′∼πθ(s′)a' \sim\pi_{\theta}(s') 9:         ygtarget←rg+γQϕ(s′,a′,g)y_g^{\mathrm{target}} \leftarrow r_g + \gamma Q_{\phi}(s', a', g)         ▹\triangleright Target for current goal 10:         ϕ←ϕ−η∇ϕLcritic(Qϕ(⋅,⋅,g)∣s,a,ygtarget)\phi\leftarrow\phi- \eta\nabla_{\phi} L_{\mathrm{critic}}(Q_{\phi}(\cdot,\cdot,g) \mid s, a, y_g^{\mathrm{target}}) 11:         θ←θ−η∇θLpolicy(πθ(⋅,g)∣s,a,Qϕ(⋅,⋅;g))\theta\leftarrow\theta- \eta\nabla_{\theta} L_{\mathrm{policy}}(\pi_{\theta}(\cdot,g) \mid s, a, Q_{\phi}(\cdot,\cdot;g))

Long-horizon reinforcement learning tasks require many time steps to complete. Sparse-rewards make credit assignment - identifying which actions contribute to task success/failure - challenging. Thus, prior work introduces hierarchical goal conditioning (HGC)2, which decomposes the problem into a sequence of subgoals G={g}g∈GG = \{g\}_{g \in G}. Traditional goal-conditioned RL assumes that the completion of individual subgoals, what we call subgoal-dones doneg(s):=I{g is completed at state s}\mathrm{done}_g(s) := \mathbb{I}\{g\text{ is completed at state }s\}, can be evaluated at both training time and during evaluation. Traditionally, doneg\mathrm{done}_g have been hand-specified, but modern VLMs can now evaluate doneg\mathrm{done}_g's specified by natural language queries (Chen et al., [12]). For simplicity, in the body of this paper we assume these goals satisfy a linear order with g1≺g2⋯≺g∣G∣∈Gg^1 \prec g^2 \cdots\prec g^{|G|} \in G, where g∣G∣g^{|G|} denotes final task completion and g1g^1 is the first goal (Appendix § 15 easily removes this sequentiality). Under this sequentiality condition, we define the current goal gcurr=min⁡{g:doneg=0}g^{\mathrm{curr}} = \min\{g : \mathrm{done}_g = 0\}, where min is taken with respect to the order ≺\prec. Algorithm 1 depicts how value-based RL methods can be modified to incorporate HGC. These modifications of standard value-based RL, depicted in blue, are: policies and critics take the current goal gt=gcurrg_t = g^{\mathrm{curr}} as an argument (Line 2), so that in particular, the policy needs gtg_t to select actions (Line 5). This necessitates that gtg_t is tracked in the replay buffer (Line 6), is used to determine the critic target (Line 9), and features in both the policy and critic loss (Lines 10, 11). However, we identify 3 limitations of HGC, outlined below.

Limitation 1: Accurate goals must be supplied to the policy at test-time, which is prohibitive if sub-goal completions requires expert engineering or larger models to evaluate. Limitation 2: Optimizing for the current goal can be overly myopic if future actions depend subtly on how the current goal is completed, such as in robotic pick and place tasks. Limitation 3: HGC often requires goals to have a sequential structure, with only one active goal at a time.

We examine these limitations in Appendix § 12. As a point of comparison, we introduce implicit hierarchical goal conditioning (iHGC), which retains goal-conditioned critics but uses a subgoal-independent policy π(a∣s)\pi(a \mid s), removing the need for test-time goal information (Limitation 1). Although iHGC can outperform traditional HGC as shown in Figure 10, both optimize the policy against the current-subgoal critic alone and remain susceptible to myopic behavior (Limitation 2). This motivates VADAR, which incorporates future-subgoal values during policy optimization.

Robomimic Toolhang (FQL) success rate comparison

Figure 10. TOOL-HANG with FQL: iHGC learns more smoothly and achieves higher final success than HGC.

The Value Disaggregation (red!50!black VaDar) Framework

VADAR training and inference diagram

Figure 2. VADAR training and inference. Left: Subgoal completion indicators weight the per-subgoal critic values, whose aggregate guides updates to a subgoal-independent policy. Right: The policy selects actions directly from the state, without subgoal information.

We now introduce Value Disaggregation (VADAR, Algorithm 2), a framework for subgoal-driven long-horizon RL that addresses the limitations of goal conditioning described in § 4. VADAR combines three key insights: (1) Like implicit hierarchical goal conditioning (iHGC), goal-conditioned critics can train a single, subgoal-independent policy. (2) Leveraging weighted combinations of critics can balance current and future goal completion, thereby reducing the myopic overexploitation described in Limitation 2. Finally, (3) The extraction from multiple critics is naturally compatible with general goal topologies, addressing Limitation 3 by allowing more than one subgoal to be active at any given time.

Algorithm 2 Value Disaggregation (VADAR)

1:  Parameters: discount factor $\gamma$; learning rate $\eta$; TD clipping parameter $c_{\mathrm{clip}}$.
2:  Initialize goal-conditioned critic $Q_{\phi}: S \times A \times G \to \mathbb{R}$; policy $\pi_{\theta}: S \to \Delta(A)$; replay buffer $\mathcal{B} \leftarrow \varnothing$.
3:  for episode $e = 1, 2, \ldots$ do
4:      while episode not done do
5:          $a_t \leftarrow \pi_{\bar{\theta}}(s_t)$, transition $s_{t+1} \sim \mathcal{P}(s_t,a_t)$
6:          Evaluate subtask completion vector: $d_t = \{\operatorname{done}_g(s_{t+1})\}_{g \in G} \in \{0,1\}^{G}$
7:          $\mathcal{B} \leftarrow \mathcal{B} \cup \{(s_t,a_t,s_{t+1},\mathbf{r}_t,\mathbf{d}_t)\}$, where $r_t[g] \leftarrow r_{\mathrm{sparse}}(d_t[g])$
8:          for number of update steps do
9:              Sample $(s,a,s',r,d) \sim \mathcal{B}$, $a' \sim \pi_{\theta}(s')$
10:             $y^{\mathrm{target}}[g] = \min\{r[g] + \gamma Q_{\phi}(s',a',g), c_{\mathrm{clip}}\cdot r_{\mathrm{done}}\}$, for all $g \in G$
11:             $\phi \leftarrow \phi - \eta \nabla_{\phi}\sum_{g \in G}\mathcal{L}_{\mathrm{critic}}(Q_{\phi}(\cdot,\cdot,g), y^{\mathrm{target}}[g])$
12:             $Q_{\mathrm{agg}}(\cdot,\cdot) \leftarrow \sum_g w_g Q_{\phi}(\cdot,\cdot,g)$, where $w \in \mathbb{R}^{G} \leftarrow \operatorname{GetWeighting}(d)$
13:             $\theta \leftarrow \theta - \eta \nabla_{\theta}\mathcal{L}_{\mathrm{policy}}(\pi_{\theta} \mid s,a,Q_{\mathrm{agg}})$

Conveniently, VADAR requires minimal modification to standard off-policy RL pipelines, wherein only the critic architecture (conditioned on subgoals) and the corresponding aggregation strategies change. We now walk through Algorithm 2 in detail, highlighting the components unique to VaDAR (shown in color in the pseudocode and below). We give a detailed practitioners guide to using VaDAR in Appendix A. In what follows, we use brackets [g] to denote goal-wise indexing.

Policy update via value weighting. The key innovation in VADAR is optimizing for weighted contributions of different critics during policy extraction, which addresses the myopic behavior described in § 4 and illustrated in Appendix § 12. At each environment step, VADAR evaluates a vector of subgoal completion indicators dt={done⁡g(st+1)}g∈G∈{0,1}G\mathbf{d}_t = \{\operatorname{done}_g(s_{t+1})\}_{g \in G} \in\{0,1\}^{G}. It uses these to computing a weighting GETWEIGHTING (Line 12) for policy extraction:

Qagg(s,a)=∑g∈Gw[g]⋅Qϕ(s,a,g),w←GETWEIGHTING⁡(d).(3)Q_{\mathrm{agg}}(s,a) = \sum_{g \in G} w[g] \cdot Q_{\phi}(s,a,g), \qquad\mathbf{w} \leftarrow\operatorname{GETWEIGHTING}(\mathbf{d}). \tag*{(3)}

The policy is then updated to maximize QaggQ_{\mathrm{agg}} via the policy loss Lpolicy(πθ∣s,Qagg)L_{\mathrm{policy}}(\pi_{\theta} \mid s,Q_{\mathrm{agg}}). Because the policy πθ:S→Δ(A)\pi_{\theta}: S \to\Delta(A) is never conditioned on gg (subgoal agnostic), subgoals are not needed at test time. A good weighting function provides strong training signal for completing current goals while improving behavior for future goals.

Weighting choices for sequential goals. As in § 3, first assume that goals are ordered as 1,…,∣G∣1,\ldots,|G|, with goal gig^i serving as a prerequisite for goal gi+1g^{i+1}, denoted gi≺gi+1g^i \prec g^{i+1}. (see Appendix § 15 for the general case. Given a vector of completion indicators d\mathbf{d}, recall that gcurr=inf⁡{g:dg=0}g^{\mathrm{curr}} = \inf\{g : d_g = 0\} denotes the current unfinished goal.

We also define the next goal as gnext:=min⁡{g+,g∣G∣}g^{\mathrm{next}} := \min\{g^+,g^{|G|}\}, where g+:=inf⁡{g′:g′≻gcurr}g^+ := \inf\{g' : g' \succ g^{\mathrm{curr}}\}; again, min/inf is taken with respect to the goal order ≺\prec, and ⪰\succeq in Table 1 denotes the negation of ≺\prec. Table 1 summarizes four choices for the weighting function along two axes. The first axis distinguishes myopic weightings, which optimize only a single goal, from future-aware weightings, which include all relevant future goals. The second axis distinguishes current weightings, which include the current goal, from next weightings, which only optimize for goals after the current goal. These choices are illustrated in Figure 3. Detailed experiments can be found in Section 6.2.

Subgoal weighting used by GETWEIGHTING

Figure 3. Subgoal weighting used by GETWEIGHTING.

All-Subgoal Critic Learning Importantly, the weighted extraction Eq. (3) requires that the critic values Q(s,a,g)Q(s,a,g) for each goal are well-estimated. Therefore, we design an all-subgoal critic update. Recall that, at each environment step, VADAR evaluates a vector of subgoal completion indicators dt\mathbf{d}_t, and computes a corresponding reward vector rt=rsparse(dt)\mathbf{r}_t = r_{\mathrm{sparse}}(\mathbf{d}_t) entrywise via the sparse reward function Eq. (3.2), Line 6). This converts a sparse task reward into ∣G∣|G| per-subgoal rewards (Line 10), each of which provides signal as soon as its corresponding subgoal is completed. The tuple (st,at,st+1,rt,dt)(s_t,a_t,s_{t+1},\mathbf{r}_t,\mathbf{d}_t) is stored in the replay buffer. Given a sampled transition (s,a,s′,r,d)(s,a,s',\mathbf{r},\mathbf{d}) from the buffer and a new action a′∼πθ(s′)a' \sim\pi_{\theta}(s'), the vector of TD targets is computed entrywise:

ytarget[g]=min⁡{r[g]+γQϕ(s′,a′,g), cclip⋅rdone[g]}∀g∈G,(4)\mathbf{y}^{\mathrm{target}}[g] = \min\{r[g] + \gamma Q_{\phi}(s',a',g),\ c_{\mathrm{clip}}\cdot r_{\mathrm{done}}[g]\} \qquad\forall g \in G, \tag*{(4)}

where the min is applied entrywise. The critic is updated by a gradient update on the critic loss across all subgoals: ∑g∈GLcritic(Qϕ(s,a,g)∣s,a,ygtarget)\sum_{g \in G} L_{\mathrm{critic}}(Q_{\phi}(s,a,g) \mid s,a,\mathbf{y}^{\mathrm{target}}_g).

Why clipping? The critic update involves a “clipping” of the maximum critic value. This is because, after a subgoal is completed, the agent keeps collecting rdoner_{\mathrm{done}} at every step, so without the clip the Q-value would accumulate to rdone/(1−γ)r_{\mathrm{done}}/(1-\gamma). The clip caps the target at cclip⋅rdonec_{\mathrm{clip}}\cdot r_{\mathrm{done}} (Line 10), capping critic values for completed subgoals.

VADAR: Key Design Decisions

  • Subgoal-agnostic policy. Subgoal information enters only through the critic at training time; deployed policy πθ(a∣s)\pi_\theta(a \mid s) needs no goal selection/conditioning at inference.

  • Weighted aggregation for policy extraction. Per-subgoal Q-values are aggregated into a single scalar signal QaggQ_{\mathrm{agg}}; the next-future (VADAR-NF) weighting is our default method which balances current and future subgoal completion.

  • Reward clipping at cclip⋅rdonec_{\mathrm{clip}} \cdot r_{\mathrm{done}}. Removes the need for critic loss gating on completed subgoals.

VADAR benefits from both dense critic supervision and weighted policy extraction (Section 6.2 and Appendix § 23.§ 1). We provide further details on non-sequential tasks in Appendix § 15, multi-objective optimization in Appendix § 19, and alternative critic parameterizations in Appendix § 10.§ 1. Finally, Appendix § 23 shows that fixed-weight value aggregation is theoretically (but only theoretically) equivalent to reward shaping but we later show (Appendix § 23.§ 1) that this equivalence does not explain VADAR’s empirical success.

Experiments

We evaluate VADAR on a suite of long-horizon manipulation benchmarks. Each method is pretrained via behavior cloning on offline data, stopped early at a low success rate so all methods begin fine-tuning from a comparable starting point, and then fine-tuned online for a fixed number of environment steps. Per-task descriptions and hyperparameters are detailed in Appendix § 16.

Baselines. We study VADAR with two different weighting schemes (Table 1) - VADAR-NF, using next-future (plotted in a solid line) and VADAR-CF, which use current-future (plotted in a dotted line). We compare against 3 alternatives, each using the same base-algorithm per plot: (1) Hierarchical Goal Conditioning (HGC), where both the policy and critic are conditioned on the current subgoal gg, where gg is set to the next incomplete subgoal at every step (Algorithm 1). (2) Implicit Hierarchical Goal Conditioning (iHGC), which retains the goal conditioned critic of HGC but does not condition the policy on the subgoal. (3) Dense Preference Reward, denoted Dense, where we adopt best-practices for a learning dense reward model via preferences (described in Appendix § 13), where the reward models are pretrained with additional rollouts on a richer dataset of successful and failed runs in order to steelman their performance. (4) The original base-algorithm that using sparse reward with no goal information, denoted as “Sparse”.

Config.Weight w[g]w[g]Interpretation
Current-myopicI[g=gcurr]\mathbb{I}[g = g^{\mathrm{curr}}]Only the current subtask contributes.
Current-futureI[g⪰gcurr]\mathbb{I}[g \succeq g^{\mathrm{curr}}]All incomplete subtasks contribute.
Next-myopicI[g=gnext]\mathbb{I}[g = g^{\mathrm{next}}]Focus on the next subtask, gated by the current subtask.
Next-futureI[g⪰gnext]\mathbb{I}[g \succeq g^{\mathrm{next}}]Incomplete subtasks with incomplete parents contribute.

Table 1. Value-weighting variants used for policy extraction in VADAR.

Environments. We evaluate on four benchmark suites spanning different task horizons and difficulties. Robocasa365 (Nasiriany et al., [46]) provides our three longest-horizon tasks, while Robomimic (Mandlekar et al., [41]) tests contact-rich, medium-horizon manipulation on Transport and Tool-Hang. We also evaluate on three multi-stage kitchen tasks from Libero (Liu et al., [38]scene-play-single-task-task4 task from Ogbench ({citet}Park et al., 2025a}). Task descriptions, subgoals, and their dependency structures are given in Appendix § 16.

Base-algorithm instantiations. We instantiate all above baselines with three value-based RL algorithms (details given in Appendix § 11). On the most challenging long-horizon environments (Robocasa365), we use OGPO (Patil et al., [52]), a state-of-the-art offline-to-online RL algorithm for flow-based policies. On the medium-horizon Robomimic tasks, we evaluate on both OGPO and the popular, though less performant FQL (Park et al., [50]), demonstrating that the benefits of VADAR apply to both. Because OGPO is considerably more sample efficient that than FQL from Sparse reward when all BC data are used, we also test OGPO in a regime with limited BC data, denoted low data, on Robomimic; details are described in Table 10. Lastly, we demonstrate VADAR’s applicability to non-generative policies with the seminal Soft Actor Critic (SAC) algorithm (Haarnoja et al., [29]) the simpler on OGBench and Libero environments.

Transport (low)Transport (full)Tool-Hang (low)Tool-Hang (full)
dataset.proportion0.11.00.301.0
offline_steps300K300K300K300K
online_steps3M3M3M3M
q_warmup_steps20K20K20K20K
start_training40K40K10K10K
τˉπ\bar{\tau}_{\pi}0.0050.0050.0050.005
bc_coeff0.10.10.10.1
clip_bc_threshold0.300.300.300.30
success buffertruetruetruetrue
subsample_bontruetruetruetrue

Table 10. Low-data BC ablation on RoboMimic. Both columns use the OGPO backbone with VADAR-style critic disaggregation. dataset.proportion controls the fraction of offline demonstrations used for BC pretraining and offline replay; all other hyperparameters remain the same.

VaDAR outperforms natural baselines

Figure 4 shows that VADAR improves various policy-extraction methods, with the largest gains on the most challenging long-horizon tasks. Not only does VADAR benefit a SOTA method like OGPO, but also enhances weaker methods like FQL and SAC with improved critic learning.

Success-rate comparisons across Robocasa, Robomimic, Libero and OGBench

Figure 4. Success rate of VADAR-NF (solid) and VADAR-CF (dotted) against HGC, its implicit variant iHGC, the Dense preference-reward baseline, and the Sparse backbone across ROBOCASA365, ROBOMIMIC, LIBERO and OGBENCH, for our three instantiations OGPO, FQL and SAC.

VADAR supplies useful values for aggressive policy extraction. OGPO aggressively optimizes critics using PPO-style policy extraction on the generative policy. We find that VADAR with OGPO matches or outperforms all baselines, with the largest gains on the most challenging long-horizon tasks (Figure 4). On Robocasa365, VADAR is the only method to achieve convergence on LoadDishwasher, while on PreSoakPan it reaches nearly double the success rate of the strongest baseline, iHGC. Similarly, on ArrangeTea, VADAR outperforms baselines with superior training stability. On ROBOMIMIC with OGPO, we find that VADAR, HGC and iHGC obtain similar performance in both the low- and full-data regimes.

The benefits extend to first-order policy extraction. VADAR also improves learning with FQL and SAC, which update the policy through action gradients of the critic. With FQL, VADAR learns faster on ROBOMIMIC TRANSPORT and TOOL-HANG, demonstrating that its benefits do not require OGPO’s policy update. With SAC, the advantage is most apparent on the longer-horizon LIBERO tasks, whilst performance remains comparable to goal-conditioned baselines on OGBENCH. Together, these comparisons support the use of value disaggregation across different policy-extraction methods. Even where performance is comparable, VADAR retains the practical benefit of requiring no goal selection at inference.

Additional baseline comparisons. In Appendix § 14, we compare to two additional algorithm families, both of which significantly underperform VADAR. First, we test hindsight experience replay (HER) (Andrychowicz et al., [4]) on contact-rich LIBERO manipulation, including variants supplied with the same subgoal information as HGC. Second, we compare against reward learning with access to subgoal information, training separate preference-based reward models for each subgoal and aggregating with next-future weighting. VLM-provided subgoal labels. On ROBOCASA365 PRESOAKPAN, VADAR also learns with VLM-provided subgoal labels, reaching similar final success to oracle supervision with only modest effects on sample efficiency (Appendix § 17.§ 1).

Understanding What Enables red!50!black VaDar's Performance

Future-Aware Policy Extraction Improves Performance To isolate the effect of policy extraction, we compare the current-myopic (VADAR-GM), current-future (VADAR-CF), and next-future (VADAR-NF) weightings described in Table 1. These ablations share all-subgoal critic supervision and goal-agnostic policies. Figure 5 shows that VADAR-GM does not outperform HGC and iHGC, while the future-aware variants improve sample efficiency and are the only methods to converge on LOAD DISHWASHER. Hence, dense critic supervision alone is insufficient when used with myopic policy extraction. Next, we test whether future-aware policy extraction also improves HGC while retaining policy goal conditioning. On ROBOCASA365 PRESOAKPAN with OGPO, we introduce three Steelman variants, namely Steelman 1 using current-myopic, Steelman 2 using current-future, and Steelman 3 using next-future, extraction respectively. Figure 6 shows that Steelman 1 remains close to HGC, while Steelman 2 and Steelman 3 approach VADAR-CF and VADAR-NF. These comparisons show that future-aware extraction benefits both policy formulations on this task, allowing VADAR to retain these benefits without requiring subgoal selection at inference.

Weighting ablation of VADAR

Figure 5. Weighting ablation of VADAR (VADAR-GM, VADAR-CF, VADAR-NF) against HGC and iHGC on ROBOCASA365 with the OGPO backbone and ROBOMIMIC TOOL-HANG with the FQL backbone.

Steelman sweep on Robocasa PreSoakPan with the OGPO backbone

Figure 6. Steelman sweep on ROBOCASA365 PRESOAKPAN with the OGPO backbone.

VariantCritic supervisionPolicy weightingPolicy
HGC (reference)current subgoalcurrent-myopicπ(a∣s,g)\pi(a \mid s,g)
iHGC (§ 4)current subgoalcurrent-myopicπ(a∣s)\pi(a \mid s)
Steelman 1all subgoalscurrent-myopicπ(a∣s,g)\pi(a \mid s,g)
Steelman 2all subgoalscurrent-futureπ(a∣s,g)\pi(a \mid s,g)
Steelman 3all subgoalsnext-futureπ(a∣s,g)\pi(a \mid s,g)
VADAR-GMall subgoalscurrent-myopicπ(a∣s)\pi(a \mid s)
VADAR-CF (ours)all subgoalscurrent-futureπ(a∣s)\pi(a \mid s)
VADAR-NF (ours)all subgoalsnext-futureπ(a∣s)\pi(a \mid s)

Table 2. Variants bridging HGC and VADAR-NF, varying critic supervision, policy weighting (Table 1), and policy goal conditioning.

Dense Subgoal Supervision Improves Critic Learning. To further test whether dense supervision across multiple temporal resolutions improves critic learning, we ask a stricter question: can intermediate subgoal supervision improve the final-task critic even when those intermediate values are never used for policy extraction? We introduce VADARLAST-Q, which trains the same multi-goal critic as VADAR (§ 5) but extracts the policy using only the final-subgoal value Qϕ(s,a,g∣G∣)Q_{\phi}(s,a,g_{\lvert G\rvert}). Thus, VADARLAST-Q uses the same policy objective as the Sparse baseline, while additionally supervising the critic on intermediate subgoals. Figure 7 shows that VADARLAST-Q approaches VADAR-NF on several tasks and substantially outperforms Sparse on the ROBOCASA365 benchmarks. Because the policy objective is unchanged, these gains isolate the effect of denser critic supervision. Full results across all benchmarks and policy-extraction methods are provided in Appendix § 21. This suggests that intermediate subgoal targets improve the critic even when their values are not used directly for policy extraction.

Success-rate curves for VADAR, VADAR-lastQ, and Sparse on four selected tasks

Figure 7. Success rate of VADARLAST-Q compared with VADAR-NF and Sparse on selected tasks. Figure 21 gives the results across all benchmarks and instantiations.

Effect of the policy extraction method. While VADARLAST-Q sometimes matches VADAR-NF, the size of this gap depends on the policy-extraction method. VADARLAST-Q is generally closer to VADAR-NF under OGPO, where the critic-derived signal supervises every denoising step of the generative policy. In FQL, where the critic provides less dense supervision than OGPO, VADARLAST-Q alone is not sufficient, and Next-future weighting provided additional gains.

Takeaway box

Figure 8. Takeaway: Intermediate supervision improves final-task learning. VADARLAST-Q, which uses VADAR for critic supervision, but only supervises on the final subgoal, improves over Sparse on the long-horizon ROBOCASA365 tasks even though intermediate subgoal values never enter the policy objective directly. This supports our hypothesis that dense subgoal supervision improves critic learning. Next-future weighting provides additional gains where final-subgoal extraction alone remains insufficient.

Further findings. We further compare against VADAR-SHAPED, which aggregates subgoal rewards before the Bellman backup and trains a scalar critic. It fails to learn on the harder ROBOCASA365 tasks and is less sample efficient than VADAR-NF elsewhere (Appendix § 23.§ 1). These and the VADARLAST-Q results suggest that reward shaping alone does not explain the gains from all-subgoal critic supervision. Multi-objective optimization. VADAR also applies when reward components represent distinct objectives. As a proof of concept, Appendix § 19.§ 1 shows how to train separate critics for task completion and obstacle avoidance improving performance and training stability over SAC trained on the aggregated rewards with optimal reward weighting.

Test-Time Steering with a Frozen Policy

We hypothesize that VADAR’s success comes from learning a better critic through denser supervision across multiple temporal resolutions. We test this directly by asking: does the learned critic better identify actions that lead to task success? To isolate critic quality from policy learning, we fix a behavior-cloned policy and update only the critic. Following Q-chunking [35], we use best-of-NN steering: a frozen behavior-cloned policy proposes NN candidate action chunks, and the critic selects the highest-valued candidate. We compare VADAR-NF and Sparse critics trained on the same demonstrations and frozen-policy rollouts. For this steering experiment, next-future aggregation uses subgoal-completion indicators at inference; the unconditional policy learned in our main experiments requires no such information.

Best-of-N steering results for Robocasa ArrangeTea, Robocasa PreSoakPan, and Robomimic Toolhang

Figure 8 (PDF p. 10). Best-of-NN steering (N=4N = 4) of a frozen BC policy with VADAR-NF and Sparse critics as critic training progresses. The dashed line marks the unsteered policy’s success rate. We use the BC-pretrained flow policy from OGPO, without online policy updates.

Figure 8 shows that the VADAR critic improves action selection most clearly on ArrangeTea and ToolHang. At the largest critic-training budget with N=4N = 4, success rises from 23% to 55% on ArrangeTea and from 32% to 77% on Tool-Hang, compared with 23% and 44% for the Sparse critic. Gains on PreSoakPan are smaller. These results support the benefit of subgoal supervision for critic learning independently of policy updates. Appendix § 20 provides the full protocol, seed variability, candidate-count sweep, and critic visualizations.

Hypothetical Mechanism: Implicit Bellman-Backup via Dense Critic Supervision

The frozen-policy experiments show that subgoal supervision improves action selection independently of policy updates. Together with the VADARLAST-Q results, they motivate a further question: how can intermediate subgoal targets improve value predictions for the final task? We hypothesize that shared representations allow supervision at shorter horizons to inform value prediction at longer horizons. We refer to this proposed mechanism as an implicit Bellman backup. For a fixed policy, the classical Bellman recursion is

Qπ(s,a)=r(s,a)+γEs′∼P(⋅∣s,a), a′∼π(⋅∣s′)[Qπ(s′,a′)].(5)Q^{\pi}(s,a) = r(s,a) + \gamma\mathbb{E}_{s'\sim\mathcal{P}(\cdot\mid s,a),\,a'\sim\pi(\cdot\mid s')} \left[ Q^{\pi}(s',a') \right]. \tag*{(5)}

With sparse final-task rewards, one-step TD learning propagates the reward signal backward through successive bootstrapped targets. Function approximation can generalize across states, but the final-task reward alone provides little direct supervision for learning representations of intermediate progress.

On the other hand, VADAR supervises Qϕ(s,a,g)Q_{\phi}(s,a,g) for every subgoal g∈Gg \in G at each sampled transition Eq. (4). These targets span multiple temporal resolutions: a state far from task completion may already be close to an intermediate subgoal. We speculate that predicting these nearer outcomes trains useful features before the final-task value is accurate. Furthermore, because the subgoal predictions share network parameters (Figure 9), features learned from intermediate targets are also available to the final-subgoal prediction Qϕ(s,a,g∣G∣)Q_{\phi}(s,a,g^{|G|}). We hypothesize that this transfer helps the critic learn dependencies across stages of the task that would otherwise have to be learned from final-task rewards alone.

Diagram showing all-subgoal TD targets updating shared critic representations

Figure 9. All-subgoal TD targets update shared critic representations.

The VADARLAST-Q comparison provides evidence of this hypothesized mechanism: intermediate targets improve learning even though only the final-subgoal value enters policy extraction (Section 6.2). These comparisons do not establish additional Bellman backups within a forward pass or a guaranteed reduction in the number of TD updates. Appendix § 25 develops a related interpretation for iterative critics, which explicitly refine a vector of value estimates over multiple integration steps.

Limitations and Conclusion

This work introduces VADAR, an approach to critic learning that utilizes subtask decomposition, which provides significant improvements across a wide range of policy extraction algorithms as well as challenging environment specifications such as long horizon or safety filtering. Still, a limitation of VADAR is that it defining a fixed DAG structure of subtasks, whereas other methods enable conditioning on a continuum of goal information ([8, 54]). We look forward to generalizing the design principles of VADAR to this continuous or combinatorially large goal spaces, such as those described by images or text, in future work. Moreover, while recent literature has focused on improved methods for policy extraction, particularly with modern expressive policy parameterizations ([56, 52, 50, 19]), VADAR demonstrates that there remains significant untapped potential for algorithmic improvements to critic learning, particularly in long-horizon tasks. Moreover, in light of the recent popularity of scaling test-time computation ([59]), our study suggests a complimentary scaling direction: scaling the density of training signals used to supervise learning, which not only improves learning efficacy, but also reduces the need for complex test-time computation (e.g. VLM queries for goal-selection) at deployment.

Acknowledgments

The authors would like to thank Nidhi Hegde for helping with the initial implementation. MS would like to thank Abhishek Gupta and Andrea Bajcsy for insightful discussions. SS would like to thank Manan Agarwal and Giri Anantharaman for their help with the baseline implementations and Shreyas Kowshik, Michael Chen, Pranav Vignesh, Elizabeth Terveen and other members of CMU MaxLab for the informative discussions and feedback on an earlier draft. This research used the Babel computational cluster resources provided by the School of Computer Science at Carnegie Mellon University. MS and SBP would like to acknowledge support from the Toyota Research Institute (TRI) University 3.0 program.

Practitioner's Guide to red!50!black VaDar

Here we give a short hands-on reference for applying VaDAR to a new task or base algorithm. Subgoal decomposition. Most of the long horizon tasks can be decomposed naturally into subtasks which should be taked as the subgoals for VaDAR. For robotics tasks, we believe that the subgoals defined by us for our environments (Appendix § 16) can serve as good examples for subgoal selections for a new task.

Weighting variant. From our experiments, we found VaDAR-NF to be the most performative variant for robotic manipulation tasks, and recommend it as the default. VaDAR-CF can train slightly faster early on when the first subgoal is hard to reach from BC initialization, at the cost of greater susceptibility to over-exploiting the current goal. The myopic variants are useful as ablations but not as defaults.

Adapting to a new base algorithm. VaDAR is a wrapper over any Value Based RL Algorithm. Given any value-based RL algorithm A\mathcal{A}, the required modifications are:

  1. Change critic output dim from 1 to ∣G∣|\mathcal{G}| (backbone unchanged); ensemble aggregation becomes entrywise.

  2. Store per-step completion vectors dt\mathbf{d}_t and per-head rewards rtr_t in the replay buffer.

  3. Replace the critic loss with its summed-over-heads version; broadcast any Lreg\mathcal{L}_{\mathrm{reg}} across heads (see Appendix § 11.§ 1 for more details).

  4. Apply the entrywise reward clip in the TD target.

  5. In the policy loss, replace Qϕ(s,a)Q_\phi(s,a) with Qagg(s,a)=⟨w(d(s)),⟨Qϕ(s,a,g),g∈G⟩⟩Q_{\mathrm{agg}}(s,a)=\langle w(\mathbf{d}(s)),\langle Q_\phi(s,a,g),g\in\mathcal{G}\rangle\rangle.

The policy is not conditioned on subgoals. We have applied this recipe unchanged to SAC, FQL, and OGPO (Appendix § 11), and these serve as good references for applying VaDAR to any new Value Based RL Algorithm.

Critic Parameterizations, Ensembling, and Target Networks

This section expands the brief mention in § 3 of critic ensembling and target exponential moving averages (EMAs), which are standard practices in modern off-policy value-based RL and which we adopt without modification in VaDAR. The recipes below apply uniformly across the base instantiations used in our experiments; specific details that interact with the policy update are covered in Appendix § 11.

Critic Parameterizations

Vector-valued multi-goal critic. As an alternative to the goal-conditioned critic Qϕ(s,a,g)Q_\phi(s,a,g), VaDAR can use a multi-goal critic Qϕ:S×A→RGQ_\phi:\mathcal{S}\times\mathcal{A}\to\mathbb{R}^{\mathcal{G}} that maps each state-action pair to a ∣G∣|\mathcal{G}|-dimensional vector of Q-values. The gg-th entry Qϕ[g](s,a)Q_\phi[g](s,a) estimates the value of the state-action pair under the MDP induced by the per-subgoal reward rgr_g. Crucially, all ∣G∣|\mathcal{G}| entries share a single network backbone, enabling representation sharing across subgoals: features learned for predicting one subgoal’s completion are available to all others, Both parameterizations provide dense supervision across subgoals; Appendix A compares them empirically. Appendix § 22 shows the vector-valued critic can be interpreted simply as a structured instantiation of successor features (Barreto et al., [7]), where each feature corresponds to subgoal performance.

Critic Ensembles

Off-policy methods are known to suffer from overestimation bias: maximizing over a noisy approximation of QπQ^\pi systematically inflates the resulting target ([23, 13]). The standard remedy is to train an ensemble of MM critic networks Qϕ1,…,QϕM:S×A→RQ_{\phi_1},\ldots,Q_{\phi_M}:\mathcal{S}\times\mathcal{A}\to\mathbb{R} and aggregate their predictions when forming the TD target. We modify the classical scalar-valued critic in each ensemble member to be a goal-conditioned critic Qϕi:S×A×G→RQ_{\phi_i}:\mathcal{S}\times\mathcal{A}\times\mathcal{G}\to\mathbb{R}. Concretely, given any aggregator agg⁡M:RM→R\operatorname{agg}_M:\mathbb{R}^M\to\mathbb{R} (e.g., mean, min), we extend it to act on the MM vector critics by applying it independently to each subgoal head: for every g∈Gg\in\mathcal{G}.

[agg⁡M(Qϕ1,…,QϕM)](s,a,g)=agg⁡M(Qϕ1(s,a,g),…,QϕM(s,a,g)).(6)\left[\operatorname{agg}_M(Q_{\phi_1},\ldots,Q_{\phi_M})\right](s,a,g)=\operatorname{agg}_M\left(Q_{\phi_1}(s,a,g),\ldots,Q_{\phi_M}(s,a,g)\right). \tag*{(6)}

We use Eq. (B.1) as the meaning of “entrywise aggregation” throughout the paper, and refer back to it whenever an aggregator is applied to the critic. The ensemble size MM is taken to be 10 (M=10M=10) by default, unless otherwise specified.

Target Aggregation

The TD target Qϕˉ(s′,a′)Q_{\bar{\phi}}(s', a') is computed by aggregating the predictions of MM target critics Qϕˉ1,…,QϕˉM:S×A×G→RQ_{\bar{\phi}_1}, \ldots, Q_{\bar{\phi}_M}: S \times A \times G \to\mathbb{R}, whose parameters are an EMA of the online critics (Appendix § 10.§ 4). We use the ensemble mean as the aggregator agg⁡M\operatorname{agg}_M throughout, applied entrywise per Eq. (B.1):

Qϕˉ(s′,a′,g)=1M∑i=1MQϕˉi(s′,a′,g)∀g∈G.(7)Q_{\bar{\phi}}(s', a', g) = \frac{1}{M} \sum_{i=1}^{M} Q_{\bar{\phi}_i}(s', a', g) \quad\forall g \in G. \tag*{(7)}

The aggregated scalar Qtarg(s′,a′)=⟨w,⟨Qϕˉ(s′,a′,g),g∈G⟩⟩Q_{\mathrm{targ}}(s', a') = \langle w, \langle Q_{\bar{\phi}}(s', a', g), g \in G \rangle\rangle used downstream by the policy update (Eq. (3)) is recovered by an inner product with the policy-extraction weights ww (w(s)=GETWEIGHTING⁡(d(s))w(s) = \operatorname{GETWEIGHTING}(d(s)) based on the weighting strategies we described in Table 1). Note that other standard alternatives like the full ensemble minimum, or a randomized two-sample minimum in the style of REDQ (Chen et al. [13]) can also be used for the curtailing overestimation bias, but we found the entrywise mean to work consistently well across all VADAR experiments and adopt it as the default.

Target Network EMAs

To stabilize the moving Bellman target, we maintain a separate set of target critic parameters ϕˉ1,…,ϕˉM\bar{\phi}_1, \ldots, \bar{\phi}_M that lag the online parameters via Polyak averaging:

ϕˉi←(1−τ)ϕˉi+τϕi,i=1,…,M,(8)\bar{\phi}_i \leftarrow(1-\tau)\bar{\phi}_i + \tau\phi_i,\qquad i=1,\ldots,M, \tag*{(8)}

with decay rate τ∈(0,1]\tau\in(0,1] applied at every gradient step. We use τ=0.005\tau= 0.005 throughout. When the base algorithm requires a target policy, the same Polyak update is applied to the policy parameters:

θˉ←(1−τ)θˉ+τθ.(9)\bar{\theta} \leftarrow(1-\tau)\bar{\theta}+\tau\theta. \tag*{(9)}

Whether a target policy is maintained, and how it is used, is algorithm-specific; see Appendix § 11.

Critic Loss

Combining Eqs. (B.2) and (B.3) with VADAR’s formulation (Eq. (4)) yields the following per-ensemble-member critic loss, summed over subgoal heads:

Lcritic(Qϕi∣s,a,ytarget)=∑g∈G(Qϕi(s,a,g)−ygtarget)2,i=1,…,M,(10)L_{\mathrm{critic}}(Q_{\phi_i}\mid s,a,\mathbf{y}^{\mathrm{target}})=\sum_{g\in G}\left(Q_{\phi_i}(s,a,g)-y_g^{\mathrm{target}}\right)^2,\qquad i=1,\ldots,M, \tag*{(10)}

where ytarget\mathbf{y}^{\mathrm{target}} is the clipped vector TD target from Eq. (4).

Base-Algorithm Details: SAC, FQL, and OGPO

We position VADAR as a wrapper around any value-based RL algorithm that replaces each scalar critic with a goal-conditioned critic Qϕi:S×A×G→RQ_{\phi_i}: S \times A \times G \to\mathbb{R}, evaluated once per subgoal, with a few variations in the QtargQ_{\mathrm{targ}} aggregation that can be plugged into the base algorithm’s policy extraction objective. Here we present the policy losses LpolicyL_{\mathrm{policy}} and any algorithm-specific implementation details for the three base algorithms used in our experiments: SAC, FQL, and OGPO. § 6.1 pairs each base algorithm with a benchmark suite (SAC with OGBENCH and LIBERO, FQL with ROBOMIMIC, and OGPO with both ROBOMIMIC and ROBOCASA365); coupling each policy extraction method to its correspondingly challenging benchmark(s). The descriptions below are restricted to the components that interact with VADAR.

Soft Actor-Critic (SAC)

SAC (Haarnoja et al. [28]) is a maximum-entropy off-policy actor-critic algorithm that augments the standard return with a per-step entropy bonus αH(π(⋅∣s))\alpha\mathcal{H}(\pi(\cdot\mid s)). The optimal policy is the one that maximizes

J(π)=∑t≥0E(st,at)[R(st,at)+αH(π(⋅∣st))].(11)J(\pi)=\sum_{t\geq0}\mathbb{E}_{(s_t,a_t)}\left[R(s_t,a_t)+\alpha\mathcal{H}(\pi(\cdot\mid s_t))\right]. \tag*{(11)}

where α>0\alpha>0 is a temperature controlling the entropy-reward trade-off. We use a Gaussian policy with a tanh-squashing transform that bounds actions in [−1,1]dim⁡A[-1,1]^{\dim A} (Haarnoja et al. [28], Eq. (11)): at=tanh⁡(μθ(st)+σθ(st)⊙ϵt)a_t=\tanh(\mu_{\theta}(s_t)+\sigma_{\theta}(s_t)\odot\epsilon_t), with ϵt∼N(0,I)\epsilon_t\sim\mathcal{N}(0,I).

Critic loss. The soft Q-function is trained against an entropy-augmented Bellman target. Plugging into the value-based RL template of § 3 (Lcritic(Qϕ∣s,a,ytarget)=(Qϕ(s,a)−ytarget)2+LregL_{\mathrm{critic}}(Q_{\phi}\mid s,a,y^{\mathrm{target}})=(Q_{\phi}(s,a)-y^{\mathrm{target}})^2+L_{\mathrm{reg}}), the entropy bonus enters through the regularizer in the target:

ytarget=r+γ[Qtarg(s′,a′)−αlog⁡πθ(a′∣s′)],a′∼πθ(⋅∣s′),(12)y^{\mathrm{target}}=r+\gamma\left[Q_{\mathrm{targ}}(s',a')-\alpha\log\pi_{\theta}(a'\mid s')\right],\qquad a'\sim\pi_{\theta}(\cdot\mid s'), \tag*{(12)}

so that Lreg(s′,a′,πθ)=−γαlog⁡πθ(a′∣s′)L_{\mathrm{reg}}(s',a',\pi_{\theta})=-\gamma\alpha\log\pi_{\theta}(a'\mid s') in the notation of § 3.

Policy loss. The actor minimizes the KL-divergence between the policy and the (exponentiated) soft Q-function. Using the reparameterization trick (Haarnoja et al. [29]), this reduces to:

Lpolicy(πθ∣s,Qϕ)=Ea∼πθ(⋅∣s)[αlog⁡πθ(a∣s)−sg⁡[Qϕ(s,a)]],(13)L_{\mathrm{policy}}(\pi_{\theta}\mid s,Q_{\phi})=\mathbb{E}_{a\sim\pi_{\theta}(\cdot\mid s)}\left[\alpha\log\pi_{\theta}(a\mid s)-\operatorname{sg}[Q_{\phi}(s,a)]\right], \tag*{(13)}

where sg⁡\operatorname{sg} is the stopgrad operator. The expectation is estimated by a single reparameterized sample per state, and the gradient flows through the Squashed Gaussian into both the policy μθ\mu_{\theta} and σθ\sigma_{\theta} heads.

Temperature auto-tuning. We tune α\alpha to match a target entropy Hˉ=−dim⁡(A)\bar{H}=-\dim(A) via the auxiliary loss (Haarnoja et al. [29]):

L(α)=Ea∼πθ(⋅∣s)[−αlog⁡πθ(a∣s)−αHˉ].(14)L(\alpha)=\mathbb{E}_{a\sim\pi_{\theta}(\cdot\mid s)}\left[-\alpha\log\pi_{\theta}(a\mid s)-\alpha\bar{H}\right]. \tag*{(14)}

Interaction with VADAR. The scalar critic Qϕ:S×A→RQ_{\phi}:S\times A\to\mathbb{R} is replaced by the goal-conditioned critic Qϕ:S×A×G→RQ_{\phi}:S\times A\times G\to\mathbb{R}, which is evaluated once for every subgoal g∈Gg\in G. The Bellman target in Eq. (C.2) is computed for all goals, with the same next action a′a' and the same entropy regularizer shared across subgoals:

ytarget[g]=r[g]+γ[Qϕˉ(s′,a′,g)−αlog⁡πθ(a′∣s′)],a′∼πθ(⋅∣s′),∀g∈G,(15)y^{\mathrm{target}}[g]=r[g]+\gamma\left[Q_{\bar{\phi}}(s',a',g)-\alpha\log\pi_{\theta}(a'\mid s')\right],\qquad a'\sim\pi_{\theta}(\cdot\mid s'),\qquad\forall g\in G, \tag*{(15)}

where QϕˉQ_{\bar{\phi}} is the target critic as defined in Appendix § 10.§ 4, and the per-ensemble-member critic loss is summed over subgoals (Appendix § 10):

Lcritic(Qϕi∣s,a,ytarget)=∑g∈G(Qϕi(s,a,g)−ytarget[g])2.(16)L_{\mathrm{critic}}(Q_{\phi_i}\mid s,a,\mathbf{y}^{\mathrm{target}})=\sum_{g\in G}\left(Q_{\phi_i}(s,a,g)-y^{\mathrm{target}}[g]\right)^2. \tag*{(16)}

The actor sees only the aggregated scalar Qagg(s,a)=∑g∈Gw[g]Qϕ(s,a,g)Q_{\mathrm{agg}}(s,a)=\sum_{g\in G}w[g]Q_{\phi}(s,a,g) of Eq. (3), so the policy loss in Eq. (C.3) becomes

Lpolicy(πθ∣s,Qagg)=Ea∼πθ(⋅∣s)[αlog⁡πθ(a∣s)−sg⁡[∑g∈Gw[g]Qϕ(s,a,g)]].(17)L_{\mathrm{policy}}(\pi_{\theta}\mid s,Q_{\mathrm{agg}})=\mathbb{E}_{a\sim\pi_{\theta}(\cdot\mid s)}\left[\alpha\log\pi_{\theta}(a\mid s)-\operatorname{sg}\left[\sum_{g\in G}w[g]Q_{\phi}(s,a,g)\right]\right]. \tag*{(17)}

The temperature loss remains unchanged.

Flow Q-Learning (FQL)

FQL (Park et al. [50]) is an off-policy method designed for expressive flow-matching policies. Its key idea is to avoid backpropagating through the multi-step ODE integration of a flow policy by training a separate, expressive one-step policy that is simultaneously (i) regularized toward a frozen behavioral-cloning (BC) flow policy via a distillation loss and (ii) optimized to maximize the critic.

Architecture. FQL maintains three networks: a critic QϕQ_{\phi}, a BC flow policy μθ(s,z)\mu_{\theta}(s,z) that defines a flow-matching velocity field vθ(t,s,xt)v_{\theta}(t,s,x^{t}), and a one-step policy μω(s,z):S×Rdim⁡A→A\mu_{\omega}(s,z):S\times\mathbb{R}^{\dim A}\to A that maps Gaussian noise z∼N(0,I)z\sim\mathcal{N}(0,I) directly to an action.

BC flow loss. The BC flow policy is trained with the standard flow-matching objective on the offline dataset D\mathcal{D} (Park et al. [50]):

LFlow(θ)=E(s,a1)∼D, x0∼N(0,I), t∼Unif⁡[0,1][∥vθ(t,s,xt)−(a1−x0)∥22],xt=(1−t)x0+ta1.(18)\mathcal{L}_{\mathrm{Flow}}(\theta)=\mathbb{E}_{(s,a^{1})\sim\mathcal{D},\,x^{0}\sim\mathcal{N}(0,I),\,t\sim\operatorname{Unif}[0,1]}\left[\left\|v_{\theta}(t,s,x^{t})-(a^{1}-x^{0})\right\|_{2}^{2}\right],\qquad x^{t}=(1-t)x^{0}+ta^{1}. \tag*{(18)}

The corresponding stochastic policy πθ(a∣s)\pi_{\theta}(a\mid s) is the push-forward of N(0,I)\mathcal{N}(0,I) under the ODE μθ(s,⋅)\mu_{\theta}(s,\cdot) defined by vθv_{\theta}. Crucially, this loss is the only training signal seen by the flow policy: μθ\mu_{\theta} is never updated against QϕQ_{\phi}.

Critic loss. The critic is trained with the standard TD residual but using the one-step policy μω\mu_{\omega} in the bootstrap action (Park et al. [50]):

Lcritic(Qϕ)=E(s,a,r,s′)∼D, z∼N(0,I)[(Qϕ(s,a)−r−γQtarg(s′,μω(s′,z)))2].(19)L_{\mathrm{critic}}(Q_{\phi})=\mathbb{E}_{(s,a,r,s')\sim\mathcal{D},\,z\sim\mathcal{N}(0,I)}\left[\left(Q_{\phi}(s,a)-r-\gamma Q_{\mathrm{targ}}(s',\mu_{\omega}(s',z))\right)^2\right]. \tag*{(19)}

where Qtarg=QϕˉQ_{\mathrm{targ}}=Q_{\bar{\phi}} is the EMA target critic (Appendix § 10.§ 4). One-step policy loss. The one-step policy is trained against an actor loss with two terms: a Q-maximization term and a distillation term that pulls μω(s,z)\mu_\omega(s,z) toward the BC flow policy’s output μθ(s,z)\mu_\theta(s,z) at the same noise zz [50 Eqs. (7),(9]:

Lpolicy(πω∣s,Qϕ)=Es∼D, z∼N(0,I)[−Qϕ(s,aπ)]aπ=μω(s,z)+αFQLEs∼D, z∼N(0,I)[∥μω(s,z)−μθ(s,z)∥22].(20)\mathcal{L}_{\mathrm{policy}}(\pi_\omega\mid s,Q_\phi)=\mathbb{E}_{s\sim\mathcal{D},\,z\sim\mathcal{N}(0,I)} \left[-Q_\phi(s,a^\pi)\right]_{a^\pi=\mu_\omega(s,z)} +\alpha_{\mathrm{FQL}}\mathbb{E}_{s\sim\mathcal{D},\,z\sim\mathcal{N}(0,I)} \left[\lVert\mu_\omega(s,z)-\mu_\theta(s,z)\rVert_2^2\right]. \tag*{(20)}

Gradients of the distillation term are stopped at μθ\mu_\theta, so the BC flow policy is not affected.

Interaction with VADAR. When applying VADAR on top of FQL, the scalar critic is replaced by the goal-conditioned critic Qϕ:S×A×G→RQ_\phi:\mathcal{S}\times\mathcal{A}\times\mathcal{G}\to\mathbb{R}, evaluated once for every subgoal g∈Gg\in G. The TD bootstrap is computed for all goals, using a single bootstrap action μω(s′,z)\mu_\omega(s',z) shared across subgoals:

y[g]target=r[g]+γQϕˉ(s′,μω(s′,z),g),z∼N(0,I),∀g∈G,(21)y^{\mathrm{target}}_{[g]}=r_{[g]}+\gamma Q_{\bar{\phi}}(s',\mu_\omega(s',z),g),\qquad z\sim\mathcal{N}(0,I),\qquad\forall g\in G, \tag*{(21)}

and the per-ensemble-member critic loss is summed over subgoals:

Lcritic(Qϕi∣s,a,ytarget)=E(s,a,r,s′)∼D, z[∑g∈G(Qϕi(s,a,g)−y[g]target)2].(22)\mathcal{L}_{\mathrm{critic}}(Q_{\phi_i}\mid s,a,\mathbf{y}^{\mathrm{target}})=\mathbb{E}_{(s,a,r,s')\sim\mathcal{D},\,z}\left[\sum_{g\in G}\left(Q_{\phi_i}(s,a,g)-y^{\mathrm{target}}_{[g]}\right)^2\right]. \tag*{(22)}

Similar to before, we define the online aggregate Qagg(s,a):=∑g∈Gw[g]Qϕ(s,a,g)Q_{\mathrm{agg}}(s,a):=\sum_{g\in G}\mathbf{w}[g]Q_\phi(s,a,g) (online critic, matching the FQL convention of Eq. (C.10)). The Q-loss term in Eq. (C.10) then becomes

Lpolicy(πω∣s,Qagg)=Es∼D, z∼N(0,I)[−∑g∈Gw[g]Qϕ(s,aπ,g)]aπ=μω(s,z)+αFQLLdistill(ω;θ).(23)\mathcal{L}_{\mathrm{policy}}(\pi_\omega\mid s,Q_{\mathrm{agg}})=\mathbb{E}_{s\sim\mathcal{D},\,z\sim\mathcal{N}(0,I)} \left[-\sum_{g\in G}\mathbf{w}[g]Q_\phi(s,a^\pi,g)\right]_{a^\pi=\mu_\omega(s,z)} +\alpha_{\mathrm{FQL}}\mathcal{L}_{\mathrm{distill}}(\omega;\theta). \tag*{(23)}

The distillation term is unchanged. The deployed policy is the one-step μω\mu_\omega, which takes no goal as input and hence requires no goal information at inference (in line with VADAR’s subgoal-agnostic property).

Off-Policy Generative Policy Optimization (OGPO)

OGPO [52] is an off-policy actor-critic for generative control policies (GCPs), like flow matching and diffusion, that side-steps the instabilities of backpropagation through the flow ODE by applying a zero-order PPO-style update to the denoising chain itself. VADAR pairs with OGPO on our most challenging long-horizon benchmarks (ROBOMIMIC, and ROBOCASA365). We summarize the policy loss here and note that OGPO follows the same aforementioned critic ensembling and EMA strategies (Appendix § 10).

Setup. A flow-based GCP samples an action at,0a_{t,0} via a KK-step denoising chain at,K→at,K−1→⋯→at,0a_{t,K}\to a_{t,K-1}\to\cdots\to a_{t,0}, where each step samples at,k−1∼πθ(⋅∣st,at,k)a_{t,k-1}\sim\pi_\theta(\cdot\mid s_t,a_{t,k}). The likelihood of a full chain factorizes as

πθ(at,K:0∣st)=∏k=1Kπθ(at,k−1∣st,at,k).(24)\pi_\theta(a_{t,K:0}\mid s_t)=\prod_{k=1}^{K}\pi_\theta(a_{t,k-1}\mid s_t,a_{t,k}). \tag*{(24)}

OGPO treats the denoising trajectory at,K:0a_{t,K:0} as a sequence in an inner “denoising MDP” embedded inside the environment MDP, and applies PPO at the level of this inner MDP only.

Annealed importance sampling ratio. The reference policy is the EMA target πˉθ\bar{\pi}_\theta (Appendix § 10.§ 4). The policy ratio is the product of per-step ratios over the entire chain:

ωθ:=∏k=1Kπθ(at,k−1∣st,at,k)πˉθ(at,k−1∣st,at,k).(25)\omega_\theta:=\prod_{k=1}^{K}\frac{\pi_\theta(a_{t,k-1}\mid s_t,a_{t,k})}{\bar{\pi}_\theta(a_{t,k-1}\mid s_t,a_{t,k})}. \tag*{(25)}

OGPO Policy Updates OGPO performs policy extraction via a PPO-style zeroth order policy gradient update that considers the entire flow-denoising chain as an MDP. The terminal action at,0a_{t,0} is the only action evaluated against the target critic. OGPO draws a group of NgroupN_{\mathrm{group}} denoising trajectories aK:0(i,j)a_{K:0}^{(i,j)} from πˉθ(⋅∣s(i))\bar{\pi}_\theta(\cdot\mid s^{(i)}) per state and uses the group mean as a value baseline:

V^(i)←1Ngroup∑j=1NgroupQtarg(s(i),a0(i,j)),A^=Qtarg(st,at,0)−V^.(26)\hat{V}^{(i)}\leftarrow\frac{1}{N_{\mathrm{group}}}\sum_{j=1}^{N_{\mathrm{group}}}Q_{\mathrm{targ}}(s^{(i)},a_{0}^{(i,j)}),\qquad\hat{A}=Q_{\mathrm{targ}}(s_t,a_{t,0})-\hat{V}. \tag*{(26)}

This GRPO-style baseline obviates the need for a separately learned value network.

Policy loss. The PPO-clipped surrogate loss is applied over the entire flow denoising chain as:

Lpolicy(πθ∣s,Qtarg)=−Est∼Bat,K:0∼πˉθ(⋅∣st)[min⁡(ωθA^,clip⁡(ωθ,1−ϵ,1+ϵ)A^)](27)L_{\mathrm{policy}}(\pi_\theta\mid s, Q_{\mathrm{targ}}) = -\mathbb{E}_{\substack{s_t \sim\mathcal{B} \\ a_{t,K:0} \sim\bar{\pi}_\theta(\cdot\mid s_t)}} \left[\min\left(\omega_\theta\hat{A}, \operatorname{clip}(\omega_\theta, 1-\epsilon, 1+\epsilon)\hat{A}\right)\right] \tag*{(27)}

where ϵ\epsilon is the PPO clip parameter. Because both ωθ\omega_\theta and the denoising trajectories can be computed in parallel without the need for environmental interactions, the loss is fully parallelizable across the Nbatch×NgroupN_{\mathrm{batch}} \times N_{\mathrm{group}} batch.

Noise injection for flow policies. Evaluating ωθ\omega_\theta requires per-step likelihoods πθ(at,k−1∣st,at,k)\pi_\theta(a_{t,k-1} \mid s_t, a_{t,k}) to be nonsingular. Following Zhang et al. [69], we replace the deterministic Euler step with a Gaussian-noised update at,k−1∼N(vθ(at,k,k/K,st),σk2I)a_{t,k-1} \sim\mathcal{N}(v_\theta(a_{t,k}, k/K, s_t), \sigma_k^2 I), and retain the marginal path-preserving correction of Albergo et al. [2] so that the per-step SDE marginals match standard ODE flow matching rollouts.

Interaction with VADAR. The critic is the goal-conditioned Qϕ:S×A×G→RQ_\phi: \mathcal{S} \times\mathcal{A} \times\mathcal{G} \to\mathbb{R}, trained on the all-goal targets ytarget[g]=r[g]+γQϕˉ(s′,a′,g)y^{\mathrm{target}}[g] = r[g] + \gamma Q_{\bar{\phi}}(s', a', g) for every g∈Gg \in G, exactly as above. VADAR composes with OGPO by replacing the scalar QtargQ_{\mathrm{targ}} in the advantage computation with the aggregate of the target critic evaluated on every subgoal, following Eq. (3):

Qagg(s,a)=∑g∈Gw[g]Qϕˉ(s,a,g),Q_{\mathrm{agg}}(s,a) = \sum_{g \in G} w[g] Q_{\bar{\phi}}(s,a,g),
V^(i)←1Ngroup∑j=1NgroupQagg(s(i),a0(i,j)),A^=Qagg(st,at,0)−V^.(28)\hat{V}^{(i)} \leftarrow\frac{1}{N_{\mathrm{group}}}\sum_{j=1}^{N_{\mathrm{group}}} Q_{\mathrm{agg}}(s^{(i)},a_0^{(i,j)}), \qquad\hat{A} = Q_{\mathrm{agg}}(s_t,a_{t,0}) - \hat{V}. \tag*{(28)}

The same weights ww are used for all NgroupN_{\mathrm{group}} samples of a given state, since they depend only on that state’s completion vector. The PPO surrogate, group baseline, and noise-injection scheme are otherwise unchanged. For our experiments, unless otherwise stated, we use the OGPO+ variant (due to its superior performance).

Limitations of Explicit and Implicit Hierarchical Goal Conditioning

We examine the limitations summarized in § 4 using implicit hierarchical goal conditioning (iHGC), which preserves goal-conditioned critics but removes goal conditioning from the policy.

Implicit Hierarchical Goal Conditioning

Definition D.1 (Implicit Hierarchical Goal Conditioning (iHGC)). Implicit goal conditioning follows the Hierarchical Goal Conditioning in Algorithm 1, with the sole modification that the policy π(a∣s)\pi(a \mid s) is not conditioned on subgoal information.

We run HGC using oracle dones, providing perfect subgoal selection for this comparison.

iHGC eliminates the first limitation described above, as now the policy need not incorporate goal information at deployment time. As an immediate example, Figure 10 compares iHGC and HGC on Robomimic Tool-hang with the FQL (Appendix § 11 for more details) policy extraction backbone. Removing goal information from the policy improves final performance and yields a smoother learning curve, suggesting that goal-conditioning the policy can introduce unnecessary instability even when goal-conditioned critics remain useful. We hypothesize that removing goal-conditioning inputs avoids aliasing at goal boundaries.

Both Explicit and Implicit Hierarchical Goal Conditioning Suffer from Myopic Behavior

However, we still find that implicit and explicit hierarchical goal conditioning run into a common failure mode: overly myopic policy exploitation. Both methods update the policy against the critic of the currently active subgoal alone. As a result, the policy is rewarded for completing the current subgoal without accounting for how its manner of completion affects subsequent subtasks. We illustrate this failure mode on ROBOCASA365 ARRANGETEA and PRESOAKPAN in Figure 11.

Rollouts illustrating myopic failure modes of goal-conditioned policies

Figure 11. Myopic failure modes of goal-conditioned policies. Rollouts of policies trained to optimize only the current subgoal on ROBOCASA365 ARRANGETEA (a, top) and PRESOAKPAN (b, bottom); frames are chronological within each row. In (a), the kettle is placed on the tray without leaving room for the mug, so the subsequent mug placement collides with the kettle and knocks it off the counter. In (b), the pan is dropped into the sink in a position that does not permit a stable placement of the sponge on top of it

In Figure 11(a), the policy places the kettle on the tray in a position that obstructs the subsequent mug placement. The “placing the kettle” subgoal is complete, but attempting to place the mug causes a collision that knocks the kettle off the counter. In Figure 11(b), the policy places the pan in the sink in a position that prevents stable placement of the sponge. Both these failure cases are manifestations of over-optimizing for the current subtask (Limitation 2). The policy does not distinguish between merely completing the current subgoal and completing it in a way that facilitates future subgoals. Our method, VADAR, addresses this limitation by optimizing the policy for future-subgoal values (§ 5), providing a learning signal for actions that enable progress toward full-task completion rather than just myopically completing the current subtask.

Dense Preference Reward: Algorithms and Implementation Details

Background and Limitations

An alternative to HGC for solving the credit assignment problem is to craft a denser reward signal rr. When oracle dense rewards are not available, the most popular data-dependent method has been preference-based dense rewards defined at the level of full trajectories τ=(s1,a1,…,sH,aH)\tau= (s_1, a_1, \ldots, s_H, a_H). Given two trajectories τ1,τ2\tau_1, \tau_2, one maximizes the Bradley-Terry-Luce (BTL) [9] likelihood, assuming the probability that τ2\tau_2 is preferred to τ1\tau_1 is given by

P(τ1≺τ2∣ψ,β)=exp⁡(βRψ(τ2))exp⁡(βRψ(τ1))+exp⁡(βRψ(τ2)),(29)P(\tau_1 \prec\tau_2 \mid\psi, \beta) = \frac{\exp(\beta R_\psi(\tau_2))}{\exp(\beta R_\psi(\tau_1)) + \exp(\beta R_\psi(\tau_2))}, \tag*{(29)}

where Rψ(τ)=∑s∈τrψ(s)R_\psi(\tau) = \sum_{s \in\tau} r_\psi(s) is cumulative trajectory return, rψ(s)r_\psi(s) is a learned, per-state dense reward, and β\beta is a temperature parameter. Preference based rewards have shown strong promise ([68, 55, 63]), but are subject to other limitations: (1) assigning consistent numerical progress values on very long, multi-step tasks is challenging; (2) statistical errors in reward learning can lead to reward hacking; and (3) effective reward learning can require large datasets. Dense rewards are imperfect proxies for desired behavior, and agents may exploit errors in the learned reward rather than complete the task ([3, 32]). For an empirical comparison with value disaggregation, see Appendix § 14.§ 2. We describe our implementation below.

Implementation

We implemented dense preference rewards for the same policy backbones as VADAR: OGPO, SAC, FQL. Our PBRL implementation was heavily inspired by [58, 66, 64]. We used trajectory-wise preferences to train a state-based reward model independently for each task. Trajectory preferences were autogenerated and fit using the Bradley–Terry [9] model. When training the reward model, we uniformly sample from the generated preference dataset. We found that the most reliable preference signal was binary trajectory success. To ensure we used the most accurate preference signal, we experimented with alternatives in Table 3, but found them to be inaccurate.

SignalObserved issue
∑s∈τV∗(s)\sum_{s\in\tau} V^{*}(s)Uses the value function V∗V^{*} from an optimal policy π∗\pi^{*}, but produced noisy preferences because V∗V^{*} gave poor estimates outside the distribution induced by π∗\pi^{*}.
∑t,st∈τγtrsparse(st)\sum_{t,s_{t}\in\tau} \gamma^{t}r_{\mathrm{sparse}}(s_{t})Monte Carlo returns depend on the policy used to generate the trajectory, making the induced reward signal noisy and not policy-invariant at the state level.
∑rdense∈τrdense\sum_{r_{\mathrm{dense}}\in\tau} r_{\mathrm{dense}}Uses rdenser_{\mathrm{dense}} as defined from the QIPM main algorithm, but it preferred longer trajectories, which introduced a harder and less reliable preference signal for the reward model to learn.

Table 3. Alternative auto generated preference signals tested for PBRL.

Offline Training PBRL was provided privileged information, a diverse set of trajectories to kick off it’s reward training from, 100 trajectories collected from (πEARLY,πMID,πLATE)(\pi_{\mathrm{EARLY}},\pi_{\mathrm{MID}},\pi_{\mathrm{LATE}}), where these were checkpoints collected during the training of an optimal policy, π∗\pi^{*}. This dataset combined with the original optimal success dataset provided on average 500 trajectories to pre-train the reward model from, with different levels of optimality and a diverse range of failure modes. We saw that this was required for PBRL to show signs of success, as demonstrated in Figure 12. This is a significant setback compared to VADAR, which is able to operate on far more limited datasets.

Comparison of success-only and mixed dataset based reward rollouts

Figure 12. Comparison of success-only (top) and mixed (bottom) dataset based reward rollouts. We see that the success-only dataset is unable to accurately estimate failure cases, not having a clear signal for reward. Both models were using preference based on V∗V^{*}

Online Training After offline reward-model pretraining, we continue updating the dense reward model during online RL following [66, 33]. The reward model is trained using the following iterative loop: the current reward model is used to train the policy, the resulting policy collects new trajectories, these trajectories are converted into additional synthetic preference comparisons, and the reward model is fine-tuned on the offline and online preference dataset.

During online RL, the environment reward is replaced by the learned state-only reward model r^θ(s)\hat{r}_{\theta}(s). The underlying RL algorithm is otherwise unchanged: OGPO, SAC, and FQL use their standard actor and critic losses, but with rewards given by r^θ\hat{r}_{\theta} instead of the environment reward.

Every KintK_{\mathrm{int}} environment steps, the reward model is fine-tuned on the union of offline and online preference pairs using the same Bradley–Terry objective as in offline pretraining. The reward model gradually adapts to the on-policy state distribution induced by the improving policy, while still being regularized by the original offline preference data.

After each reward-model update, we relabel the live RL replay buffer with the updated reward model: rt←r^θ(st)r_t \leftarrow\hat{r}_{\theta}(s_t). This step makes the online reward update immediately affect subsequent policy learning.

Hyper Parameters Table 4 mentions the primary hyper parameters used to train the model.

HyperparameterValue
Reward model
ensemble size3
model architectureMLP of (512, 512, 512, 512)
learning rate3×10−43 \times10^{-4}
training epochs50
segment sizeepisode
Online PBRL schedule
RM update interval5,000 env steps
maximum online feedback10,000 preference pairs
feedback bufferFIFO

Table 4. Reward-model hyperparameters used for PBRL experiments.

Additional Baseline Comparisons

Hindsight experience replay (HER) struggles with contact-rich control

An alternative approach to mitigating the credit assignment problem in goal-conditioned RL is hindsight experience replay (HER) [4], which relabels states reached during a rollout as goals, so that the policy receives learning signal even when the task reward is never observed. HER has largely been evaluated in reaching, pick-and-place and maze settings, where any state is a reasonable goal. To test whether it applies to contact-rich manipulation settings, we evaluate HER with SAC on LIBERO-Task 6, with goals given by object states and relabelled from future states of the same episode, and with the same BC pretraining, online step budget and networks as our other SAC baselines. We also steelman HER with two variants that provide it with the same subgoal information that HGC receives: an oracle that supplies a representative goal state for each subgoal in turn, and a learned goal embedding trained to recognize subgoal completions. In both the variants, we find HER to be incapable of performing multi stage complex manipulation tasks, with the eventual success rate remaining quite close to the initial BC success rate.

TD critic learning outperforms preference-based per-subgoal reward

The Dense baseline in § 6 learns rewards from the full trajectory preference, so it does not receive the same subgoal information as VaDAR. To steelman the comparison, we learn separate preference reward models for each of the subtasks, aggregate them using our Next-Future weighting same as VaDAR-NF, and then train a scalar critic on aggregated reward. We call this method Pref-Subgoal. Figure 13 shows that the Pref-Subgoal underperforms VaDAR-NF on all four tasks. On PreSoakPan, Pref-Subgoal appears to be more sample efficient in the early stages but later fails to reach the same final performance as VaDAR-NF. On ArrangeTea and LoadDishwasher, it does not improve over its BC initialization. With FQL on Tool-Hang, Pref-Subgoal trains more slowly and plateaus below VaDAR-NF.

Success rate comparisons for VaDAR-NF and Pref-Subgoal across four tasks

Figure 13. Success rate of VaDAR-NF against Pref-Subgoal, which learns a dense reward per subgoal from preferences, on RoboCasa365 with the OGPO backbone and Robomimic Tool-Hang with the FQL backbone.

These results suggest that directly fitting the critic on multiple temporal resolutions is more reliable than learning dense per-subgoal rewards from preferences. We conjecture that small reward-model errors accumulate over the long horizon, while VADAR directly supervises per-subgoal TD targets at multiple temporal resolutions while sharing representations across subgoals. This structure may benefit from the implicit Bellman-backup mechanism discussed in § 6.4.

DAG Extension for Non-Sequential Subtasks

Multiple subgoals may be active simultaneously, as in bimanual robotic manipulation. The value-weighting function in Table 1 extends naturally to this setting: a DAG can express both parallel subgoals and prerequisites that must be completed before a subsequent goal becomes available.

In the sequential setting, subtasks must be completed in a fixed linear order. However, many real-world tasks have more complex structures where multiple subtasks can be pursued simultaneously for optimal behavior. Moreover, some subtasks can form a precondition boundary for others, where policies can be rolled out in the future subtasks only if the preconditions are satisfied. VADAR can naturally be applied to a litany of these subtask dependencies often represented as directed acyclic graphs (DAGs) with each node corresponding to an individual Q-function. This provides the core benefits of value disaggregation to rich dependency structures dominant in long-horizon RL settings.

Illustrative Example: Robomimic Transport. To illustrate this, consider the robomimic Transport task, a multi-robot task where two robot arms must coordinate to transport a payload from one bin to another while managing trash removal. The task decomposes into 7 interdependent subtasks forming a DAG:

  1. lid_removed: Lid removed from start bin

  2. payload_grasped: Payload grasped from start bin (Robot 0)

  3. payload_lifted: Payload lifted above bins (Robot 0)

  4. trash_grasped: Trash grasped from target bin (Robot 1)

  5. trash_in_bin: Trash placed in trash bin (Robot 1)

  6. payload_handed_off: Payload transferred from Robot 0 to Robot 1

  7. payload_in_target: Payload placed in target bin (SUCCESS)

The dependencies form two parallel execution paths: (1 → 2 → 3) for payload handling by Robot 0, and (4 → 5) for trash management by Robot 1. The handoff step (6) requires both paths to complete, and the final success (7) depends on the handoff, an illustration is provided in Figure 14.

Transport Subgoal Structure

Figure 14. Transport Subgoal Structure.

DAG Formalism. Recall from § 4 that the subgoals GG are partially ordered, where gi≺gjg^{i} \prec g^{j} means that gig^{i} must be completed before gjg^{j}. We represent this structure with a DAG D=(G,E)\mathcal{D} = (G,E), with

(gi,gj)∈E⟺gi≺gj and ∄gk∈G such that gi≺gk≺gj.(g^{i}, g^{j}) \in E \Longleftrightarrow g^{i} \prec g^{j} \text{ and } \nexists g^{k} \in G \text{ such that } g^{i} \prec g^{k} \prec g^{j}.

This graph is acyclic: a directed cycle would imply gi≺gig^{i} \prec g^{i} for some subgoal gig^{i}, contradicting the prerequisite relation. The linear ordering from § 4 is the special case in which the DAG is a chain: every subgoal except the initial one has exactly one parent. We define the following graph functions:

pred⁡(q):={p∈G:(p,q)∈E}desc⁡(S):={q∈G:∃s∈S such that s≺q}.\operatorname{pred}(q) := \{p \in G : (p,q) \in E\} \qquad\operatorname{desc}(S) := \{q \in G : \exists s \in S \text{ such that } s \prec q\}.

Gate Weights for DAG Structures. For a DAG, the “current” subgoals are the incomplete goals whose prerequisites have all been completed:

Gcurr(d):={g∈G:dg=0 and dp=1 ∀p∈pred⁡(g)},G^{\mathrm{curr}}(\mathbf{d}) := \{g \in G : d_g = 0 \text{ and } d_p = 1 \ \forall p \in\operatorname{pred}(g)\},

We also define the terminal subgoals and the immediate successors of the current frontier as

Gterm:={g∈G:∄h∈G such that (g,h)∈E},G^{\mathrm{term}} := \{g \in G : \nexists h \in G \text{ such that } (g,h) \in E\},
Gnext(d):={g∈G:∃p∈Gcurr(d) such that (p,g)∈E}∪(Gcurr(d)∩Gterm).G^{\mathrm{next}}(\mathbf{d}) := \{g \in G : \exists p \in G^{\mathrm{curr}}(\mathbf{d}) \text{ such that } (p,g) \in E\} \cup(G^{\mathrm{curr}}(\mathbf{d}) \cap G^{\mathrm{term}}).

For gnextg^{\mathrm{next}}, we handle the edge case where an element from gcurrg^{\mathrm{curr}} is a terminal node, putting those nodes into the set. Using these sets, the DAG value-weighting variants are given in Table 5, provided that we are given d\mathbf{d}.

Config.Weight wgw_gInterpretation
Current-myopicI[g∈Gcurr(d)]\mathbb{I}[g \in G^{\mathrm{curr}}(\mathbf{d})]Only currently available subtasks contribute.
Current-futureI[g∈desc⁡(Gcurr(d))]\mathbb{I}[g \in\operatorname{desc}(G^{\mathrm{curr}}(\mathbf{d}))]Current subtasks and their descendants contribute.
Next-myopicI[g∈Gnext(d)]\mathbb{I}[g \in G^{\mathrm{next}}(\mathbf{d})]Only direct successors of current subtasks contribute.
Next-futureI[g∈desc⁡(Gnext(d))]\mathbb{I}[g \in\operatorname{desc}(G^{\mathrm{next}}(\mathbf{d}))]Next subtasks and their descendants contribute.

Table 5. DAG value-weighting variants used for policy extraction in VADAR.

Environment Details: Per-Task Descriptions: Subgoals, Dones, and DAG Structure

We list the subgoals used by VADAR and HGC on each benchmark task, taken directly from the simulator wrappers in our codebase. For each task we list (i) the high-level objective, (ii) the per-subgoal completion indicators done⁡g(s)\operatorname{done}_g(s) as evaluated from low-level simulator state (no VLM calls are used for done evaluation), and (iii) the prerequisite structure used to gate subgoal completion. Subgoal completion is evaluated with monotonic completion semantics: once done⁡g\operatorname{done}_g becomes true and all prerequisites are satisfied, done⁡g\operatorname{done}_g remains true for the rest of the episode. The default prerequisite structure is a sequential chain (gig^i requires g1,…,gi−1g^1,\ldots,g^{i-1}); tasks that override this with an explicit DAG are noted below. See Appendix § 15 for an in-depth explanation of the DAG structure.

Robocasa365

Our longest-horizon tasks come from the ROBOCASA365 (Nasiriany et al., [46]) suite. Each task involves multiple sequential phases of long-horizon kitchen manipulation. For our experiments, we use the state-based environments of ROBOCASA365 composite tasks (having multiple subtasks to be done in order to achieve success), which do not involve the movement of the robot’s base. Concretely, we evaluate on the following ROBOCASA365 tasks:

PRESOAKPAN. The agent must place a pan in the sink, place a sponge inside the pan in the sink, and then turn on the faucet. This task is decomposed into 5 sequential subgoals:

  1. g1=pan_grasped_liftedg^1 = \texttt{pan\_grasped\_lifted}: pan (obj1) is grasped and lifted to height z>0.88z > 0.88 m.

  2. g2=pan_in_sinkg^2 = \texttt{pan\_in\_sink}: pan is contained inside the sink basin (obj_inside_of(obj1, sink)).

  3. g3=sponge_grasped_liftedg^3 = \texttt{sponge\_grasped\_lifted}: sponge (obj2) is grasped and lifted to height z>0.88z > 0.88 m.

  4. g4=sponge_in_pan_and_sinkg^4 = \texttt{sponge\_in\_pan\_and\_sink}: sponge is inside the sink, with xyxy-distance to pan <0.25< 0.25 m and vertical offset zsponge−zpan∈(−0.05,0.20)z_{\mathrm{sponge}} - z_{\mathrm{pan}} \in(-0.05, 0.20) m.

  5. g5=water_ong^5 = \texttt{water\_on}: faucet is on, both pan and sponge are in the sink, and the gripper is far from both objects (matches the environment’s original success check).

Final task success is done⁡g5\operatorname{done}_{g^5}.

LOADDISHWASHER. The agent must load two dishes (a cup and a bowl) into the dishwasher rack and close the dishwasher. Decomposed into 5 sequential subgoals:

  1. g1=cup_graspedg^1 = \texttt{cup\_grasped}: cup (dish0) is grasped (check_obj_grasped).

  2. g2=cup_in_dwg^2 = \texttt{cup\_in\_dw}: cup is in contact with the dishwasher rack.

  3. g3=bowl_graspedg^3 = \texttt{bowl\_grasped}: bowl (dish1) is grasped.

  1. g4=bowl_in_dwg^{4} = \texttt{bowl\_in\_dw}: bowl is in contact with the dishwasher rack.

  1. g5=dw_closedg^{5} = \texttt{dw\_closed}: cup in rack and bowl in rack and dishwasher door closed (angular tolerance 0.05).

Final task success is doneg5_{g^{5}}.

ARRANGETEA: This is the third ROBOCASA365 task we evaluate on. In this task, the agent must pick up the kettle and place it on a tray on the kitchen counter, then pick up a mug from inside the kitchen cabinet and then place it next to the kettle on the tray, and then close the hinged cabinet that initially holds the mug. This task is decomposed into 6 natural subgoals:

  1. g1=kettle_graspedg^{1} = \texttt{kettle\_grasped}: kettle is grasped (\texttt{check_obj_grasped(kettle)}).

  1. g2=kettle_on_trayg^{2} = \texttt{kettle\_on\_tray}: kettle is contained on the tray (\texttt{check_obj_in_receptacle(kettle, container)}).

  1. g3=kettle_on_tray & mug_graspedg^{3} = \texttt{kettle\_on\_tray \& mug\_grasped}: kettle still on tray and mug is grasped.

  1. g4=kettle_on_tray & mug_on_trayg^{4} = \texttt{kettle\_on\_tray \& mug\_on\_tray}: kettle still on tray and mug is now on the tray.

  1. g5=both_on_tray & right_door_closedg^{5} = \texttt{both\_on\_tray \& right\_door\_closed}: both objects on tray and the right cabinet door is closed.

  1. g6=both_on_tray & cab_closedg^{6} = \texttt{both\_on\_tray \& cab\_closed}: both objects on tray and the full hinged cabinet is closed.

Final task success is doneg6_{g^{6}} which matches the original success of the ARRANGETEA task.

Robomimic

ROBOMIMIC (Mandlekar et al. [41]) contributes two contact-rich, medium-horizon tasks.

TRANSPORT. A bimanual task in which two arms coordinate to lift a lid, transport a payload between bins, and dispose of trash. The codebase exposes 7 subgoal indicators; the user-facing description of the DAG is given in Appendix § 15. The 7 subgoal checks are:

  1. g1=lid_removedg^{1} = \texttt{lid\_removed}: lid handle has slid leftward by more than 0.08 m from its initial xx-position (xlid<xlidinit−0.08x_{\mathrm{lid}} < x_{\mathrm{lid}}^{\mathrm{init}} - 0.08).

  1. g2=payload_graspedg^{2} = \texttt{payload\_grasped}: Robot 0’s gripper is in contact with the payload geoms (\texttt{_check_grasp}).

  1. g3=payload_liftedg^{3} = \texttt{payload\_lifted}: payload zz-position exceeds the table height by >0.10> 0.10 m.

  1. g4=trash_graspedg^{4} = \texttt{trash\_grasped}: Robot 1’s gripper is in contact with the trash geoms.

  1. g5=trash_in_bing^{5} = \texttt{trash\_in\_bin}: \texttt{transport.trash_in_trash_bin} flag is true.

  1. g6=payload_handed_offg^{6} = \texttt{payload\_handed\_off}: Robot 1 is grasping the payload and Robot 0 is not.

  1. g7=payload_in_targetg^{7} = \texttt{payload\_in\_target}: \texttt{transport.payload_in_target_bin} flag is true.

Structure: two parallel chains (g1→g2→g3g^{1} \to g^{2} \to g^{3}) and (g4→g5g^{4} \to g^{5}), both prerequisites for g6g^{6}, which is itself a prerequisite for g7g^{7} (Figure 14). Final task success is doneg7_{g^{7}}.

TOOL-HANG. A single-arm precision task: insert a hook frame into a base stand, then hang a wrench on it. Decomposed into 4 sequential subgoals:

  1. g1=frame_graspedg^{1} = \texttt{frame\_grasped}: gripper is grasping the frame and the frame is lifted >0.02> 0.02 m above the table (\texttt{_check_grasp_frame}).

  1. g2=frame_insertedg^{2} = \texttt{frame\_inserted}: frame is assembled into the stand (\texttt{_check_frame_assembled}).

  1. g3=frame_inserted_and_wrench_graspedg^{3} = \texttt{frame\_inserted\_and\_wrench\_grasped}: frame is still inserted and the wrench is grasped and lifted >0.02> 0.02 m above the table.

  1. g4=wrench_attachedg^{4} = \texttt{wrench\_attached}: wrench is hung on the inserted frame (\texttt{_check_tool_on_frame}).

The final success is doneg4_{g^{4}} which matches the success of the original TOOL-HANG env.

OGBench

Next we use the scene-play-singletask-task4 (put_in_drawer) from OGBENCH [49]. In this task, the agent needs to first unlock the drawer (by pressing a button), then open the drawer, then lift the cube and place it inside the drawer and finally close the drawer. The subgoal decomposition is as follows:

  1. g1=g^{1} =press_button_0: button 0 state is set to pressed (_cur_button_states[0] == 1); this unlocks the drawer.

  2. g2=g^{2} =open_drawer: drawer slide xx-position <−0.12< -0.12 (slide range [−0.16,0][-0.16, 0]).

  3. g3=g^{3} =block_in_drawer: block xyzxyz-position satisfies the env’s _is_in_drawer predicate.

  4. g4=g^{4} =close_drawer: ∣xslide∣<0.04\lvert x_{\mathrm{slide}}\rvert< 0.04 (drawer back to nearly fully closed).

Final task success is doneg4_{g^{4}} which is coonsisitent with the origianl success.

Libero

We evaluate VADAR on 3 multi stage kitchen tasks from the LIBERO-10 [38] suite. These are: KITCHEN_SCENE3, KITCHEN_SCENE4, KITCHEN_SCENE6. We use the following subgoal decomposition for these tasks: KITCHEN_SCENE3 (turn on the stove and place the moka pot). BDDL goal: (Turnon flat_stove_1) ∧\land (On moka_pot_1 flat_stove_1_cook_region). Decomposed into 3 sequential subgoals:

  1. g1=g^{1} =stove_on: (turnon flat_stove_1).

  2. g2=g^{2} =moka_pot_grasped: gripper is grasping any geom containing moka_pot_1.

  3. g3=g^{3} =moka_pot_on_stove: (on moka_pot_1 flat_stove_1_cook_region).

Final task success is doneg3_{g^{3}}.

KITCHEN_SCENE4 (place black bowl in bottom drawer and close it). The bottom drawer of white_cabinet_1 is open at reset. BDDL goal: (Close white_cabinet_1_bottom_region) ∧\land (In akita_black_bowl_1 white_cabinet_1_bottom_region). Decomposed into 3 sequential subgoals:

  1. g1=g^{1} =bowl_grasped: gripper is grasping any geom containing akita_black_bowl.

  2. g2=g^{2} =bowl_in_drawer: (in akita_black_bowl_1 white_cabinet_1_bottom_region).

  3. g3=g^{3} =drawer_closed: (close white_cabinet_1_bottom_region).

Final task success is doneg3_{g^{3}}.

KITCHEN_SCENE6 (place mug in microwave and close it). The microwave is open at reset. BDDL goal: (In white_yellow_mug_1 microwave_1_heating_region) ∧\land (Close microwave_1). Decomposed into 3 sequential subgoals:

  1. g1=g^{1} =mug_grasped: gripper is grasping any geom containing white_yellow_mug.

  2. g2=g^{2} =mug_in_microwave: (in white_yellow_mug_1 microwave_1_heating_region).

  3. g3=g^{3} =microwave_closed: (close microwave_1).

Final task success is doneg3_{g^{3}}.

PushT-Obstacle

The multi-objective PUSHT-OBSTACLE task does not use a sequential subgoal decomposition; instead, the two “heads” correspond to two independent per-step signals defined in Appendix § 19:

  1. g1=g^{1} =task_done: block-goal coverage (intersection-over-union proxy) ≥0.95\ge0.95. Ratchets up.

  2. g2=g^{2} =collision_occurred: continuous proximity cost ct∈[0,1]c_t \in[0, 1] around the obstacle boundary (1 inside, decaying linearly across a 40-px soft margin).

Neither head is a prerequisite for the other, so GETWEIGHTING collapses to a constant weighting w=(1,λsafety)\mathbf{w} = (1, \lambda_{\mathrm{safety}}).

SuiteTask∣G∣\lvert G\rvertStructure
ROBOCASA365PRESOAKPAN5sequential chain
ROBOCASA365LOADDISHWASHER5sequential chain
ROBOCASA365ARRANGETEA6sequential chain
ROBOMIMICTRANSPORT7DAG, two parallel chains, joint handoff
ROBOMIMICTOOL-HANG4sequential chain
LIBEROKITCHEN_SCENE33sequential chain
LIBEROKITCHEN_SCENE43sequential chain
LIBEROKITCHEN_SCENE63sequential chain
OGBENCHscene-play-singletask-task44sequential chain
–PUSHT-OBSTACLE (multi-objective)2independent heads (no prerequisites)

Table 6. Summary of subgoal counts and dependency structure across benchmark tasks.

Hyperparameters

VADAR is tested with three off policy value based instantiations. Hyperparameters for each instantiation follows the best practices recommended in the corresponding original papers; we tune only the small set of task-level knobs (training-step budget, discount, BC regularization strength) needed to keep each baseline competitive on the task at hand, and we use the same setting across all baselines (Sparse, Dense, HGC, VADAR) on a given task so that comparisons are not confounded by tuning.

Common Hyperparameters

Most of our hyper parameters remain the same for all are experiments, unless explicitly mentioned. The detailed list of these are listed in Table 8.

HyperparameterValue
Optimization
critic learning rate3×10−43 \times10^{-4}
actor / BC learning rate3×10−43 \times10^{-4}
batch size256
replay buffer size2×1062 \times10^{6}
Networks
actor MLP hidden dims(512, 512, 512, 512)
critic MLP hidden dims(512, 512, 512, 512)
critic ensemble size MM10
ensemble aggregationmean
target EMA τ\tau (critic)0.05
target EMA τˉπ\bar{\tau}_{\pi} (actor)0.05
VADAR critic
critic shapeS×A→R∣G∣S \times A \to\mathbb{R}^{\lvert G\rvert}
advantage gating modegrandparent_future (VADAR-NF) / goal_future (VADAR-CF)
Online schedule
UTD (warmup, online)(1, 1)

Table 8. Hyperparameters shared across all tasks and backbones (OGPO, FQL, SAC)

EnvironmentMax episode length
ROBOCASA365 (all envs: ARRANGETEA, PRESOAKPAN, LOADDISHWASHER)1200
ROBOMIMIC TRANSPORT800
ROBOMIMIC TOOL-HANG1000
LIBERO KITCHEN_SCENE3/4/6500
OGBENCH scene-play-singletask-task4100
PUSHT-OBSTACLE300

Table 7. Maximum episode length per environment (used both for evaluation and as the truncation horizon during online finetuning)

OGPO specific hyperparameters

OGPO [52] is our flow-based backbone for the RoboCasa365 and RoboMimic suites. Table 9 lists the OGPO hyperparameters that are held fixed across all five OGPO-backed tasks; the per-task differences are then enumerated below.

HyperparameterValue
hh (action chunk length)8
q_warmup_steps20,000
flow_steps10
best_of_n8
critic target EMA τ\tau0.05

Table 9. OGPO hyperparameters held fixed across all five OGPO-backed tasks (PreSoakPan, LoadDishwasher, ArrangeTea, Transport, Tool-Hang)

The remaining hyper parameters vary per task. For the RoboCasa365 suite of tasks (PreSoakPan, LoadDishwasher, ArrangeTea), we use the bc data obtained from the official VLA checkpoint (NVIDIA et al., 2025). Since we deal with state-based environments and RoboCasa365 environments change the layout/scenes/fixtures for every environment reset, we fix them across resets to avoid partial observability in state-based settings. For RoboCasa365 environments, we set bc_coeff to 1 and offline_steps to 500, 000. We use around 150 demos collected from the VLA checkpoint. On RoboMimic, we do two sets of experiments: (1) with the full set of the offline demos present (2) reducing the offline demonstration set to a fraction of the original (the dataset.proportion parameter): only this fraction of the offline trajectories is used for BC pretraining. The full list of our hyper parameter settings for RoboMimic experiments on OGPO is present in Table 10.

FQL per-task hyperparameters

FQL [50] is the secondary flow-based backbone we use on RoboMimic Transport and Tool-Hang. Table 11 lists the values used. We use the same network architecture and learning rates as OGPO (i.e., Table 8); We do behavior cloning using all the available data a bit longer, due to FQL being less performative. The detailed list of hyperparameters used for FQL are listed in Table 11.

TRANSPORTTOOL-HANG
offline_steps1M500K
online_steps6M3M
q_warmup_steps50K50K
start_training10K10K
hh (chunk)88
γ\gamma0.9990.999
τˉπ\bar{\tau}_{\pi}0.0050.005
q_aggmeanmean
∣G∣|G|, structure7, DAG4, chain
group_num_samples3232
best_of_n, flow_steps8, 108, 10
αFQL\alpha_{\mathrm{FQL}} (distillation coeff.)10001000
bc_coeff1.01.0
clip_bc_threshold0.450.45

Table 11. Per-task hyperparameters for the FQL backbone on ROBOMIMIC. Values not listed match Table 8.

Learning with VLM-Provided Subgoal Labels

Goal-conditioned policies are computationally prohibitive with frontier model feedback

For our main experiments, the subgoal completion indicators done⁡g\operatorname{done}_{g} are evaluated using privileged simulator state. In real-world settings, these intermediate subgoal labels may instead be obtained from a VLM given visual observations and natural-language descriptions of the subgoals [20].

We test whether VADAR can learn when these intermediate training signals are provided by a VLM rather than by the simulator. On ROBOCASA365 PRESOAKPAN, we replace the oracle subgoal completion indicators with predictions from a VLM judge. The environment provides only the final task-success signal, while all intermediate subgoal supervision used by VADAR comes from the VLM. The policy and critic remain state-based, and the VLM is used only to generate training-time subgoal labels. Full details and the judge prompt are provided in Appendix § 18.

Figure 15 compares VADAR-NF trained with oracle and VLM-predicted subgoal completions. VLM supervision is somewhat less sample efficient early in training, but reaches a similar final success rate. These results show that VLMs can serve as training-time evaluators for the subgoal rewards used by VADAR, without introducing any VLM dependence at deployment time, while HGC needs such subgoal completion information during inference time, adding to the cost and latency of querying VLMs during deployment.

VADAR-NF on ROBOCASA365 PRESOAKPAN with the OGPO base algorithm, trained with oracle subgoal completions vs. with completions judged by a VLM

Figure 15. VADAR-NF on ROBOCASA365 PRESOAKPAN with the OGPO base algorithm, trained with oracle subgoal completions vs. with completions judged by a VLM (gemini-3.6-flash).

VLM Judge Prompt

Protocol. We use gemini-3.6-flash as the VLM judge for the experiment in Appendix § 17.§ 1. For every 8-step action chunk, the judge receives the frames at step offsets [0, 2, 5, 7] from all three cameras, together with the prompt below, and returns a JSON list of ∣G∣|G| booleans, which we use as the subgoal completion vector dt\mathbf{d}_{t} for the final step of the chunk. The prompt used for ROBOCASA365 PRESOAKPAN is shown below.

VLM judge prompt for ROBOCASA365 PRESOAKPAN

You are a strict robot-manipulation subgoal evaluator.

Task: PreSoakPan

The input is one 8-step action chunk. It contains 4 chronological timestamps at step offsets [0, 2, 5, 7]. Every timestamp has exactly three synchronized camera views. Timestamps are ordered oldest to newest.

Judge whether each subgoal's completion criterion is satisfied at the FINAL timestamp. Use the earlier timestamps as visual history to disambiguate grasping, motion, placement, and occlusion. A transient condition such as 'gripper is holding' must still be true at the final timestamp. Do not infer completion only from the expected task order, and do not require criteria that are not stated below.

Subgoals: 0. pan_grasped_lifted Completion criterion: The gripper is holding the pan and the pan is lifted off the counter. 1. pan_in_sink Completion criterion: The pan is resting inside the sink basin. 2. sponge_grasped_lifted Completion criterion: The gripper is holding the sponge and the sponge is lifted up. 3. sponge_in_pan_and_sink Completion criterion: The sponge is inside the sink and is placed on or in the pan. 4. water_on Completion criterion: The faucet water is running, the pan and sponge are in the sink, and the gripper has moved away from both objects.

Return only strict JSON in this form: {"done":[true,false,...]}. The list must contain exactly 5 booleans in the listed order.

Multi-Objective Optimization on PushT-Obstacle: Full Details

This appendix gives the full environment, algorithm, and training details for the multi-objective experiment in Appendix § 19.§ 1, together with the reward-shaping ablation that controls for the safety signal’s smoothness.

Generalization to multi-objective losses. We observe that VADAR does not require that the disaggregated value functions correspond to rewards associated with subgoals of a long-horizon task. Any collection of independently valuable rewards, such as safety constraints or velocity penalties, can be included in the fitting of the multi-objective critics. We validate the utility of VADAR for such multi-objective problems in Appendix § 19.§ 1.

red!50!black VaDar enables stable multi-objective optimization

Next, we test VADAR’s sample efficient multi-objective optimization capability. Specifically, we consider a setting where a reward function is decomposed into GG groups r=∑g∈Grgr = \sum_{g \in G} r^g. While classical RL uses an single critic for the aggregate reward rr, we propose VADAR for separate per-reward critics, which are aggregated after TD updates in the policy extraction step. Full reward and environment details are provided in Appendix § 19. Here we present a high level summary.

To test VADAR for multi-objective optimization, we design PUSHT-OBSTACLE, a variant of the popular env PushT [14] where a blue-dot pushes a gray “T” to the green goal configuration. We add a fixed circular obstacle and include both a sparse task reward for pushing the T-block to the goal pose rttaskr_t^{\mathrm{task}}, and a dense safety reward that penalizes proximity to the obstacle rtsafetyr_t^{\mathrm{safety}}. We instantiate VADAR with SAC and with G=2G = 2 heads (one per signal) and use a vector-valued multigoal critic (Qϕ)(Q_\phi), using a constant weighting for the aggregated critic in policy extraction: Qagg=Qϕ[task]+λsafetyQϕ[safety]Q_{\mathrm{agg}} = Q_\phi[\mathrm{task}] + \lambda_{\mathrm{safety}} Q_\phi[\mathrm{safety}]. We compare against standard RL with aggregated reward rt=rttask+λsafetyrtsafetyr_t = r_t^{\mathrm{task}} + \lambda_{\mathrm{safety}} r_t^{\mathrm{safety}} as a baseline (AGGREGATED-SAC). See Appendix § 19 and Appendix § 19.§ 6 for more details on the env and comparison of VADAR and AGGREGATED-SAC on PUSHT-OBSTACLE. Figure 16 shows that VADAR outperforms aggregated rewards, even

Success rate for PushT

Figure 16. Multi-Objective Optimization using VADAR on a variant of PushT created: PUSHT-OBSTACLE. Here the goal of the agent is to align the T with the given position whilst avoiding contact with the obstacle

when λsafety\lambda_{\mathrm{safety}} is chosen to maximize baseline performance. Moreover, we observe added training stability with VADAR safety, compared to AGGREGATED-SAC.

Environment

PUSHT-OBSTACLE extends the standard PushT [15] planar block-pushing task with a static circular obstacle of radius 30 px placed at (320, 220) in the 512×\times512 workspace. The agent (a circular pusher) must move a T-shaped block to a target pose while avoiding the obstacle. We use the keypoint observation variant (pusht-keypoints-v0) with a 5D state (xagent,yagent,xblock,yblock,θblock)(x_{\mathrm{agent}}, y_{\mathrm{agent}}, x_{\mathrm{block}}, y_{\mathrm{block}}, \theta_{\mathrm{block}}) and a 2D continuous action setting the target position of the PD-controlled pusher. Episodes terminate after 300 steps.

Depiction of the PUSHT-OBSTACLE environment

Figure 17. Depiction of our PUSHT-OBSTACLE env. The agent needs to place the T into the T shaped slot (measured by IoU ≥95%\ge95\%) whilst avoiding the red shaped obstacle placed in the vicinity of the final position.

Each environment step yields two scalar signals:

  • Task signal: a sparse coverage indicator rttask∈{−1,+5}r^{\mathrm{task}}_t \in\{-1, +5\}, equal to +5 when the block-goal IoU exceeds 95% and −1 otherwise.

  • Safety signal: a continuous soft-margin proximity cost ct∈[0,1]c_t \in[0, 1] that takes value 1 if either the pusher or the block penetrates the obstacle, decays linearly to 0 across a safety margin of 10 px around the obstacle boundary, and is 0 outside the margin. The corresponding per-step safety reward used by the critic is rtsafety=−5ctr^{\mathrm{safety}}_t = -5c_t.

VaDar Instantiation

We provide the full pseudocode in Algorithm 3. We instantiate VADAR on top of a SAC backbone with G=2G = 2: head 0 tracks task completion and head 1 tracks the safety cost. The vector critic Qϕ:S×A→R2Q_\phi: \mathcal{S} \times\mathcal{A} \to\mathbb{R}^2 is a 4-layer MLP of width 512 with layer norm, ensembled over 10 members and aggregated by mean. The actor is a tanh-squashed Gaussian over a 4-step action chunk, also a 4-layer MLP of width 512. The TD targets follow Eq. (4): the task head uses the standard reward-clipped VADAR target with cap cclip⋅rdone=15c_{\mathrm{clip}} \cdot r_{\mathrm{done}} = 15, while the safety head uses the backup −5ct+γHQ[safety](s′,a′)-5c_t + \gamma^H Q[\mathrm{safety}](s', a') with γ=0.99\gamma= 0.99 and chunk length H=4H = 4. Because neither subtask is a prerequisite for the other, the policy weighting GETWEIGHTING collapses to a constant w=(1,λsafety)\mathbf{w} = (1, \lambda_{\mathrm{safety}}), yielding the aggregated objective

Qagg(s,a)=Qϕ[task](s,a)+λsafetyQϕ[safety](s,a),Q_{\mathrm{agg}}(s, a) = Q_\phi[\mathrm{task}](s, a) + \lambda_{\mathrm{safety}} Q_\phi[\mathrm{safety}](s, a),

with λsafety=1.0\lambda_{\mathrm{safety}} = 1.0 in our experiments. (We also do a sweep over the values of λsafety\lambda_{\mathrm{safety}} Appendix § 19.§ 6). Apart from the vector-valued critic and this aggregation, the agent is parameterized as a standard Squashed Gaussian policy with learnable temperature.

Algorithm 3 VADAR for Multi-Objective RL (instantiated on PUSHT-OBSTACLE with G=2G=2).

  1. Parameters: discount γ\gamma; learning rate η\eta; safety weight λsafety\lambda_{\mathrm{safety}}; task-head TD clip cclipc_{\mathrm{clip}}.

  2. Initialize multi-objective critic Qϕ:S×A→R2Q_{\phi}:S\times A\to\mathbb{R}^{2} with heads {task, safety}; policy πθ:S→Δ(A)\pi_{\theta}:S\to\Delta(A); replay buffer B←∅\mathcal{B}\leftarrow\varnothing.

  3. for episode =1,2,…=1,2,\ldots do

  4. &nbsp;&nbsp;while episode not done do

  5. &nbsp;&nbsp;&nbsp;&nbsp;at←πθ(st)a_t\leftarrow\pi_{\theta}(s_t), transition st+1∼P(st,at)s_{t+1}\sim P(s_t,a_t)

  6. &nbsp;&nbsp;&nbsp;&nbsp;Read both signals: task indicator done⁡[task]t∈{0,1}\operatorname{done}[\mathrm{task}]_t\in\{0,1\} and continuous proximity cost ct∈[0,1]c_t\in[0,1].

  7. &nbsp;&nbsp;&nbsp;&nbsp;Form reward vector rt=(rsparse(done⁡[task]t),−5ct)r_t=(r_{\mathrm{sparse}}(\operatorname{done}[\mathrm{task}]_t),-5c_t).

  8. &nbsp;&nbsp;&nbsp;&nbsp;B←B∪{(st,at,st+1,rt,done⁡[task]t)}\mathcal{B}\leftarrow\mathcal{B}\cup\{(s_t,a_t,s_{t+1},r_t,\operatorname{done}[\mathrm{task}]_t)\}

  9. &nbsp;&nbsp;&nbsp;&nbsp;for number of updates do

  10. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;Sample (s,a,s′,r,done⁡[task])∼B(s,a,s',r,\operatorname{done}[\mathrm{task}])\sim\mathcal{B}, a′∼πθ(s′)a'\sim\pi_{\theta}(s')

  11. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;ytarget[task]=min⁡{r[task]+γQϕ[task](s′,a′),cclip⋅rdone}y^{\mathrm{target}}[\mathrm{task}]=\min\{r[\mathrm{task}]+\gamma Q_{\phi}[\mathrm{task}](s',a'),c_{\mathrm{clip}}\cdot r_{\mathrm{done}}\} ▷ reward clipping

  12. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;ytarget[safety]=r[safety]+γQϕ[safety](s′,a′)y^{\mathrm{target}}[\mathrm{safety}]=r[\mathrm{safety}]+\gamma Q_{\phi}[\mathrm{safety}](s',a')

  13. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;ϕ←ϕ−η∇ϕ∑g∈{task,safety}Lcritic(Qϕ[g]∣s,a,ytarget,g)\phi\leftarrow\phi-\eta\nabla_{\phi}\sum_{g\in\{\mathrm{task},\mathrm{safety}\}}\mathcal{L}_{\mathrm{critic}}(Q_{\phi}[g]\mid s,a,y^{\mathrm{target},g})

  14. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;Qagg(⋅,⋅)←Qϕ[task](⋅,⋅)+λsafetyQϕ[safety](⋅,⋅)Q_{\mathrm{agg}}(\cdot,\cdot)\leftarrow Q_{\phi}[\mathrm{task}](\cdot,\cdot)+\lambda_{\mathrm{safety}}Q_{\phi}[\mathrm{safety}](\cdot,\cdot) ▷ constant weighting

  15. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;θ←θ−η∇θLpolicy(πθ∣s,a,Qagg)\theta\leftarrow\theta-\eta\nabla_{\theta}\mathcal{L}_{\mathrm{policy}}(\pi_{\theta}\mid s,a,Q_{\mathrm{agg}})

Baselines

To isolate the contribution of value disaggregation from the contribution of the underlying reward terms, we compare against a dense-reward baseline that shares VADAR’s base algorithm (SAC), critic ensemble, network sizes, BC pretraining, and online step budget, and that consumes the exact same per-step task and safety signals but combines them into a single scalar reward fitted by a scalar critic. We call this AGGREGATED-SAC (Algorithm 4), and it combines the task and safety signals term-for-term as rt=rttask+λsafetyrtsafetyr_t=r_t^{\mathrm{task}}+\lambda_{\mathrm{safety}}r_t^{\mathrm{safety}}, with the same λsafety=1.0\lambda_{\mathrm{safety}}=1.0. This is the most direct baseline: it differs from VADAR only in collapsing the two signals before the Bellman backup rather than after.

Algorithm 4 Dense-reward shaping baseline for PUSHT-OBSTACLE (AGGREGATED-SAC).

  1. Parameters: discount γ\gamma; learning rate η\eta; safety weight λsafety\lambda_{\mathrm{safety}}.

  2. Initialize scalar critic Qϕ:S×A→RQ_{\phi}:S\times A\to\mathbb{R}; policy πθ:S→Δ(A)\pi_{\theta}:S\to\Delta(A); replay buffer B←∅\mathcal{B}\leftarrow\varnothing.

  3. for episode =1,2,…=1,2,\ldots do

  4. &nbsp;&nbsp;while episode not done do

  5. &nbsp;&nbsp;&nbsp;&nbsp;at←πθ(st)a_t\leftarrow\pi_{\theta}(s_t), transition st+1∼P(st,at)s_{t+1}\sim P(s_t,a_t)

  6. &nbsp;&nbsp;&nbsp;&nbsp;Read both signals: task indicator done⁡ttask∈{0,1}\operatorname{done}^{\mathrm{task}}_t\in\{0,1\} and continuous proximity cost ct∈[0,1]c_t\in[0,1]

  7. &nbsp;&nbsp;&nbsp;&nbsp;Combine before the Bellman backup into a single scalar reward:

  8. &nbsp;&nbsp;&nbsp;&nbsp;rt←rsparse(done⁡ttask)+λsafety(−5ct)r_t\leftarrow r_{\mathrm{sparse}}(\operatorname{done}^{\mathrm{task}}_t)+\lambda_{\mathrm{safety}}(-5c_t) ▷ shared signals with Algorithm 3; same λsafety\lambda_{\mathrm{safety}}

  9. &nbsp;&nbsp;&nbsp;&nbsp;B←B∪{(st,at,st+1,rt)}\mathcal{B}\leftarrow\mathcal{B}\cup\{(s_t,a_t,s_{t+1},r_t)\}

  10. &nbsp;&nbsp;&nbsp;&nbsp;for number of updates do

  11. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;Sample (s,a,s′,r)∼B(s,a,s',r)\sim\mathcal{B}, a′∼πθ(s′)a'\sim\pi_{\theta}(s')

  12. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;ytarget=r+γQϕ(s′,a′)y^{\mathrm{target}}=r+\gamma Q_{\phi}(s',a') ▷ single scalar TD target

  13. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;ϕ←ϕ−η∇ϕLcritic(Qϕ∣s,a,ytarget)\phi\leftarrow\phi-\eta\nabla_{\phi}\mathcal{L}_{\mathrm{critic}}(Q_{\phi}\mid s,a,y^{\mathrm{target}})

  14. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;θ←θ−η∇θLpolicy(πθ∣s,a,Qϕ)\theta\leftarrow\theta-\eta\nabla_{\theta}\mathcal{L}_{\mathrm{policy}}(\pi_{\theta}\mid s,a,Q_{\phi})

Training and Evaluation Protocol

All methods are pre-trained for 150,000 behavior-cloning steps on a fixed dataset of transitions collected from a previously trained SAC policy on the obstacle-free task, and then fine-tuned online for 800,000 environment steps. We measure the success rate in terms of success: block-goal coverage ≥95%\geq95\% and no collision with the obstacle occurs the episode.

Parameter Sweep for safety

We sweep over values of λsafety∈{0.1,1,5,10}\lambda_{\mathrm{safety}}\in\{0.1,1,5,10\} (Figure 18) and observe two trends. First, VADAR is far less sensitive to the choice of λsafety\lambda_{\mathrm{safety}}: it reaches comparable final success rates at λ=0.1\lambda=0.1 and λ=1\lambda=1, degrades only moderately at λ=5\lambda=5, and still makes meaningful progress at λ=10\lambda=10. AGGREGATED-SAC matches VADAR at small λ\lambda but collapses to near-zero success once λ≥5\lambda\geq5, since the safety penalty starts dominating the aggregate reward. Second, at any fixed λsafety\lambda_{\mathrm{safety}}, VADAR trains more stably than AGGREGATED-SAC, with narrower seed bands and fewer fluctuations during training. We conjecture that this is due to ease of composition at the Qfunction level: each VADAR head fits a λ\lambda-independent target, while AGGREGATED-SAC absorbs λ\lambda into the target itself.

Plots of success rate for different values of $\lambda_{\mathrm{safety}}$ comparing VADAR-multi and Agg-SAC on PushT-Obstacle

Figure 18. Plots of success rate for different values of λsafety\lambda_{\mathrm{safety}} comparing both VADAR and AGGREGATED-SAC on our PUSHT-OBSTACLE environment. Across all the values of λsafety\lambda_{\mathrm{safety}} that VADAR is more performative and stable than AGGREGATED-SAC

Test-Time Policy Steering

This appendix gives the full protocol, critic training details, and results for the test-time steering experiment in § 6.3.

Setup

We evaluate a critic by using it to select among candidate actions of a frozen policy. The policy is the BC-pretrained flow policy of OGPO, trained once and never updated. At every step, the policy proposes N∈{1,4,16,64}N \in\{1, 4, 16, 64\} candidate action chunks of length h=8h = 8, the critic scores each candidate, and the highest-valued chunk is executed. With N=1N = 1 the critic is not used, so this setting is the unsteered BC policy and is shared across critics and budgets, and we report a single N=1N = 1 baseline per task. The Sparse critic is scored by its ensemble mean (Appendix § 10). The VADAR critic is scored by the next-future aggregate Qagg(s,a)=∑g∈Gw[g]Qϕ(s,a,g)Q_{\mathrm{agg}}(s, a) = \sum_{g \in G} w[g] Q_{\phi}(s, a, g) of Eq. (3), where the goal-conditioned critic Qϕ(s,a,g)Q_{\phi}(s, a, g) is evaluated once per subgoal and w=GETWEIGHTING⁡(dt)\mathbf{w} = \operatorname{GETWEIGHTING}(\mathbf{d}_{t}) is computed from the environment’s subgoal completion indicators.

Critic Training

Starting from the frozen BC policy, only the critic is trained; the policy is never updated against the critic. The replay buffer is first seeded with 20,000 environment steps of frozen-policy rollouts. The critic is then updated with 4 gradient steps per environment step on batches drawn half from the offline demonstrations and half from frozen-policy rollouts, for up to 300,000 environment steps, and we evaluate checkpoints at 40,000, 100,000, 200,000 and 280,000 steps. The two critics share all hyperparameters (discount, value bounds, normalization and network sizes, Table 8) and differ only in the critic loss. We report the mean and standard deviation over 3 seeds, each evaluated on 96 episodes.

Results

Figure 19 and Table 12 give the full sweep, and we observe three trends. First, steering with the VADAR critic improves with the critic training budget at every NN on ARRANGETEA and TOOL-HANG, and at the largest budget on PRESOAKPAN. Second, the Sparse critic, trained on identical data with an identical budget, does not improve on the unsteered policy on ARRANGETEA at any budget or NN, is flat in both budget and NN on PRESOAKPAN, and on TOOL-HANG improves on the unsteered policy at small budgets but plateaus well below the VADAR critic. The only setting in which the Sparse critic is ahead is TOOL-HANG at the smallest budget, and the VADAR critic overtakes it from 100,000 steps onward. Third, steering degrades with large NN for both critics, since the maximum over more candidates increasingly selects out-of-distribution actions, and the degradation is milder for the VADAR critic.

Bar charts showing best-of-N steering for ArrangeTea, PreSoakPan, and Toolhang with Sparse and VADAR critics for N = 4, 16, and 64

Figure 19. Best-of-NN steering of the frozen OGPO BC policy with the Sparse and VADAR critics for N∈{4,16,64}N \in\{4, 16, 64\} (rows) on ROBOCASA365 ARRANGETEA, PRESOAKPAN and ROBOMIMIC TOOL-HANG (columns), as a function of the critic training budget. Error bars are one standard deviation over seeds; the dashed line is the unsteered BC policy.

TaskCritic budgetSparse critic N=4N = 4N=16N = 16N=64N = 64VADAR critic N=4N = 4N=16N = 16N=64N = 64
ARRANGETEAN=1N = 10.23 ± 0.04
40K0.17 ± 0.030.05 ± 0.040.01 ± 0.000.20 ± 0.070.10 ± 0.060.02 ± 0.02
100K0.24 ± 0.120.11 ± 0.150.05 ± 0.080.33 ± 0.100.23 ± 0.110.05 ± 0.05
200K0.27 ± 0.080.09 ± 0.030.05 ± 0.060.48 ± 0.070.30 ± 0.110.16 ± 0.10
280K0.23 ± 0.040.09 ± 0.010.02 ± 0.020.55 ± 0.050.38 ± 0.050.22 ± 0.06
PRESOAKPANN=1N = 10.22 ± 0.03
40K0.27 ± 0.020.29 ± 0.040.26 ± 0.040.250.280.28
100K0.29 ± 0.020.26 ± 0.050.26 ± 0.020.30 ± 0.100.30 ± 0.120.27 ± 0.16
200K0.22 ± 0.010.27 ± 0.030.20 ± 0.020.31 ± 0.020.31 ± 0.010.31 ± 0.04
280K0.28 ± 0.040.22 ± 0.050.25 ± 0.040.28 ± 0.050.32 ± 0.050.36 ± 0.02
TOOL-HANGN=1N = 10.32 ± 0.04
40K0.42 ± 0.010.23 ± 0.040.17 ± 0.010.25 ± 0.060.16 ± 0.050.03 ± 0.02
100K0.32 ± 0.080.21 ± 0.110.19 ± 0.100.47 ± 0.180.21 ± 0.110.12 ± 0.01
200K0.45 ± 0.120.28 ± 0.050.24 ± 0.120.62 ± 0.100.36 ± 0.120.21 ± 0.04
280K0.44 ± 0.170.38 ± 0.080.32 ± 0.050.77 ± 0.050.53 ± 0.150.40 ± 0.11

Table 12. Success rate of best-of-NN steering of the frozen OGPO BC policy with the Sparse and VADAR critics, as a function of the critic training budget (mean ± std over seeds). The N=1N = 1 row is the unsteered BC policy, and the best cell in each row is highlighted.

To understand this gap, we roll out the frozen policy and, at each step, score the same 64 candidate action chunks with both critics. Figure 20 shows the two critics at one such step on TOOL-HANG, just before subgoal g2g^{2} (frame_inserted, Appendix § 16) is completed. The insertion is the step that demands the most precision, and hence one of its most important decision steps. We take the plane spanned by the two principal directions of the 64 candidates, evaluate each critic on a grid of actions in this plane, and plot its contours together with the direction of its action-gradient. As both critics predict returns in the same units, we use the same contour spacing for both, so that a critic with little variation over the plane shows few contours. The VADAR critic varies by an order of magnitude more than the Sparse critic over this plane. The executed chunk is the action taken by the successful trajectory, i.e. the correct action at this step. The VADAR critic identifies it: it assigns its highest values to the candidates near the executed chunk, and its gradient points towards the executed chunk from across the candidate set. The Sparse critic does not: it assigns nearly the same value to all candidates, and its gradient does not point towards the executed chunk. This suggests that VADAR learns

Q-value landscapes for the VADAR and Sparse critics

Figure 20. Q-values of the VADAR (middle) and Sparse (right, labelled OGPO in the panel) critics, each trained with the largest critic budget of 280K steps, at a single step of ROBOMIMIC TOOL-HANG before the frame is inserted into the stand (left), plotted over the plane of the first two principal components of the BC policy’s 64 candidate action chunks. Dots are the candidates and the green square is the executed chunk; streamlines follow the gradient of QQ with respect to the action, and the contour spacing is the same in both panels.

a better critic: one that correctly ranks the actions at the step that decides the task, whilst the Sparse critic trained on the same data cannot tell them apart. We conjecture that this difference in critic quality is what drives VADAR’s advantage in both the online RL experiments, where the critic supervises policy extraction, and the steering experiments, where it ranks candidate actions directly.

Last-QQ Extraction: Full Benchmark Results

Figure 21 gives the full comparison accompanying Section 6.2.

Success-rate curves across all benchmarks and instantiations

Figure 21. Success rate of VADARLAST-Q compared with VADAR-NF and Sparse across all benchmarks and instantiations.

Connections to Successor Features and Reward Shaping

VADAR’s aggregation step (Eq. (3)) is structurally similar to a success feature decomposition of a QQ-function. Classical success features assume reward is linear in a feature map ϕ:S×A→R∣G∣\phi: S \times A \to\mathbb{R}^{|G|} ([16, 7]), r(s,a)=⟨w,ϕ(s,a)⟩r(s,a) = \langle w,\phi(s,a)\rangle for some fixed w∈R∣G∣w \in\mathbb{R}^{|G|}. The QQ-function then inherits the same decomposition Qπ(s,a)=⟨w,ψπ(s,a)⟩Q^\pi(s,a) = \langle w,\psi^\pi(s,a)\rangle, where ψ\psi is the successor feature, which represents the expected discounted sum of returns under policy π\pi. Crucially as motivated by Barreto et al. [7], this allows efficient policy extraction from arbitrary reward weightings without retraining policies across tasks.

VADAR instantiates the reward features ϕ(s,a)\phi(s,a) as per-subgoal rewards ϕg(s,a)=rg(s,a)\phi_g(s,a)=r_g(s,a) so the successor features are precisely the per-subgoal Q-values Qϕg(s,a)=ψgπ(s,a)Q_{\phi_g}(s,a)=\psi_g^\pi(s,a). The weight vector w(s)=GETWEIGHTING⁡(d(s))\mathbf{w}(s)=\operatorname{GETWEIGHTING}(d(s)) is a state-dependent current subgoal completion vector. This allows VADAR to recover sufficient statistics for evaluating the entire ancestral tree of weighted objectives in a single multi-step RL training run with w(s)\mathbf{w}(s) as the contraction operator.

Fixed-Weight Value Aggregation and Reward Shaping

For fixed weights and true per-goal Q-functions under a common policy, value aggregation is equivalent to evaluating an aggregated reward. Consider family of per-goal MDPs Mg=(S,A,P0,P,Rg,γ)M_g=(S,A,P_0,P,R_g,\gamma) which share common dynamics and discount factor, but differ in their reward RgR_g. For example, in the goal-conditioned setting described in § 4, choosing Rg(s,a)←rsparse(done⁡g(s))R_g(s,a)\leftarrow r_{\mathrm{sparse}}(\operatorname{done}_g(s)) yields that MgM_g is the MDP where rewards are associated with subgoal gg. In principle, however, the theory accomodates any choices of rewards. Finally, we let Qπ,MgQ^{\pi,M_g} denote the true Q-function (as described in the preliminaries) associated with π\pi under MDP MgM_g.

We now show that the Q-function achieved by aggregating per-goal Q-functions is equivalent to the Q-function of a policy in an MDP whose reward aggregates per-goal rewards:

Proposition O.1. Fix a weighting vector w∈RG\mathbf{w}\in\mathbb{R}^{G}. Then, for any goal-unconditioned policy π:S→Δ(A)\pi:S\to\Delta(A), and any s,as,a, we have the equality

Qaggπ(s,a):=∑gw[g]Qπ,Mg(s,a)=Eπ[∑t≥0γtRw(st,at)∣(s0,a0)=(s,a)]=Qπ,Mw(s,a),(30)Q_{\mathrm{agg}}^\pi(s,a):=\sum_g w[g]Q^{\pi,M_g}(s,a)=\mathbb{E}^{\pi}\left[\sum_{t\geq0}\gamma^tR_{\mathbf{w}}(s_t,a_t)\mid(s_0,a_0)=(s,a)\right]=Q^{\pi,M_{\mathbf{w}}}(s,a), \tag*{(30)}

where above Rw:=∑gw[g]RgR_{\mathbf{w}}:=\sum_g w[g]R_g and MwM_{\mathbf{w}} is the MDP defined by (S,A,P0,P,Rw,γ)(S,A,P_0,P,R_{\mathbf{w}},\gamma)

Proof. Observe that for any MDP MgM_g with reward RgR_g, we have

Qπ,Mg(s,a)=Eπ[∑t≥0γtRg(st,at)∣(s0,a0)=(s,a)],Q^{\pi,M_g}(s,a)=\mathbb{E}^{\pi}\left[\sum_{t\geq0}\gamma^tR_g(s_t,a_t)\mid(s_0,a_0)=(s,a)\right],

where Eπ\mathbb{E}^{\pi} is expectation under the dynamics of the MDP and policy π\pi. As the dynamics are shared between all per-goal MDPs, we may leverage linearity of expectation to commute the expectation with the weighted summation.

This identity applies to fixed weights and exact Q-functions. It does not establish equivalence between VADAR’s state-dependent weighting and clipped, approximate critic updates and learning a scalar critic on an aggregated reward. The experiments in Section 6.2 and Appendix § 23.§ 1 distinguish reward aggregation from the benefits of dense critic supervision.

Reward shaping does not explain Vadar’s success

It might be tempting to attribute VADAR’s success to reward shaping. However, in Section 6.2, we find that even using the final subgoal critic can lead to strong performance in many benchmarks. We conclude that VADAR’s benefits come as much from representation learning induced by denser critic supervision as they do from explicit reward shaping during policy extraction.

Subgoal Reward Shaping Does Not Reliably Enable Long-Horizon Learning

Appendix O establishes that, for fixed weights and true Q-functions under a common policy, a weighted sum of subgoal values equals the value of the correspondingly weighted reward. This identity does not guarantee equivalent learning with state-dependent weights or clipped, approximate critics. This suggests a natural alternative to VADAR: aggregate the per-subgoal rewards before the Bellman backup, and train a standard scalar critic and policy on the resulting shaped reward. We call this variant VADAR-SHAPED.

Figure 22 shows that this equivalence does not translate to learning in practice. VADAR-SHAPED fails to learn on the harder ROBOCASA365 tasks and is substantially less sample efficient than VADAR-NF elsewhere. Since VADAR-SHAPED combines the subgoal rewards before the Bellman update, the critic is trained on single scalar target rather than learning each subgoal value separately. Together with the VADARLAST-Q results, this suggests that VADAR’s gains come from multi-resolution temporal supervision, rather than from reward shaping alone.

Subgoal reward shaping ablation on Robocasa LoadDishwasher, Robocasa ArrangeTea, and Robomimic Toolhang

Figure 22. Subgoal reward shaping ablation. VADAR-NF against VADAR-SHAPED, which sums the per-subgoal rewards into a single shaped reward before the Bellman backup and trains a standard scalar critic on it, on ROBOCASA365 with OGPO and ROBOMIMIC TOOL-HANG with FQL.

Alternative critic parameterizations yield limited additional benefit

An alternative to VADAR’s goal-conditioned critic Qϕ(s,a,g)Q_{\phi}(s, a, g), which is evaluated once per subgoal to form the vector of values used for policy extraction, is a vector-valued critic Qϕ:S×A→R∣G∣Q_{\phi}: S \times A \to\mathbb{R}^{|G|} that predicts all subgoal values in a single forward pass. We call this variant VECTOR-VADAR.

Figure 23 shows that it performs roughly the same as VADAR across tasks and weightings. This is consistent with the previous subsections: both critics are supervised on every subgoal at every transition, and it is this dense supervision, together with non-myopic policy extraction, that accounts for VADAR’s performance rather than the particular critic parameterization.

Critic parameterization ablation on Robocasa PreSoakPan, Robocasa LoadDishwasher, Robocasa ArrangeTea, and Robomimic Toolhang

Figure 23. Critic parameterization ablation. VADAR with its goal-conditioned critic S×A×G→RS \times A \times G \to\mathbb{R} (blue) against VECTOR-VADAR, which uses a vector-valued critic S×A→R∣G∣S \times A \to\mathbb{R}^{|G|} (red), under next-future (solid) and current-future (dotted) weightings on ROBOCASA365 with OGPO and ROBOMIMIC TOOL-HANG with FQL.

A second alternative is an iterative critic, which refines its subgoal values over several passes and can be viewed as simulating a Bellman update inside the critic (Appendix § 25). We instantiate this with MIQ (Appendix § 25.§ 2) and again find performance on par with VADAR’s monolithic critic, with the iterative critic training somewhat more stably (Appendix § 25.2.5).

Combining VaDAR with Iterative Critics

VaDAR is compatible with critics that estimate values through iterative computation [1]. Here we explore how iterative computation and multi-goal prediction could implement an implicit Bellman update during value prediction. Empirically, iterative critics provide limited additional benefit over the monolithic critic (Appendix § 24).

Simulating the Bellman Update via Iterative Critic Parametrizations

Standard RL critics compute Q-values in a single monolithic forward pass. In the context of VaDAR, a monolithic critic produces all ∣G∣|G| heads in a single shot, so the subgoal heads never communicate at inference time. Here we argue that combining iterative critic parameterizations with VaDAR can be viewed as implementing a Bellman update at inference time.

An iterative VaDAR critic proceeds in steps j=0,…,Niterj = 0, \ldots, N_{\mathrm{iter}}, producing an estimates q^j∈RG\hat{q}^{j} \in\mathbb{R}^{G} of the critic values per step. At each step, the critic updates its estimate q^j+1←Qϕ(s,a,q^j,j)\hat{q}^{j+1} \leftarrow Q_{\phi}(s, a, \hat{q}^{j}, j). Therefore, each updated qk+1q^{k+1} can depend on previous estimates of critic values qkq^{k}. Informally, this computation graph can express belief-propagation updates across critic time steps, such as the classical Bellman update [61] which defines the Q-values. We depict this intuition in Figure 24. For training an iterative critic, we introduce a two-step minimal iterative Q-network (MIQ) inspired by Pan et al. [48] described in Appendix § 25.§ 2, which we anecdotally find more stable to train than flow-based critics [1].

Iterative Critic Prediction

Figure 24. Iterative Critic Prediction

The MIQ Critic

VaDAR’s vector critic synergizes with iterative critic parameterizations: each iteration can propagate information across subgoal heads, so the second iteration’s prediction for one subgoal can be conditioned on the first iteration’s prediction for the others. We instantiate this with MIQ, a critic-side adaptation of the Minimal Iterative Policy of Pan et al. [48] that retains the two-step iteration with t⋆=0.9t^{\star} = 0.9.

Architecture

A MIQ critic is a single MLP Qϕ:S×A×R×[0,1]→RQ_{\phi}: \mathcal{S} \times\mathcal{A} \times\mathbb{R} \times[0, 1] \to\mathbb{R} that takes a state-action pair (s,a)(s, a), a scalar zz that plays the role of the noise variable in a flow-matching linear interpolant, and a time index t∈[0,1]t \in[0, 1]. To recall the flow-matching analogy: a flow model is trained to map the linear interpolant It=tx+(1−t)zI_t = t x + (1-t) z between Gaussian noise zz and a clean target xx at time t∈[0,1]t \in[0, 1] back to xx, so that at t=0t = 0 the input is pure noise and as t→1t \to1 it approaches the target. MIP [48] retains this interpolant but supervises the network with the same target twice, at t=0t = 0 and at t=t⋆t = t^{\star} (with t⋆=0.9t^{\star} = 0.9), using a regression loss instead of a velocity-matching loss. MIQ is the same construction with the supervision target xx instantiated as the scalar Q-target ytargety^{\mathrm{target}}: the flow-matching interpolant is replaced by an interpolant in Q-value space, and z∼Unif⁡[−σ,σ]z \sim\operatorname{Unif}[-\sigma,\sigma] is the noise variable that lives on the noisy end of that interpolant. At inference, MIQ runs two autoregressive steps:

q^0←Qϕ(s,a,z,t=0),q^1←Qϕ(s,a,q^0,σ/10,t⋆),(31)\hat{q}^{0} \leftarrow Q_{\phi}(s, a, z, t=0), \qquad\hat{q}^{1} \leftarrow Q_{\phi}(s, a, \hat{q}^{0}, \sigma/10, t^{\star}), \tag*{(31)}

i.e., the second iteration substitutes its own iteration-zero output q^0\hat{q}^{0} in place of an interpolated input, since q^0\hat{q}^{0} has been trained to approximate ytargety^{\mathrm{target}}.

Training Objective

Following Pan et al. [48], Eq. (4.4), MIQ supervises both iterations against the standard scalar TD target ytarget=r+γQtarget(s′,a′)y^{\mathrm{target}} = r + \gamma Q_{\mathrm{target}}(s', a'). To train the iteration-one mapping, we cannot feed the network the iteration-zero output (it is a moving target during training); instead, we feed a noise-corrupted version of the supervision signal itself. Concretely, we sample z∼Unif⁡[−σ,σ]z \sim\operatorname{Unif}[-\sigma,\sigma] and form the linear interpolant

zt⋆=(1−t⋆)z+ytarget,(32)z_{t^{\star}} = (1-t^{\star})z + y^{\mathrm{target}}, \tag*{(32)}

which is a noisy copy of ytargety^{\mathrm{target}} (smaller noise the larger t⋆t^{\star} is, hence t⋆t^{\star} near 1). The MIQ critic loss for a single ensemble member is

LcriticMIQ(Qϕ∣s,a,ytarget,z)=(Qϕ(s,a,z,0)−ytarget)2+(Qϕ(s,a,zt⋆,t⋆)−ytarget)2.(33)L_{\mathrm{critic}}^{\mathrm{MIQ}}(Q_{\phi} \mid s, a, y^{\mathrm{target}}, z) = \left(Q_{\phi}(s, a, z, 0) - y^{\mathrm{target}}\right)^{2} + \left(Q_{\phi}(s, a, z_{t^{\star}}, t^{\star}) - y^{\mathrm{target}}\right)^{2}. \tag*{(33)}

The first term trains the iteration-zero pathway to predict ytargety^{\mathrm{target}} from a fully random zz; the second term trains the iteration-one pathway to denoise a near-correct estimate zt⋆z_{t}^{\star} back to ytargety^{\mathrm{target}}. At inference we substitute the iteration-zero output q^0\hat{q}^{0} for zt⋆z_{t}^{\star}, which is well-behaved because the iteration-zero pathway has already been trained to produce a good approximation of ytargety^{\mathrm{target}}.

Ensembling

We use an explicit ensemble of MM independently parameterized MIQ networks Qϕ1,…,QϕMQ_{\phi_1},\ldots,Q_{\phi_M} (ValueMIPEnsemble in our codebase). At each update, a single noise draw zz is shared across the ensemble; each member produces its own q^0,(i)\hat{q}^{0,(i)} at iteration zero and uses that member’s own q^0,(i)\hat{q}^{0,(i)} as the iteration-one input. Aggregation across the ensemble (Eq. (B.2)) is unchanged from Appendix § 10: we average the iteration-one predictions to form QtargQ_{\mathrm{targ}}.

MIQ with VaDAR

We have seen how MIQ can be used as a general critic. Adapting it to our VaDAR framework requires minimal modifications. The scalar Qϕ:S×A×R×[0,1]→RQ_{\phi}:S \times A \times\mathbb{R} \times[0,1] \to\mathbb{R} now becomes a vector critic: Qϕ:S×A×R∣G∣×[0,1]→R∣G∣Q_{\phi}:S \times A \times\mathbb{R}^{|G|} \times[0,1] \to\mathbb{R}^{|G|}. The noise variable z∼Unif⁡[−σ,σ]∣G∣z \sim\operatorname{Unif}[-\sigma,\sigma]^{|G|} also becomes a ∣G∣|G| dimensional vector. And the TD target is the vector ytarget∈R∣G∣y^{\mathrm{target}} \in\mathbb{R}^{|G|} from Eq. (4), and Eq. (Q.1) becomes

q^0←Qϕ(s,a,z,0),q^1←Qϕ(s,a,q^0,t⋆).\hat{q}^{0} \leftarrow Q_{\phi}(s,a,z,0), \qquad\hat{q}^{1} \leftarrow Q_{\phi}(s,a,\hat{q}^{0},t^{\star}).

Because the second pass’ input is the full vector q^0\hat{q}^{0}, head gg’s second estimate depends on every head’s previous estimate.

Policy Extraction via MIQ + VaDAR

To validate the hypothesis that iterative critic parameterizations learn implicit Bellman updates over the subgoal DAG, we test MIQ+VaDAR on LIBERO (∣G∣=3|G| = 3) with SAC and ROBOMIMIC TRANSPORT (∣G∣=7|G| = 7) with OGPO as the backbone. We compare the VaDAR’s monolithic multi-head critic against MIQ with varying Q-function update-to-data (UTD) ratios.

Results. Across the four panels in Figure 25, success rates with MIQ are broadly comparable to the monolithic critic, with a minor improvement in policy extraction for the higher critic UTD (q=4q = 4) ablation.

MIQ+VaDAR versus monolithic vector critics across four environments

Figure 25. MIQ+VaDAR (orange/purple) vs. monolithic vector critics (red/blue) at critic UTD q=1q = 1 (solid) and q=4q = 4 (dashed); the policy update frequency is held fixed across all runs. The largest MIQ effect is on OGPO TRANSPORT, where the monolithic critic degrades sharply at q=4q = 4 and MIQ restores parity with q=1q = 1.

The more informative differences show up on the critic itself: in Appendix § 25.§ 2 (Figure 26), per-subgoal TD residuals on LIBERO KITCHEN_SCENE6 drop under MIQ across all three subtasks and stay flat throughout training, while the monolithic critic’s TD error drifts upward late in training. The mean-QQ trajectory under MIQ is more stable as hypothesized due to the implicit message passing down the subgoal DAG. We posit this being more beneficial to use-cases where critic stability even at subgoal level is a large concern.

TD error, success rate, and mean-Q curves for Kitchen Task 6

Figure 26. We compare monolithic VADAR critics with MIQ+VADAR at critic UTD q∈{1,4}q \in\{1, 4\}, holding the actor update frequency fixed. MIQ substantially reduces total and per-subgoal TD residuals and produces lower-variance mean-QQ estimates, while maintaining comparable or slightly improved success rates.

This view connects to the VADARLAST-Q ablation of Section 6.2, where policy extraction uses only the final-subgoal value and approaches VADAR-NF on several tasks. As discussed in § 6.4, these results are consistent with intermediate targets improving the shared representations used for final-subgoal prediction. MIQ introduces an explicit dependence between successive value estimates: its second pass conditions every subgoal prediction on the full first-pass vector q^\hat{\mathbf{q}}. The comparison therefore tests the additional benefit of iterative refinement beyond shared subgoal supervision; comparable success rates do not establish that the non-iterative critic performs the same internal computation.

References

  1. [1]Bhavya Agrawalla, Michal Nauman, Khush Agrawal, and Aviral Kumar. floq: Training critics via flow-matching for scaling compute in value-based rl. arXiv preprint arXiv:2509.06863, 2025.arxiv.org/abs/2509.06863
  2. [2]Michael S Albergo, Nicholas M Boffi, and Eric Vanden-Eijnden. Stochastic interpolants: A unifying framework for flows and diffusions. arXiv preprint arXiv:2303.08797, 2023.arxiv.org/abs/2303.08797
  3. [3]Dario Amodei, Chris Olah, Jacob Steinhardt, Paul Christiano, John Schulman, and Dan Mané. Concrete problems in ai safety, 2016. URL https://arxiv.org/abs/1606.06565.
  4. [4]Marcin Andrychowicz, Filip Wolski, Alex Ray, Jonas Schneider, Rachel Fong, Peter Welinder, Bob McGrew, Josh Tobin, Pieter Abbeel, and Wojciech Zaremba. Hindsight experience replay, 2018. URL https://arxiv.org/abs/1707.01495.
  5. [5]Pierre-Luc Bacon, Jean Harb, and Doina Precup. The option-critic architecture, 2016. URL https://arxiv.org/abs/1609.05140.
  6. [6]Philip J Ball, Laura Smith, Ilya Kostrikov, and Sergey Levine. Efficient online reinforcement learning with offline data. In International Conference on Machine Learning, pp. 1577–1594. PMLR, 2023.arxiv.org/abs/2302.02948
  7. [7]André Barreto, Will Dabney, Rémi Munos, Jonathan J Hunt, Tom Schaul, Hado P Van Hasselt, and David Silver. Successor features for transfer in reinforcement learning. Advances in neural information processing systems, 30, 2017.arxiv.org/abs/1606.05312
  8. [8]Kevin Black, Mitsuhiko Nakamoto, Pranav Atreya, Homer Walke, Chelsea Finn, Aviral Kumar, and Sergey Levine. Zero-shot robotic manipulation with pretrained image-editing diffusion models. In International Conference on Learning Representations (ICLR), 2024.arxiv.org/abs/2310.10639
  9. [9]Ralph Allan Bradley and Milton E. Terry. Rank analysis of incomplete block designs: I. the method of paired comparisons. Biometrika, 39(3/4):324–345, 1952. ISSN 00063444, 14643510. URL http://www.jstor.org/stable/2334042.DOI
  10. [10]Daniel Brown, Wonjoon Goo, Prabhat Nagarajan, and Scott Niekum. Extrapolating beyond suboptimal demonstrations via inverse reinforcement learning from observations. In Kamalika Chaudhuri and Ruslan Salakhutdinov (eds.), Proceedings of the 36th International Conference on Machine Learning, volume 97 of Proceedings of Machine Learning Research, pp. 783–792. PMLR, 09–15 Jun 2019. URL https://proceedings.mlr.press/v97/brown19a.html.arxiv.org/abs/1904.06387
  11. [11]Thomas P Cannon and Özgür Simsek. Accelerating task generalisation with multi-level skill hierarchies, 2025. URL https://arxiv.org/abs/2411.02998.
  12. [12]Kaiyuan Chen, Shuangyu Xie, Zehan Ma, Pannag R Sanketi, and Ken Goldberg. Robo2VLM: Improving visual question answering using large-scale robot manipulation data. In The Thirty-ninth Annual Conference on Neural Information Processing Systems Datasets and Benchmarks Track, 2026. URL https://openreview.net/forum?id=OChorZcZnY.
  13. [13]Xinyue Chen, Che Wang, Zijian Zhou, and Keith Ross. Randomized ensembled double q-learning: Learning fast without a model. arXiv preprint arXiv:2101.05982, 2021.arxiv.org/abs/2101.05982
  14. [14]Cheng Chi, Siyuan Feng, Yilun Du, Zhenjia Xu, Eric Cousineau, Benjamin Burchfiel, and Shuran Song. Diffusion policy: Visuomotor policy learning via action diffusion. arXiv preprint arXiv:2303.04137, 2023.arxiv.org/abs/2303.04137
  15. [15]Cheng Chi, Zhenjia Xu, Siyuan Feng, Eric Cousineau, Yilun Du, Benjamin Burchfiel, Russ Tedrake, and Shuran Song. Diffusion policy: Visuomotor policy learning via action diffusion. The International Journal of Robotics Research, 2024.
  16. [16]Peter Dayan. Improving generalization for temporal difference learning: The successor representation. Neural computation, 5(4):613–624, 1993.DOI
  17. [17]Peter Dayan and Geoffrey E Hinton. Feudal reinforcement learning. In S. Hanson, J. Cowan, and C. Giles (eds.), Advances in Neural Information Processing Systems, volume 5. Morgan-Kaufmann, 1992. URL https://proceedings.neurips.cc/paper_files/paper/1992/file/d14220ee66aeec73c4903885428ec4c-Paper.pdf.
  18. [18]Thomas G. Dietterich. Hierarchical reinforcement learning with the maxq value function decomposition, 1999. URL https://arxiv.org/abs/cs/9905014.
  19. [19]Perry Dong, Qiyang Li, Dorsa Sadigh, and Chelsea Finn. Expo: Stable reinforcement learning with expressive policies. arXiv preprint arXiv:2507.07986, 2025.arxiv.org/abs/2507.07986
  20. [20]Yuqing Du, Ksenia Konyushkova, Misha Denil, Akhil Raju, Jessica Landon, Felix Hill, Nando De Freitas, and Serkan Cabi. Vision-language models as success detectors. arXiv preprint arXiv:2303.07280, 2023.arxiv.org/abs/2303.07280
  21. [21]Joshua B. Evans and Özgür Şimşek. Creating multi-level skill hierarchies in reinforcement learning, 2024. URL https://arxiv.org/abs/2306.09980.
  22. [22]Roy Fox, Sanjay Krishnan, Ion Stoica, and Ken Goldberg. Multi-level discovery of deep options, 2017. URL https://arxiv.org/abs/1703.08294.
  23. [23]Scott Fujimoto, Herke Hoof, and David Meger. Addressing function approximation error in actor-critic methods. In International conference on machine learning, pp. 1587–1596. PMLR, 2018.
  24. [24]Scott Fujimoto, Pierluca D’Oro, Amy Zhang, Yuandong Tian, and Michael Rabbat. Towards general-purpose model-free reinforcement learning. In International Conference on Learning Representations, volume 2025, pp. 83509–83552, 2025.arxiv.org/abs/2501.16142
  25. [25]Caelan Reed Garrett, Rohan Chitnis, Rachel Holladay, Beomjoon Kim, Tom Silver, Leslie Pack Kaelbling, and Tomás Lozano-Pérez. Integrated task and motion planning. Annual review of control, robotics, and autonomous systems, 4(1):265–293, 2021.
  26. [26]Abhishek Gupta, Vikash Kumar, Corey Lynch, Sergey Levine, and Karol Hausman. Relay policy learning: Solving long-horizon tasks via imitation and reinforcement learning. arXiv preprint arXiv:1910.11956, 2019.arxiv.org/abs/1910.11956
  27. [27]Abhishek Gupta, Vikash Kumar, Corey Lynch, Sergey Levine, and Karol Hausman. Relay policy learning: Solving long-horizon tasks via imitation and reinforcement learning. In Leslie Pack Kaelbling, Danika Kragic, and Komei Sugiura (eds.), Proceedings of the Conference on Robot Learning, volume 100 of Proceedings of Machine Learning Research, pp. 1025–1037. PMLR, 30 Oct–01 Nov 2020. URL https://proceedings.mlr.press/v100/gupta20a.html.arxiv.org/abs/1910.11956
  28. [28]Tuomas Haarnoja, Aurick Zhou, Pieter Abbeel, and Sergey Levine. Soft actor-critic: Off-policy maximum entropy deep reinforcement learning with a stochastic actor. In International conference on machine learning, pp. 1861–1870. Pmlr, 2018a.
  29. [29]Tuomas Haarnoja, Aurick Zhou, Kristian Hartikainen, George Tucker, Sehoon Ha, Jie Tan, Vikash Kumar, Henry Zhu, Abhishek Gupta, Pieter Abbeel, et al. Soft actor-critic algorithms and applications. arXiv preprint arXiv:1812.05905, 2018b.arxiv.org/abs/1812.05905
  30. [30]Leslie Pack Kaelbling. Learning to achieve goals. In International Joint Conference on Artificial Intelligence, 1993. URL https://api.semanticscholar.org/CorpusID:5538688.
  31. [31]Leslie Pack Kaelbling and Tomás Lozano-Pérez. Hierarchical task and motion planning in the now. In 2011 IEEE international conference on robotics and automation, pp. 1470–1477. IEEE, 2011.
  32. [32]Victoria Krakovna, Jonathan Uesato, Vladimir Mikulik, Matthew Rahtz, Tom Everitt, Ramana Kumar, Zachary Kenton, Jan Leike, and Shane Legg. Specification gaming: The flip side of ai ingenuity. https://deepmind.google/blog/specification-gaming-the-flip-side-of-ai-ingenuity/, 2020. DeepMind Blog.
  33. [33]Kimin Lee, Laura Smith, Anca Dragan, and Pieter Abbeel. B-pref: Benchmarking preference-based reinforcement learning. In Thirty-fifth Conference on Neural Information Processing Systems Datasets and Benchmarks Track (Round 1), 2021. URL https://openreview.net/forum?id=ps95-mkHF_.arxiv.org/abs/2111.03026
  34. [34]Andrew Levy, George Konidaris, Robert Platt, and Kate Saenko. Learning multi-level hierarchies with hindsight, 2019. URL https://arxiv.org/abs/1712.00948.
  35. [35]Qiyang Li, Zhiyuan Zhou, and Sergey Levine. Reinforcement learning with action chunking. arXiv preprint arXiv:2507.07969, 2025.arxiv.org/abs/2507.07969
  36. [36]Qiyang Li, Zhiyuan Paul Zhou, and Sergey Levine. Reinforcement learning with action chunking. Advances in Neural Information Processing Systems, 38:55518–55553, 2026.arxiv.org/abs/2507.07969
  37. [37]Anthony Liang, Yigit Korkmaz, Jiahui Zhang, Minyoung Hwang, Abrar Anwar, Sidhant Kaushik, Aditya Shah, Alex S Huang, Luke Zettlemoyer, Dieter Fox, et al. Robometer: Scaling general-purpose robotic reward models via trajectory comparisons. arXiv preprint arXiv:2603.02115, 2026.arxiv.org/abs/2603.02115
  38. [38]Bo Liu, Yifeng Zhu, Chongkai Gao, Yihao Feng, Qiang Liu, Yuke Zhu, and Peter Stone. Libero: Benchmarking knowledge transfer for lifelong robot learning, 2023a. URL https://arxiv.org/abs/2306.03310.
  39. [39]Songtao Liu, Zhengkai Tu, Minkai Xu, Zuobai Zhang, Lu Lin, Rex Ying, Jian Tang, Peilin Zhao, and Dinghao Wu. Fusionretro: molecule representation fusion via in-context learning for retrosynthetic planning. In International Conference on Machine Learning, pp. 22028–22041. PMLR, 2023b.arxiv.org/abs/2209.15315
  40. [40]Jiafei Lyu, Zichuan Lin, Scott Fujimoto, Kai Yang, Yangkun Chen, Saiyong Yang, Zongqing Lu, and Deheng Ye. Debiased model-based representations for sample-efficient continuous control, 2026. URL https://arxiv.org/abs/2605.11711.
  41. [41]Ajay Mandlekar, Danfei Xu, Josiah Wong, Soroush Nasiriany, Chen Wang, Rohun Kulkarni, Li Fei-Fei, Silvio Savarese, Yuke Zhu, and Roberto Martín-Martín. What matters in learning from offline human demonstrations for robot manipulation. In arXiv preprint arXiv:2108.03298, 2021.arxiv.org/abs/2108.03298
  42. [42]Ofir Nachum, Shixiang Shane Gu, Honglak Lee, and Sergey Levine. Data-efficient hierarchical reinforcement learning. Advances in neural information processing systems, 31, 2018.arxiv.org/abs/1805.08296
  43. [43]Ashvin Nair, Vitchyr Pong, Murtaza Dalal, Shikhar Bahl, Steven Lin, and Sergey Levine. Visual reinforcement learning with imagined goals, 2018. URL https://arxiv.org/abs/1807.04742.
  44. [44]Ashvin Nair, Abhishek Gupta, Murtaza Dalal, and Sergey Levine. Awac: Accelerating online reinforcement learning with offline datasets, 2021. URL https://arxiv.org/abs/2006.09359.
  45. [45]Soroush Nasiriany, Vitchyr Pong, Steven Lin, and Sergey Levine. Planning with goal-conditioned policies. Advances in neural information processing systems, 32, 2019.arxiv.org/abs/1911.08453
  46. [46]Soroush Nasiriany, Sepehr Nasiriany, Abhiram Maddakuri, and Yuke Zhu. Robocasa365: A large-scale simulation framework for training and benchmarking generalist robots. arXiv preprint arXiv:2603.04356, 2026.arxiv.org/abs/2603.04356
  47. [48]Chaoyi Pan, Giri Anantharaman, Nai-Chieh Huang, Claire Jin, Daniel Pfrommer, Chenyang Yuan, Frank Permenter, Guannan Qu, Nicholas Boffi, Guanya Shi, and Max Simchowitz. Much ado about noising: Dispelling the myths of generative robotic control, 2026. URL https://arxiv.org/abs/2512.01809.
  48. [49]Seohong Park, Kevin Frans, Benjamin Eysenbach, and Sergey Levine. Ogbench: Benchmarking offline goal-conditioned rl, 2025a. URL https://arxiv.org/abs/2410.20092.
  49. [50]Seohong Park, Qiyang Li, and Sergey Levine. Flow q-learning, 2025b. URL https://arxiv.org/abs/2502.02538.
  50. [51]Ronald Parr and Stuart Russell. Reinforcement learning with hierarchies of machines. In Proceedings of the 11th International Conference on Neural Information Processing Systems, NIPS’97, pp. 1043–1049, Cambridge, MA, USA, 1997. MIT Press.
  51. [52]Sarvesh Patil, Mitsuhiko Nakamoto, Manan Agarwal, Shashwat Saxena, Jesse Zhang, Giri Anantharaman, Cleah Winston, Chaoyi Pan, Douglas Chen, Nai-Chieh Huang, Zeynep Temel, Oliver Kroemer, Sergey Levine, Abhishek Gupta, Hongkai Da, Paarth Shah, and Max Simchowitz. Ogpo: Sample efficient full-finetuning of generative control policies, 2026. URL https://arxiv.org/abs/2605.03065.
  52. [53]Shaohui Peng, Xing Hu, Rui Zhang, Ke Tang, Jiaming Guo, Qi Yi, Ruizhi Chen, Xishan Zhang, Zidong Du, Ling Li, Qi Guo, and Yunji Chen. Causality-driven hierarchical structure discovery for reinforcement learning, 2022. URL https://arxiv.org/abs/2210.06964.
  53. [54]Physical Intelligence, Bo Ai, Ali Amin, Ashwin Balakrishna, Kevin Black, Danny Driess, Michael Equi, Chelsea Finn, Lachlan Groom, Karol Hausman, Brian Ichter, Sergey Levine, Karl Pertsch, Allen Z. Ren, Lucy Xiaoyang Shi, et al. π0.7: a steerable generalist robotic foundation model with emergent capabilities. arXiv preprint arXiv:2604.15483, 2026.arxiv.org/abs/2604.15483
  54. [55]Rafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D Manning, Stefano Ermon, and Chelsea Finn. Direct preference optimization: Your language model is secretly a reward model. Advances in neural information processing systems, 36:53728–53741, 2023.arxiv.org/abs/2305.18290
  55. [56]Allen Z Ren, Justin Lidard, Lars L Ankile, Anthony Simeonov, Pulkit Agrawal, Anirudha Majumdar, Benjamin Burchfield, Hongkai Dai, and Max Simchowitz. Diffusion policy policy optimization. arXiv preprint arXiv:2409.00588, 2024.arxiv.org/abs/2409.00588
  56. [57]Tom Schaul, Daniel Horgan, Karol Gregor, and David Silver. Universal value function approximators. In Francis Bach and David Blei (eds.), Proceedings of the 32nd International Conference on Machine Learning, volume 37 of Proceedings of Machine Learning Research, pp. 1312–1320, Lille, France, 07–09 Jul 2015. PMLR. URL https://proceedings.mlr.press/v37/schaul15.html.
  57. [58]Daniel Shin, Anca D. Dragan, and Daniel S. Brown. Benchmarks and algorithms for offline preference-based reward learning, 2023. URL https://arxiv.org/abs/2301.01392.
  58. [59]Charlie Snell, Jaehoon Lee, Kelvin Xu, and Aviral Kumar. Scaling llm test-time compute optimally can be more effective than scaling model parameters. arXiv preprint arXiv:2408.03314, 2024.
  59. [60]Richard S Sutton and Andrew G Barto. Reinforcement learning: An introduction second edition. Adaptive computation and machine learning: The MIT Press, Cambridge, MA and London, 2018.
  60. [61]Richard S Sutton, Andrew G Barto, et al. Reinforcement learning: An introduction, volume 1. MIT press Cambridge, 1998.
  61. [62]Richard S. Sutton, Doina Precup, and Satinder Singh. Between mdps and semi-mdps: A framework for temporal abstraction in reinforcement learning. Artificial Intelligence, 112(1):181–211, 1999. ISSN 0004-3702. doi: https://doi.org/10.1016/S0004-3702(99)00052-1. URL https://www.sciencedirect.com/science/article/pii/S0004370299000521.
  62. [63]Fahim Tajwar, Anikait Singh, Archit Sharma, Rafael Rafailov, Jeff Schneider, Tengyang Xie, Stefano Ermon, Chelsea Finn, and Aviral Kumar. Preference fine-tuning of llms should leverage suboptimal, on-policy data. arXiv preprint arXiv:2404.14367, 2024.
  63. [64]Sreyas Venkataraman, Yufei Wang, Ziyu Wang, Navin Sriram Ravie, Zackory Erickson, and David Held. Real-world offline reinforcement learning from vision language model feedback. In IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), 2025.
  64. [65]Alexander Sasha Vezhnevets, Simon Osindero, Tom Schaul, Nicolas Heess, Max Jaderberg, David Silver, and Koray Kavukcuoglu. Feudal networks for hierarchical reinforcement learning, 2017. URL https://arxiv.org/abs/1703.01161.
  65. [66]Yufei Wang, Zhanyi Sun, Jesse Zhang, Zhou Xian, Erdem Biyik, David Held, and Zackory Erickson. RL-vlm-f: Reinforcement learning from vision language foundation model feedback, 2024. URL https://arxiv.org/abs/2402.03681.
  66. [67]Jason Wei, Xuezhi Wang, Dale Schuurmans, Maarten Bosma, Fei Xia, Ed Chi, Quoc V Le, Denny Zhou, et al. Chain-of-thought prompting elicits reasoning in large language models. Advances in neural information processing systems, 35:24824–24837, 2022.arxiv.org/abs/2201.11903
  67. [68]Christian Wirth, Riad Akrour, Gerhard Neumann, and Johannes Fürnkranz. A survey of preference-based reinforcement learning methods. Journal of Machine Learning Research, 18(136):1–46, 2017.
  68. [69]Tonghe Zhang, Chao Yu, Sichang Su, and Yu Wang. Reinflow: Fine-tuning flow matching policy with online reinforcement learning. arXiv preprint arXiv:2505.22094, 2025.

Paper details

vlnc:W_8y7sx4am
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents