Poke &Wiggle

Leaderboard

Model

DiT-Flow

Carnegie Mellon University · UC Berkeley

#4

Rank · PAW-GEN-10

Model information

Website
Website⁠ (opens in a new tab)
Paper
arXiv⁠ (opens in a new tab)
Code
SudeepDasari/dit-policy⁠ (opens in a new tab)
Parameters
115M
Released
October 2024
Licence
MIT

PAW-GEN-10

Share of episodes rated Ok or Excellent.

Summary

Overall Success
14%
Evaluations
5400
NominalEvaluated on the placements the model was trained on.InterpolationEvaluated on unseen placements between the training placements.ExtrapolationEvaluated on unseen placements outside the training placements' convex hull.
EnvironmentNominal · D10Trained on ~10 demonstrations per environment.Nominal · D100Trained on ~100 demonstrations per environment.Nominal · D300Trained on ~300 demonstrations per environment.Interpolation · D10Trained on ~10 demonstrations per environment.Interpolation · D100Trained on ~100 demonstrations per environment.Interpolation · D300Trained on ~300 demonstrations per environment.Extrapolation · D10Trained on ~10 demonstrations per environment.Extrapolation · D100Trained on ~100 demonstrations per environment.Extrapolation · D300Trained on ~300 demonstrations per environment.
Basic skills
Sort screws into bins7%#195% CI 2%–21%0%#595% CI 0%–11%0%#495% CI 0%–11%0%#195% CI 0%–11%0%#295% CI 0%–11%3%#395% CI 1%–17%0%#195% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%
Datum-corner alignment3%#495% CI 1%–17%33%#495% CI 19%–51%27%#595% CI 14%–44%0%#595% CI 0%–11%20%#495% CI 10%–37%17%#595% CI 7%–34%0%#195% CI 0%–11%0%#395% CI 0%–11%0%#395% CI 0%–11%
Pour screws into the funnel23%#295% CI 12%–41%60%#495% CI 42%–75%87%#395% CI 70%–95%0%#595% CI 0%–11%53%#495% CI 36%–70%67%#395% CI 49%–81%0%#395% CI 0%–11%0%#595% CI 0%–11%13%#495% CI 5%–30%
High precision
Connect DC jack0%#195% CI 0%–11%7%#495% CI 2%–21%0%#395% CI 0%–11%0%#195% CI 0%–11%3%#295% CI 1%–17%0%#295% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%
Route cable through closed hoops0%#195% CI 0%–11%0%#495% CI 0%–11%0%#495% CI 0%–11%0%#195% CI 0%–11%0%#495% CI 0%–11%0%#495% CI 0%–11%0%#195% CI 0%–11%0%#295% CI 0%–11%0%#395% CI 0%–11%
Put tools in standing toolbox0%#295% CI 0%–11%3%#195% CI 1%–17%3%#495% CI 1%–17%0%#195% CI 0%–11%0%#395% CI 0%–11%0%#495% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%
Bimanual
Place loaded boxes in container (bimanual)3%#395% CI 1%–17%30%#395% CI 17%–48%73%#295% CI 56%–86%0%#295% CI 0%–11%33%#295% CI 19%–51%53%#395% CI 36%–70%0%#195% CI 0%–11%0%#395% CI 0%–11%0%#395% CI 0%–11%
Place spray bottles upright, nozzle forward (bimanual handover)7%#595% CI 2%–21%90%#195% CI 74%–97%100%#195% CI 89%–100%0%#495% CI 0%–11%90%#195% CI 74%–97%90%#195% CI 74%–97%0%#195% CI 0%–11%53%#195% CI 36%–70%67%#195% CI 49%–81%
Tool use
Open the toolbox with a screwdriver0%#295% CI 0%–11%30%#395% CI 17%–48%30%#295% CI 17%–48%3%#195% CI 1%–17%7%#495% CI 2%–21%27%#395% CI 14%–44%0%#195% CI 0%–11%0%#395% CI 0%–11%0%#595% CI 0%–11%
Clamp electrical component0%#195% CI 0%–11%23%#295% CI 12%–41%3%#395% CI 1%–17%0%#195% CI 0%–11%0%#395% CI 0%–11%0%#395% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%0%#195% CI 0%–11%
Pooled4%#495% CI 3%–7%28%#395% CI 23%–33%32%#495% CI 27%–38%0%#595% CI 0%–2%21%#495% CI 16%–26%26%#495% CI 21%–31%0%#395% CI 0%–1%5%#495% CI 3%–8%8%#495% CI 5%–12%

Data Scaling

Scaling results for DiT-Flow: SuccessDiT-Flow nominal: D10 4% (95% CI 3%–7%), D100 28% (95% CI 23%–33%), D300 32% (95% CI 27%–38%). DiT-Flow interpolation: D10 0% (95% CI 0%–2%), D100 21% (95% CI 16%–26%), D300 26% (95% CI 21%–31%). DiT-Flow extrapolation: D10 0% (95% CI 0%–1%), D100 5% (95% CI 3%–8%), D300 8% (95% CI 5%–12%).Success rate (%)03060DiT-Flow nominal: 4% (95% CI 3%–7%) on D10DiT-Flow nominal: 28% (95% CI 23%–33%) on D100DiT-Flow nominal: 32% (95% CI 27%–38%) on D300DiT-Flow interpolation: 0% (95% CI 0%–2%) on D10DiT-Flow interpolation: 21% (95% CI 16%–26%) on D100DiT-Flow interpolation: 26% (95% CI 21%–31%) on D300DiT-Flow extrapolation: 0% (95% CI 0%–1%) on D10DiT-Flow extrapolation: 5% (95% CI 3%–8%) on D100DiT-Flow extrapolation: 8% (95% CI 5%–12%) on D300D10D100D300

Spatial Generalization

Spatial generalization for DiT-Flow: SuccessDiT-Flow: Nominal 21% (95% CI 19%–24%), Interpolation 16% (95% CI 13%–18%), Extrapolation 4% (95% CI 3%–6%).Success rate (%)03060DiT-Flow: 21% (95% CI 19%–24%) on NominalDiT-Flow: 16% (95% CI 13%–18%) on InterpolationDiT-Flow: 4% (95% CI 3%–6%) on ExtrapolationNominalInterpolationExtrapolation

Per Environment (Success)

Model performance by environmentDiT-Flow overall: Sort screws into bins 1%, Datum-corner alignment 11%, Pour screws into the funnel 34%, Connect DC jack 1%, Route cable through closed hoops 0%, Put tools in standing toolbox 1%, Place loaded boxes in container (bimanual) 21%, Place spray bottles upright, nozzle forward (bimanual handover) 55%, Open the toolbox with a screwdriver 11%, Clamp electrical component 3%.Sort screws into binse01Datum-corner alignmente02Pour screws into the funnele03Connect DC jacke04Route cable through closed hoopse05Put tools in standing toolboxe06Place loaded boxes in container (bimanual)e07Place spray bottles upright, nozzle forward (bimanual handover)e08Open the toolbox with a screwdrivere09Clamp electrical componente10Basic skillsHigh precisionBimanualTool use

Training parameters

Starting checkpoint
From scratch
Frozen part
None
Total batch size
256
Learning rate
2e-4
Schedule
Warmup, then cosine to 0
Action chunk
32
Action space
Absolute TCP pose

Compare every model's parameters

Rollouts

Mid-training dataM0
Task-specific dataD300
SpatialNominal

One clip per environment from this entry's evaluation, beside where each evaluation episode placed the environment's objects, coloured by its rating. Use the button beside each environment to open its evaluation dataset in the dataset viewer.

  1. Sort screws into bins
    Success
    0%
    Progress
    44%
  2. Pour screws into the funnel
    Success
    87%
    Progress
    90%
  3. Connect DC jack
    Success
    0%
    Progress
    13%
  4. Route cable through closed hoops
    Success
    0%
    Progress
    39%
  5. Open the toolbox with a screwdriver
    Success
    30%
    Progress
    51%
  6. Held-out benchmark environment
    Datum-corner alignment
    Success
    27%
    Progress
    27%
  7. Held-out benchmark environment
    Put tools in standing toolbox
    Success
    3%
    Progress
    22%
  8. Held-out benchmark environment
    Place loaded boxes in container (bimanual)
    Success
    73%
    Progress
    83%
  9. Held-out benchmark environment
    Place spray bottles upright, nozzle forward (bimanual handover)
    Success
    100%
    Progress
    100%
  10. Held-out benchmark environment
    Clamp electrical component
    Success
    3%
    Progress
    42%

Success by dataset size

e01 · Sort screws into bins
D10: 2% (95% CI 1%–8%)D100: 0% (95% CI 0%–4%)D300: 1% (95% CI 0%–6%)D10D100D300
e02 · Datum-corner alignment
D10: 1% (95% CI 0%–6%)D100: 18% (95% CI 11%–27%)D300: 14% (95% CI 9%–23%)D10D100D300
e03 · Pour screws into the funnel
D10: 8% (95% CI 4%–15%)D100: 38% (95% CI 28%–48%)D300: 56% (95% CI 45%–65%)D10D100D300
e04 · Connect DC jack
D10: 0% (95% CI 0%–4%)D100: 3% (95% CI 1%–9%)D300: 0% (95% CI 0%–4%)D10D100D300
e05 · Route cable through closed hoops
D10: 0% (95% CI 0%–4%)D100: 0% (95% CI 0%–4%)D300: 0% (95% CI 0%–4%)D10D100D300
e06 · Put tools in standing toolbox
D10: 0% (95% CI 0%–4%)D100: 1% (95% CI 0%–6%)D300: 1% (95% CI 0%–6%)D10D100D300
e07 · Place loaded boxes in container (bimanual)
D10: 1% (95% CI 0%–6%)D100: 21% (95% CI 14%–31%)D300: 42% (95% CI 33%–53%)D10D100D300
e08 · Place spray bottles upright, nozzle forward (bimanual handover)
D10: 2% (95% CI 1%–8%)D100: 78% (95% CI 68%–85%)D300: 86% (95% CI 77%–91%)D10D100D300
e09 · Open the toolbox with a screwdriver
D10: 1% (95% CI 0%–6%)D100: 12% (95% CI 7%–21%)D300: 19% (95% CI 12%–28%)D10D100D300
e10 · Clamp electrical component
D10: 0% (95% CI 0%–4%)D100: 8% (95% CI 4%–15%)D300: 1% (95% CI 0%–6%)D10D100D300

Success by spatial configuration

e01 · Sort screws into bins
Nominal: 2% (95% CI 1%–8%)Interpolation: 1% (95% CI 0%–6%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e02 · Datum-corner alignment
Nominal: 21% (95% CI 14%–31%)Interpolation: 12% (95% CI 7%–21%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e03 · Pour screws into the funnel
Nominal: 57% (95% CI 46%–66%)Interpolation: 40% (95% CI 30%–50%)Extrapolation: 4% (95% CI 2%–11%)NominalInterp.Extrap.
e04 · Connect DC jack
Nominal: 2% (95% CI 1%–8%)Interpolation: 1% (95% CI 0%–6%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e05 · Route cable through closed hoops
Nominal: 0% (95% CI 0%–4%)Interpolation: 0% (95% CI 0%–4%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e06 · Put tools in standing toolbox
Nominal: 2% (95% CI 1%–8%)Interpolation: 0% (95% CI 0%–4%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e07 · Place loaded boxes in container (bimanual)
Nominal: 36% (95% CI 26%–46%)Interpolation: 29% (95% CI 21%–39%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e08 · Place spray bottles upright, nozzle forward (bimanual handover)
Nominal: 66% (95% CI 55%–75%)Interpolation: 60% (95% CI 50%–70%)Extrapolation: 40% (95% CI 30%–50%)NominalInterp.Extrap.
e09 · Open the toolbox with a screwdriver
Nominal: 20% (95% CI 13%–29%)Interpolation: 12% (95% CI 7%–21%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.
e10 · Clamp electrical component
Nominal: 9% (95% CI 5%–17%)Interpolation: 0% (95% CI 0%–4%)Extrapolation: 0% (95% CI 0%–4%)NominalInterp.Extrap.