FlowDPG

FlowDPG Deterministic Policy Gradient on Flow Matching Policies
for Real-World Manipulation

Kexin Shi1,2 Junyao Shi1,3 Poorvi Hebbar1 Zhuolun Zhao1
Tarun Amarnath1 Yifan Su1 Shikhar Bahl1 Deepak Pathak1,2

Skild AI1 Carnegie Mellon University2 University of Pennsylvania3
CoRL 2026
Abstract

Real-world reinforcement learning for robotic manipulation remains challenging, and this difficulty is amplified for flow matching policies: applying policy gradient methods to these policies is fundamentally limited by the need to backpropagate through time (BPTT) along the multi-step ODE that maps noise to actions, which is computationally prohibitive and numerically fragile. We propose FlowDPG, a DDPG-style method specifically designed for flow matching policies that distills the critic gradient into the velocity field at training time, bypassing BPTT entirely. Intuitively, FlowDPG combines two complementary vectors: the demonstration-driven velocity that keeps the action feasible, and the critic-driven correction that steers it toward higher value. Our contributions are threefold: (1) a BPTT-free in-place framework that leaves the deployed multi-step inference path unchanged, (2) a formal connection between the FlowDPG update direction and vanilla Deterministic Policy Gradient via three explicit approximations, and (3) real-world validation on two long-horizon, multi-stage, dual-arm tasks on two different robots — AirPods assembly on a Franka and scrambled-egg cooking on a YAM.

AirPods Assembly on Franka

One hour uncut · 25 cases back to back · no human intervention

Scrambled Eggs on YAM

Recovery from Failures

AirPods · emergent re-grasp
AirPods · two pods grasped by accident, one still inserted
Eggs · re-grasping a dropped eggshell
Eggs · retrying a misplaced spatula

Method Overview

FlowDPG vs. BC and QAM

BCQAM offlineQAM +onlineFlowDPG offlineFlowDPG +onlineerror bars: s.e. over 5 checkpoints
AirPods · overall success (%)
025507510062BC7880QAM8690FlowDPG
AirPods · rubric score (%)
025507510066.6BC79.481.4QAM88.292.3FlowDPG
Eggs · rubric score (%), offline
025507510064.2BC76.9QAM85.8FlowDPG

50 rollouts per cell (10 at each of 5 checkpoints). Rubric: 0–3 per stage (criteria). Eggs: offline only.

All baseline families (AirPods)

BC (base)64%
Value-conditioning
RA-BC76%
AWR76%
RECAP72%
Auxiliary module
DSRL68%
PLD76%
RLT80%
Adjoint critic gradient
QAM80%
Ours
FlowDPG (offline)88%
FlowDPG (+online)92%
End-to-end success (%)
020406080100
BC base (64%)

25 cases per method; baselines trained offline only.

Human Disturbances

Recovery rate under human disturbances (AirPods)

offline+onlinerecovery rate (%), 50 trials each
Re-graspobject put back in the tray
78
92
+14
Re-open / closecase state inverted
82
94
+12
Re-insertpod dislodged
86
92
+6
Adapt grasptray reshuffled mid-reach
70
88
+18

Stage-Aware Reward Model

Successful rollout · predicted progress and stage
Failed rollout · progress stalls, then drops

Reward ablation

Full reward
100%
96%
96%
96%
96%
92%
92%
92%
w/o progress
100%
96%
92%
88%
84%
84%
80%
80%
w/o stage transition
100%
96%
92%
88%
84%
80%
76%
76%
Only terminal
100%
80%
76%
72%
68%
64%
60%
60%
Grasp
case
Open
case
Grasp
R pod
Insert
R pod
Grasp
L pod
Insert
L pod
Close
case
Place
case

Share of AirPods episodes that get through each stage.

Consistency Loss and Adaptive Shift

(a) Projection error0.050.10510training steps (k)‖â − x₁‖w/o consistencyw/ consistency
(b) Q-value discrepancy0.250.50510training steps (k)|Q(s,â) − Q(s,x₁)|w/o consistencyw/ consistency
(c) Training loss0.080.160510training steps (k)lossw/o adaptive shiftw/ adaptive shift
(d) Success rate050100overall success (%)72w/oboth76w/oshift80w/oconsistency92Full

Swipe sideways for all four panels.

All variants include the online phase.

Evaluation rubrics
0–3 per stage · AirPods: 8 stages, 24 points · Eggs: 14 stages, 42 points
Click to expandCollapse

A stage never reached scores 0. Criteria were fixed before scoring.

AirPods assembly (Franka) · 8 stages, 24 points

Stage3 – Successful2 – Minor error1 – Major error0 – Failed
Grasp caseSecure grasp, correct orientation, no slipGrasp with adjustment or delayCase lifted but shifted or dropped onceFails to grasp case
Open caseLid fully opened in one clean motionOpens with adjustment or hesitationOpens only after repeated attempts, or case displacedLid not opened
Grasp right podSecure grasp, correct approach directionGrasp with adjustment or delayWrong approach direction, pod dislodged then re-graspedFails to grasp pod
Insert right podSeated on the first attempt, fully inSeated after one retrySeated after two or more retries, or insertion dislodges the other podPod not seated
Grasp left podSecure grasp, correct approach directionGrasp with adjustment or delayWrong approach direction, pod dislodged then re-graspedFails to grasp pod
Insert left podSeated on the first attempt, fully inSeated after one retrySeated after two or more retries, or insertion dislodges the right podPod not seated
Close caseLid closed cleanly with both pods seatedCloses after minor adjustmentCloses with a pod unseated, or needs repositioningCase not closed
Place casePlaced upright within the target regionPlaced with minor misalignmentDropped or placed outside the regionCase not placed

Scrambled eggs (YAM) · 14 stages, 42 points

Pick up and crack occur twice, once per egg: 14 scored stages from 12 types.

Stage3 – Successful2 – Minor error1 – Major error0 – Failed
Pick up eggSecure grasp, no slip or damageGrasp with adjustment or delayLifts egg but drops or damages itFails to grasp egg
Crack eggClean crack, proper shell separationMinor shell fragments or awkward motionIncorrect or messy crackFails to crack egg
Put saltCorrect amount applied accuratelyMinor spillage or hesitationIncorrect amount or placementSalt not applied, or bottle dropped
Pick up forkSecure and correct graspGrasp with adjustmentUnstable or incorrect graspFails to pick up fork, or drops it
Whisk eggThorough whisking with correct motionPartial or inefficient whiskingMinimal or incorrect motionNo whisking
Place fork backPlaced neatly in correct locationMinor misalignmentDropped or poorly placedFork not placed back
Put eggs into panClean transfer into panMinor spillage or hesitationSignificant spillageEggs not transferred
Pick up spatulaSecure and correct graspGrasp with adjustmentUnstable or incorrect graspFails to pick up spatula
Stir eggsProper stirring across panIncomplete or uneven stirringMinimal or incorrect stirringNo stirring
Transfer eggsClean and accurate transferMinor spillage or inefficiencyPartial or messy transferEggs not transferred
Place spatula backPlaced correctly and safelyMinor misplacementDropped or unsafe placementSpatula not placed back
Serve eggsServed correctly and fullyMinor error or delayIncorrect or incomplete servingEggs not served

BibTeX

@inproceedings{shi2026flowdpg,
  title     = {FlowDPG: Deterministic Policy Gradient on Flow Matching Policies
               for Real-World Manipulation},
  author    = {Shi, Kexin and Shi, Junyao and Hebbar, Poorvi and Zhao, Zhuolun and
               Amarnath, Tarun and Su, Yifan and Bahl, Shikhar and Pathak, Deepak},
  booktitle = {Conference on Robot Learning (CoRL)},
  year      = {2026}
}