Robo-Dopamine 2.0: History-Conditioned and OOD-Aware Process Reward Modeling for Robotic Manipulation
The paper presents Robo-Dopamine 2.0, a process reward model that uses history conditioning and a signed progress space to distinguish valid progress, robustness variations, failures, and recovery
1. Introduction: The Reward Modeling Bottleneck in Visuomotor Control
Large-scale Vision-Language-Action (VLA) models and imitation learning frameworks have advanced the capacity of robotic systems to execute manipulation tasks across diverse environments. However, long-horizon, contact-rich manipulation tasks remain highly susceptible to compounding execution errors, unpredicted environmental variations, and off-trajectory states. While Reinforcement Learning (RL) provides a formal mechanism to refine pretrained VLA policies through environmental interaction, policy optimization relies heavily on dense, informative reward signals.
Standard sparse terminal rewards require extensive exploration, rendering RL fine-tuning inefficient for long-horizon trajectories. Conversely, manually engineering dense reward functions requires task-specific heuristic modeling, making them brittle, difficult to scale, and costly to design. Learned visual reward models address this bottleneck by estimating dense progress directly from visual observations and task instructions. Nevertheless, conventional visual reward models rely predominantly on static before-and-after observation pairs (), which introduces two primary failure modes during execution:
- Temporal Ambiguity and Temporal Aliasing: Static frame comparisons struggle to resolve latent progress when visually similar observations correspond to distinct execution phases. These ambiguities often stem from unobserved physical contact events, repeated motion phases, or reordered subgoals where static endpoint observations appear nearly identical despite representing different temporal states.
- Weak Out-of-Distribution (OOD) Discrimination: Standard visual reward models often conflate semantics-preserving, benign scene variations with catastrophic, task-invalid failures. While perturbations such as transient visual occlusions, distractor objects, or background shifts ought to preserve progress estimates (robustness), true execution failures—such as empty grasps, wrong-object interactions, or missed object releases—require strict negative reward feedback. In-distribution demonstration datasets rarely contain negative execution states, leaving reward models without explicit supervision to distinguish benign variations from task-invalid failures.
To resolve these challenges, Xu et al. (Aug 2026) introduced Robo-Dopamine 2.0, a history-conditioned and OOD-aware process reward model (PRM). Designed to mitigate compounding errors and off-trajectory policy divergence, Robo-Dopamine 2.0 incorporates temporal context panels and a signed progress representation into a unified pairwise prediction framework. Built upon the Qwen3-VL-8B vision-language backbone and evaluated against prior architectures such as GRM-8B-Pro, Robo-Dopamine 2.0 provides an infrastructure for dense visual credit assignment under off-trajectory and OOD execution.
2. History-Conditioned Pairwise Reward Architecture
Robo-Dopamine 2.0 maintains a pairwise prediction interface while integrating history conditioning to resolve static temporal ambiguity.
Mathematical Formulation
The foundational prediction interface is governed by the General Reward Model (GRM) pairwise progress equation:
Where:
- represents the textual task instruction.
- specifies static task-reference anchors corresponding to the task start state () and task goal state ().
- represents the ordered temporal context frames preceding and separating the queried endpoints.
- denotes the ordered pair of queried multi-view endpoint observations, where aggregates canonical view slots: head/third-person camera (), left wrist camera (), and right wrist camera ().
- represents the predicted signed relative progress.
Temporal Context Structure & Token Budget Trade-offs
The temporal context structures historical frames into two ordered grid layouts:
- Pre-query panels (): Sampled uniformly from execution states preceding the earlier queried endpoint index (), summarizing historical execution leading into state .
- Between-state panels (): Sampled uniformly from execution states strictly between the queried endpoints (), summarizing state transitions separating and .
Images within panels are arranged in standard grid layouts across canonical camera slots. Missing or unavailable auxiliary views are padded with fixed mask images to maintain structural format consistency.
# Static Baseline Input Layout
I_static = [r_s, r_g, X_b, X_a]
# Robo-Dopamine 2.0 Reference Panel Input Layout
I_ref = [r_s, r_g, M_{1:K}^{pre}, X_b, M_{1:L}^{mid}, X_a]
Ablations conducted by Xu et al. reveal critical engineering trade-offs regarding visual-token capacity. While adding temporal context panels improves visual order consistency, expanding panel grid resolution beyond (e.g., to ) or increasing the number of between-state panels () results in non-monotonic performance degradation. This saturation stems from visual-token budget constraints within the underlying VLM backbone, where excessive context frames dilute attention allocation over key endpoint evidence. Consequently, a panel layout with between-state panels provides the optimal architectural balance.
Context Provenance Contract Across Operational Modes
The provenance of the temporal context frames adapts to the operational query context under a strict provenance contract:
- Standard Training Queries: Context panels () are compiled directly from same-episode expert execution history.
- Synthetic OOD Queries: Context panels are compiled exclusively from the unedited, successful source episode, while only the queried endpoints contain the counterfactual OOD edits. This source-only context contract ensures that the reference panels supply nominal, valid execution progress history without obscuring or replacing the edited endpoint evidence.
- Online RL Inference: Context panels are constructed dynamically from the agent’s observed rollout history prefix up to rollout step , requiring no access to expert source trajectories or ground-truth step labels.
Multi-Anchor Online Potential Fusion Mechanism
During closed-loop RL inference, Robo-Dopamine 2.0 evaluates three complementary, normalized query projections at each rollout step :
- Forward-anchored query (): Evaluates progress from the initial rollout observation to the current observation using rollout prefix context :
- Backward-anchored query (): Evaluates progress from the static goal reference to current observation without temporal panels:
- Incremental query (): Evaluates progress from the immediate previous state to using step-adjacent rollout context :
Because the initial state possesses a defined potential of zero, a negative forward score operationally signals entry into a task-invalid negative branch. Xu et al. formalize this via a dynamic online branch gate indicator (with ). This indicator establishes the query-specific branch floor parameters:
To invert normalized signed Hop targets back into potential space, the framework applies the invertibility function :
The individual potential estimates are recovered as:
These three components are aggregated into a unified, context-conditioned online state potential :
3. The Signed Progress Space & Counterfactual OOD Pipeline
Mathematical Structure of the Signed Progress Space
Robo-Dopamine 2.0 organizes trajectory states into a continuous signed progress space , categorizing execution states across positive, robustness-preserving, and negative branches.
Given a successful source trajectory of length , normalized progress for state index is defined as . The signed space assigns scalar targets based on state branch classification:
Here, positive states () and robustness-preserving states () share identical positive progress values . Task-invalid negative states () are assigned a negative sign with absolute magnitude , indexing failure depth by the source trajectory phase at which the failure occurred.
[ Positive Branch: Valid Executions (P_i) ] -----> Progress: +g_i
/
[State] -- [ Robustness Branch: Semantics-Preserving (R_i) ] -> Progress: +g_i
\
[ Negative Branch: Task-Invalid Failures (N_i) ] ---> Progress: -g_i
Continuous Signed Hop Target Formulation and Floor Invertibility
For a queried pair with assigned potentials and , the progress differential is explicitly defined as .
To map transitions seamlessly across valid progress, failure, and recovery, the framework incorporates a normalized branch floor parameter:
Where is a binary indicator outputting if state belongs to a negative branch (), and otherwise. The Continuous Signed Hop target is formulated as:
0, & \Delta_{ba} = 0 \\ \frac{\Delta_{ba}}{1 - \phi_b}, & \Delta_{ba} > 0 \\ \frac{\Delta_{ba}}{\phi_b - L_{ba}}, & \Delta_{ba} < 0 \end{cases}$$ The regression label supplied to the loss function is defined as: $$y_{ba} = 100 \cdot \text{clip}(H_{ba}, -1, 1)$$ The constant scaling factor of 100 preserves the output magnitude of the original GRM regression interface and does not convert the continuous regression target into a discrete three-class classification objective. The normalized floor $L_{ba}$ ensures mathematical invertibility via $\Gamma(h; \phi, L)$. When both queried states are valid ($P$ or $R$), $L_{ba} = 0$, normalizing positive progress differentials toward the goal value $1$. When either state involves a negative execution state ($N$), $L_{ba} = -1$. This shift extends the continuous mapping across failure transitions ($P \to N$, where $H_{ba} < 0$) and recovery transitions ($N \to P$, where $H_{ba} > 0$), allowing the model to represent off-trajectory credit assignment without density engineering. ### 6-Step Synthetic OOD Data Construction Pipeline To generate supervision across negative and robust branches, Xu et al. developed a 6-step counterfactual editing pipeline that applies task- and phase-aware interventions to successful source trajectories (drawn from RoboCasa, LIBERO, AgiBot World, and RoboTwin 2.0): 1. **Task Parsing & Primitive Extraction:** Instructions are normalized and mapped across 7 primitive categories: *grasp-lift*, *place-transfer*, *open*, *close*, *toggle-on*, *toggle-off*, and *push-move*. Target phrase extraction isolates task objects from non-target scene elements. 2. **Operator Assignment & Semantic Guards:** Edits are categorized into task-invalid negative operators ($Z_{neg}$) and robustness-preserving operators ($Z_{rob}$): * *Negative Operators ($Z_{neg}$):* Wrong object, empty grasp, non-release, target replacement. * *Robustness Operators ($Z_{rob}$):* Foreground occlusion, background change, distractor addition/removal, non-target color change. * *Guards:* Identity-changing edits are avoided when multi-object ambiguity exists, and non-target color edits protect instruction-specified target colors. 3. **Phase-Aware Temporal Localization:** Interventions are bounded by primitive-specific phase windows to maintain semantic coherence: * *Grasp-lift:* $[0.18, 0.80]$ * *Place-transfer:* $[0.28, 0.95]$ * *Open / Close:* $[0.15, 0.85]$ * *Toggle-on / Toggle-off:* $[0.10, 0.75]$ * *Push-move:* $[0.12, 0.85]$ 4. **Multi-Reference Edit Construction:** For a queried frame index $i$, up to $K_i \le 13$ chronological reference frames from the identical camera stream are appended to the editing prompt to preserve target identity, scene layout, and lighting conditions. 5. **Structured Prompt Design & Geometric Constraints:** Functional prompts are composed as $P = P_{input} \oplus P_{scene} \oplus P_{task} \oplus P_{target} \oplus P_{operator} \oplus P_{view} \oplus P_{physical} \oplus P_{output}$. Head-camera prompts enforce global corridor constraints, while wrist-camera prompts restrict edits to local gripper boundaries and visible finger geometry. 6. **Batch Execution & Provenance Tracking:** Image editing is executed via Gemini 3.1 Flash Image in batch mode. Source frame IDs, camera indices, primitive states, and operator metadata are logged to guarantee exact pair alignment ($\tilde{o}_{i}^v \longleftrightarrow o_{i}^v$). --- ## 4. Signed-Hop Curriculum with Transition-Aware Replay To prevent large-transition pairs from overwhelming local progress calibration, Xu et al. implemented a two-stage training schedule comprising 400K pairwise examples (200K per stage) built on the Qwen3-VL-8B backbone. ### Stage Division and Replay Mechanism Pairwise training examples are ranked by absolute magnitude $|H_{ba}|$ within each pair family to establish two training pools: * **$D_{large}$:** Top 40% of examples by $|H_{ba}|$ per family, capturing global progress, failure, and recovery geometry. * **$D_{fine}$:** Remaining 60% of examples, containing small-Hop and zero-Hop pairs for fine-grained calibration. #### Stage 1 (0–200K steps) Sampling is drawn strictly from $D_{large}$ to establish global trajectory ordering and branch boundaries. #### Stage 2 (200K–400K steps) Sampling mixes fine-grained calibration examples with replayed large-Hop transitions using a sampling ratio $\rho$: $$D_2 = (1 - \rho)D_{fine} + \rho D_{large}$$ Empirical sweeps demonstrate that **$\rho = 0.25$** (25% large-Hop replay) yields optimal overall Visual Order Consistency (VOC), preventing catastrophic forgetting of global progress geometry during fine-grained calibration. ### Trajectory-Level Visual Order Consistency (VOC) Across Curriculum Variants Visual Order Consistency (VOC) evaluates predicted progress along full trajectories using Spearman rank correlation against ground-truth progress indices across 400K training examples. | Curriculum Strategy ID | Strategy Description | Stage 1 Pool | Stage 2 Pool | Stage 2 Replay ($\rho$) | Mean Trajectory VOC | | :--- | :--- | :--- | :--- | :--- | :---: | | **$E_0$** | Random Mixture | Unordered | Unordered | N/A | Baseline (0.9841) | | **$E_1$** | Matched Shuffled Control | Matched | Matched | Shuffled | 0.9858 | | **$E_2$** | Reverse Curriculum | $D_{fine}$ | $D_{large}$ | N/A | Degradation vs. $E_0$ | | **$E_3$** | Large-to-Fine (No Replay) | $D_{large}$ | $D_{fine}$ | $0.00$ | 0.9866 | | **$E_4$** | **Signed-Hop + 25% Replay** | **$D_{large}$** | **$D_2$** | **$0.25$** | **0.9872** | *Note: The matched shuffled control ($E_1$) uses the exact sample pool as $E_4$ without structured ordering, confirming that the performance gains of $E_4$ stem from progressive curriculum staging rather than data composition alone.* --- ## 5. Empirical Results & Downstream RL Evaluation ### Offline Trajectory VOC Across Benchmark Families Trajectory-level VOC performance across five benchmark evaluation families (In-Distribution, Temporal-Memory, OOD-Negative, OOD-Robust, and OOD-Temporal) highlights the impact of structured context panels: * **Static Endpoint Baseline ($I_{static}$):** Achieves a mean VOC of **0.967** (with OOD-Robust VOC dropping to **0.906** due to occlusion sensitivity). * **GRM-8B-Pro Baseline (Static):** Achieves an unweighted mean VOC of **0.958**. * **Robo-Dopamine 2.0-8B ($3\times3$ Panel Input):** Achieves a mean VOC of **0.986** across all families, raising OOD-Robust VOC to **0.958**, In-Distribution (ID) to **0.991**, Temporal-Memory to **0.994**, OOD-Negative to **0.994**, and OOD-Temporal to **0.991**. ### Task-Completion Classification Accuracy Evaluated on discrete task-completion classification across 60 held-out test episodes, Robo-Dopamine 2.0-8B outperforms baseline generalist VLMs: | Model | Success Rate (Out of 60 Episodes) | Classification Accuracy (%) | | :--- | :---: | :---: | | **Robo-Dopamine 2.0-8B** | **57.7 / 60** | **96.1%** | | GPT-5.5 | 48.6 / 60 | 81.1% | | Claude-Opus-4.8 | 47.6 / 60 | 79.4% | | Gemini-3-Pro | 46.3 / 60 | 77.2% | | Qwen3.5-397B-A17B-fp8 | 45.7 / 60 | 76.1% | | RoboBrain 2.0 | 37.0 / 60 | 61.7% | ### Simulation RL Policy Fine-Tuning (RoboTwin 2.0) Group Relative Policy Optimization (GRPO) was applied to fine-tune OpenVLA-OFT policies across three RoboTwin 2.0 tasks (`place_empty_cup`, `place_container_plate`, `handover_block`). | Condition Code | OOD Supervision | Rollout History Panels | Training Budget | Cup (%) | Plate (%) | Hand. (%) | Mean Success Rate (%) | | :--- | :---: | :---: | :---: | :---: | :---: | :---: | :---: | | **RLinf Sparse** | — | — | 1,000 Epochs | 94.2 | **95.0** | 68.2 | 85.8% | | **C00 Controlled** | $\times$ | $\times$ | 300 Epochs | 77.5 | 82.1 | 57.0 | 72.2% | | **C01 History-Only** | $\times$ | $\checkmark$ | 300 Epochs | 79.5 | 88.6 | 57.1 | 75.1% | | **C10 OOD-Only** | $\checkmark$ | $\times$ | 300 Epochs | 85.2 | 89.5 | 67.1 | 80.6% | | **C11 Full Model** | $\checkmark$ | $\checkmark$ | **300 Epochs** | **96.2** | 94.6 | **69.7** | **86.8%** | The full model (C11) achieves an **86.8%** mean success rate within 300 epochs, surpassing the 1,000-epoch sparse reward baseline (**85.8%**). While C11 dominates overall and achieves top performance on `place_empty_cup` (96.2%) and `handover_block` (69.7%), the sparse baseline remains slightly higher by 0.4% on `place_container_plate` (95.0% vs. 94.6%). ### Real-World Dual-Franka Repeated Insertions Evaluated on a physical dual-arm Franka setup running ConRFT on the multi-attempt *Insert Square* task ($K=4$ insertion attempts per trial, total 20 trials / 80 insertion attempts): | Model / Condition | Reward Signal Type | Online Training Budget | Successful Insertions (Out of 80) | Full Sequence Completions (Out of 20) | | :--- | :--- | :---: | :---: | :---: | | **Event Sparse** | Per-insertion Event Reward | 300 Episodes | 48 / 80 | 8 / 20 | | **C00 Controlled** | GRM Shaping (No History/OOD) | 200 Episodes | 46 / 80 | 7 / 20 | | **C01 History-Only** | GRM Shaping + History Panels | 200 Episodes | 67 / 80 | 13 / 20 | | **C10 OOD-Only** | GRM Shaping + OOD Supervision | 200 Episodes | 50 / 80 | 9 / 20 | | **C11 Full Model** | **GRM Shaping + Full Model** | **200 Episodes** | **71 / 80** | **15 / 20** | Rollout-history panels eliminate temporal aliasing during repeated insertion loops, enabling the full model (C11) to achieve **71/80** successful insertions and **15/20** full-sequence completions using $33\%$ fewer online interaction episodes than the sparse baseline. --- ## 6. Key Takeaways for AI Safety & Embodied RL Research 1. **Rollout History Context Eliminates Temporal Aliasing:** Static before-after observation pairs inherent to traditional process reward models fail under repeated execution loops and hidden contact phases. Conditioning process reward models on rollout prefixes enables agents to disambiguate identical state observations based on historical context. 2. **Explicit Failure Bounds via Signed Progress:** Structuring negative states into a signed potential space ($\Phi < 0$) establishes clear mathematical bounds for task-invalid execution. This explicit representation prevents reward models from rewarding off-trajectory failure states while providing fine-grained credit assignment for recovery maneuvers. 3. **Curriculum Replay Retains Global Trajectory Geometry:** Training process reward models using a Signed-Hop curriculum—initiating optimization on large-Hop transitions ($D_{large}$) followed by fine-grained calibration ($D_{fine}$) with moderate replay ($\rho = 0.25$)—preserves global progression geometry while sharpening local step accuracy. 4. **Potential-Based Reward Shaping Accelerates Fine-Tuning:** Incorporating learned process potentials into potential-based reward shaping: $$\hat{r}_t^{\text{GRM}} = r_t^{\text{task}} + \gamma_{\text{RL}} \hat{\Phi}_{t+1} - \hat{\Phi}_t$$ delivers continuous, OOD-aware progress feedback that accelerates downstream policy convergence in both simulation and real-world deployment without manual reward engineering. *Read the [full paper on arXiv](https://arxiv.org/abs/2608.15680) · [PDF](https://arxiv.org/pdf/2608.15680.pdf)*