OneSteer: Unlocking Steerable General Humanoid Motion Tracking via One-Step Diffusion Whole-Body Control

Anonymous authors

Unitree G1 · simulation in MuJoCo / mjlab

Teaser. Left: OneSteer tracking reference motions on a real Unitree G1. Right: the same frozen policy steered by test-time costs or body-group masks in simulation: passing under a low ceiling, stepping over bars, reaching and carrying, a weak knee motor, velocity commands, and partial references.
One frozen general tracker, steerable at test time. Left: tracking arbitrary references on a Unitree G1. Right: with identical weights, differentiable costs or body-group masks adapt the behavior zero-shot in simulation (blue: earlier executed poses, the latest in the robot’s own materials; gold: the reference pose at the critical moment).

Abstract

Deployment objectives often specify desired outcomes without providing revised trajectories, while parts of the original reference must still be followed. We present OneSteer, a one-step diffusion whole-body controller that combines general motion tracking with selective test-time steering. To accommodate new objectives, differentiable task costs guide optimization of the generator’s input noise with frozen weights, while body-group reference masks relax selected tracking requirements. Predicted state–action plans provide a shared interface for task optimization and feedback-conditioned execution. For nominal tracking, the model trained from scratch achieves the lowest local tracking error among the evaluated baselines on successful held-out AMASS and BONES clips. For steering, one frozen distilled controller adapts across diverse deployment objectives in simulation, raising the clean passage rate under a 1.25 m bar from 2% to 90%. On a physical Unitree G1, OneSteer completes all 27 nominal tracking trials across nine diverse whole-body motion sequences without falling. With a steering cost derived from a signed distance field built using only onboard LiDAR, it also passes under a 1.1 m low bar in four of five attempts and steps over a 12 cm hurdle without falling in all three attempts. Beyond control, prediction variation in the model trained from scratch ranks likely tracking failures without physics rollouts. This scoring method is more than 10× faster than simulation-based scoring with five rollouts per clip and guides targeted fine-tuning.

Track
One-step generator plans future states and actions at 50 Hz; an inverse-dynamics head executes the plan.
Steer
Gradients of any differentiable cost on the input noise; masked body groups yield, the rest keeps tracking.
Assess
Where plans from different noise disagree, the tracker usually fails: no simulator, no extra model.
Real robot

Running on a Unitree G1

OneSteer tracking reference motions on the real robot, exported to ONNX and run on the onboard Jetson Orin NX at 50 Hz. The overhead gantry is a slack safety line.

Highly dynamic motions

Backflips on the real robot, plain tracking with the same controller.

Steering on the robot

The same frozen policy with a whole-body distance cost from a LiDAR map built before each run and held fixed. It passed under a 1.1 m bar in four of five attempts and stepped over a 12 cm hurdle without falling in all three (once without touching it); the unsteered controller failed every attempt. With a dance reference instead of walking and the same cost, the robot keeps dancing while it passes under the bar (two of four attempts; the large arm swings of the dance disturb its balance while it crouches) and steps over the hurdle (all three attempts). In these trials the steered policy ran on a tethered RTX 4090 computer.

1.1 m low bar

unsteered
steered

12 cm hurdle

unsteered
steered

Low bar, while dancing

unsteered
steered

Hurdle, while dancing

unsteered
steered
Simulation rollouts

Steering a frozen tracker

Steering bends the plan

Each clip shows the generator’s 0.96 s plan as ghosts. Top: without steering (grey ghosts, red where the plan hits the obstacle; gold is the reference motion). Bottom: the same reference after test-time steering (blue ghosts). The cost moves the plan itself, and the robot executes the moved plan.

Bars, reference motion A

Bars, reference motion B

Bars, reference motion C

Low ceiling, reference motion A

Low ceiling, reference motion B

Low ceiling, reference motion C

Other objectives

Five tasks, each shown without (top) and with (bottom) a test-time cost: weak knee motor, tray balancing, high-five, velocity command and VR wrist targets.
The same frozen policy without (top) and with (bottom) a test-time cost: one clip each, same reference; earlier poses in grey without and blue with steering. Over the test sets, steering raises knee-fault survival from 0.71 to 0.79, halves ball loss (0.16 to 0.08) and lowers the VR wrist error from 24.3 to 16.4 cm.

Side by side

Every clip is a closed-loop physics rollout. Panels run left to right. The leftmost panel is the reference or the RL teacher. The others are OneSteer, without and with steering.

High-five

A hand-target cost, 12 descent steps. Only the arm group is masked to yield; the gait keeps tracking.

Tray balancing

Teacher, OneSteer without steering, OneSteer with a tray-tilt cost (0.3).

Weak knee motor

Teacher, OneSteer without steering, steered when told about the fault, steered with a fault detector.

Velocity commands

Joystick walk-and-turn command (vx 0.83 m/s, yaw rate −0.67 rad/s; green disc). Teacher, OneSteer with the leg reference masked and no steering, and OneSteer with a root-velocity cost. Measured velocities are printed under each panel.
Adding a condition

Terrain is one more input

The generator can take a new condition without changing its structure. We encode a 21×21 height scan around the robot into 36 terrain tokens and append them to the generator’s input sequence. The reference stays a flat-ground motion, the inverse-dynamics head is unchanged, and the model is fine-tuned with RL on stairs, slopes, boxes and rough ground.

Stairs up Stairs down Slope Boxes Rough ground
One terrain-conditioned model on five terrain families (simulation): stairs up, stairs down, slope, boxes, rough ground. Each panel overlays four poses, about 0.55 m apart, of one rollout (earlier ones in blue); the reference is on flat ground, and the policy sees the terrain only through the height-scan tokens.

Stairs up

Stairs down

Slope

Boxes

Boxes, another motion

Rough ground

Left: the flat-ground reference motion. Right: the terrain-conditioned model on the terrain shown. The label gives the terrain family, the difficulty level (L0–L9) and the final distance to the end of the reference path. These clips are picked from cells the model passes; the table below covers all cells.

Success (no fall, ends within 1 m) ↑FlatStairs downStairs upSlopeBoxesRoughAll
Without terrain input0.870.530.160.550.190.320.43
Terrain tokens switched off0.900.810.250.670.420.370.55
With terrain tokens0.970.990.911.000.860.760.90

88 walking and jogging clips that cross the terrain, each on six terrain families and ten difficulty levels, simulation only. All rows are means of four evaluation passes (the spread of the last column is about 0.02–0.05). “Switched off” keeps the fine-tuned weights and masks the tokens, so the gain comes from the scan. The terrain model starts from the tracker trained from scratch. On flat ground it tracks nearly as before (success 0.963 vs. 0.963, local error 27.6 vs. 26.0 mm), and steering still helps on hurdles, with about half the gain of the model without terrain input. We have not run this model on the real robot.

Method

Plan, then execute

Method overview: reference window and proprioception condition a one-step generator that maps noise to a plan of future states and actions; an inverse-dynamics head turns the plan into the action; costs back-propagate to the noise.

A deterministic tracker maps a reference to an action, so the only handle at test time is the reference itself. OneSteer instead models the future. The plan is regenerated from the robot's real state at every tick, so a cost applied to the plan acts in closed loop by construction.

Plain tracking needs one generator call (3.8 ms on an RTX 4090); the controller on the robot keeps two calls (5.3 ms). The real-time steering setting guides the second call with two gradient steps (10.6–13.7 ms on an RTX 4090; 14.4 ms on the Orin NX, timed offline), within the 20 ms period of the 50 Hz loop.

Plain tracking

A competitive general tracker

All methods run in one common harness: MuJoCo sim-to-sim, official checkpoints; 150 AMASS windows (held-out subjects) and all 512 BONES windows (unseen actors). Errors are on successful clips. On BONES, 18 windows need stairs, ladders or boxes that the flat-ground harness does not have; every method fails them, so success is capped at 96.5%. Best in bold, second underlined.

AMASSBONES
MethodSucc. (%) ↑El (mm) ↓Eg med. (mm) ↓Succ. (%) ↑El (mm) ↓Eg med. (mm) ↓Steerable
SONIC (release)98.731.999.295.526.856.2no
SONIC (v1.1)98.030.5126.295.127.467.8no
HoloMotion95.343.5119.194.338.796.1no
HEFT (pmg)92.729.4127.095.323.367.1no
Humanoid-GPT95.344.5254.391.031.8106.1no
MimicLite Huge (teacher)94.029.9138.695.327.068.0no
OneSteer (distilled)95.331.4135.594.927.370.0yes
OneSteer (RL from scratch)96.726.5107.696.522.655.0yes

The distilled model is on par with its RL teacher. The model trained from scratch with RL, with no teacher, has the lowest local error El on both sets, and on BONES it fails only the 18 scene-dependent windows.

Steering vs. editing the reference

Steering inside the generator beats editing the reference

79 walking motions of unseen actors in simulation, three runs per steered condition. The RL teacher tracks a reference edited with the same cost, offline or at every tick (closed loop); J: task-group joints, R: also root height and pitch. OneSteer is steered inside the generator and tracks the original reference. Hurdles: feasible passes may touch the hurdle with a foot, clean passes do not. Latency per control step on one RTX 4090. Bold: best real-time row (95th percentile within the 20 ms period of the 50 Hz loop on the robot’s Orin NX); underlined: best slower row.

Hurdles ↑Low bar: clean pass ↑
Methodfeasibleclean1.30 m1.25 m1.20 mReach hit ↑Latency (ms) ↓
Teacher, original reference0.480.010.360.020.000.012.5
Teacher, offline edit (J)0.580.010.720.060.000.382.5
Teacher, offline edit (J+R)0.470.000.550.030.000.312.5
Teacher, closed-loop edit (J)0.570.020.870.110.020.60330.5
Teacher, closed-loop edit (J+R)0.450.000.670.040.020.60277
Teacher, tracks OneSteer's steered plan (J)0.530.100.860.740.510.3710.6–13.7
OneSteer, unsteered0.400.000.230.020.000.025.3
OneSteer, 2nd call, 2 steps (real-time)0.680.110.950.900.560.5710.6–13.7
OneSteer, both calls, 2 + 2 steps0.660.120.950.840.520.5417.2–20.5
OneSteer, 6 steps + line search per call0.740.210.980.830.560.62100–112

The edited reference asks for crouches the deterministic tracker cannot execute, so editing even at every tick stays at 0.11 under a 1.25 m bar, while steering the generator in real time reaches 0.90. A planner followed by a separate tracker also falls short: the teacher tracking OneSteer's own steered plan passes the 1.25 m bar in 0.74 of the trials, so the gain also relies on the action head trained with the generator.

Self-assessment without a simulator

The tracker knows which motions it cannot track

Give the generator the same reference and proprioception with K = 8 different noise samples. If the model has learned a motion well, the eight plans agree. If it has not, they spread out. We use the 95th percentile, over time, of the per-joint standard deviation of the planned joint positions as the score. It needs only the reference and the model: no simulator rollout and no extra network. Scoring 100 clips takes about 7 seconds.

Evaluation set: 7,884 library clips, including those the base policy finds hardest. Labels: whether the RL-from-scratch tracker (60k iterations) falls in mjlab. 537 clips fail (6.8%).

91%
of the top-2% most uncertain clips really fail (13× the 6.8% base rate)
74%
real failures among the top 329 clips, vs. 50% for plan deviation and 34% for reference atypicality
48±4%
of the tracker's 501 real failures fixed by fine-tuning on the 329 clips it is least sure about (three training seeds), against 27±2% for random clips
11–28×
cheaper than labeling clips with five simulated rollouts each, and no simulator needed

Failure rate in the most uncertain clips

Clips ranked by score, in bins of 2% of the set. Dashed line: base rate 6.8%.

AUC for real failures

All scores are simulator-free. Orange: needs only the diffusion generator. Grey: needs a critic, a world model or the library.

Precision and recall at the top of the ranking

TopNoise spreadPlan deviationCriticSpread + critic
1%0.91 / 0.130.87 / 0.131.00 / 0.151.00 / 0.15
2%0.91 / 0.270.68 / 0.200.85 / 0.250.97 / 0.28
3%0.79 / 0.350.58 / 0.250.74 / 0.330.86 / 0.38
5%0.68 / 0.500.44 / 0.330.57 / 0.420.67 / 0.49
10%0.41 / 0.610.35 / 0.510.37 / 0.540.41 / 0.60
20%0.24 / 0.690.22 / 0.640.23 / 0.670.24 / 0.71

Precision / recall. Imagined world-model rollouts reach 0.93 at the top 2% and 0.71 / 0.52 at 5%, close to the noise spread. A threshold picked on half of the chunks (top ~6%) gives precision 0.61 and recall 0.49 on the other half.

Uncertainty and rarity are complementary

Failure rate when each score is split at its median. Rarity is the kNN distance of a clip's reference statistics to the rest of the library.

common motion
rare motion
low spread
0.4%
2,556 clips
6.1%
1,386 clips
high spread
3.5%
1,386 clips
15.5%
2,556 clips

Combined AUC 0.87. Rarity alone ranks broadly (0.81), but the spread is twice as precise at the very top.

From diagnosis to improvement

Fine-tune the RL-from-scratch tracker for 8k iterations on the 329 clips a score ranks hardest, then re-run 7,788 library clips in simulation and compare clip by clip with the base model (501 failures). To keep what it already masters, half of the environments replay the library with a consolidation loss toward the frozen base and keep their RL return. Mean ± s.d. over three training seeds (base model: two re-evaluations).

Clips selected byFailures fixed ↑… outside the list ↑Broken ↓Library failures ↓Unseen families ↑Seen families ↑
None (base re-evaluated)12.2±4.8%12.2±4.8%0.8±0.3%6.4%24.5±2.6%9.7±2.3%
Random clips26.9±2.0%27.0±2.0%1.6±0.2%6.2±0.1%34.8±3.8%17.2±6.5%
Kinematic rarity30.5±3.1%29.8±2.8%1.3±0.1%5.7±0.3%35.2±3.7%19.9±0.9%
RL critic value41.2±2.3%41.2±3.3%1.3±0.1%5.0±0.2%37.9±2.8%31.7±3.4%
Imagined world-model rollout46.3±2.8%42.1±4.1%1.3±0.1%4.7±0.2%43.0±1.0%39.2±0.9%
Noise spread (ours)48.3±4.2%44.2±5.6%1.7±0.2%5.0±0.4%47.9±2.3%38.7±4.3%

Unseen and seen families: success on held-out failures of families the lists never contain, or of trained families. The noise spread fixes nearly twice as many failures as random clips at the same budget and generalizes best to unseen families; imagined rollouts are comparable on the library but need a policy rollout and a learned world model. The RL return on mastered clips keeps regressions near the noise floor: with the consolidation loss alone, 4.6% of tracked clips break.

Citation

BibTeX

@misc{onesteer2027,
  title     = {OneSteer: Unlocking Steerable General Humanoid Motion Tracking
               via One-Step Diffusion Whole-Body Control},
  author    = {Anonymous},
  year      = {2027}
}