Skip to content

Latest commit

 

History

History
94 lines (73 loc) · 4.58 KB

File metadata and controls

94 lines (73 loc) · 4.58 KB

Custom PPO pilot results

All pilots use PushCube-v1, training seed 9351, and the same fixed deterministic 20-episode evaluator. Checkpoints are evaluated every 500,000 interactions and the best one is selected by success rate, then final distance, then return. One seed is useful evidence, but not a final comparison.

3M reward-alignment comparison

Both 3M conditions use 4,096 CUDA environments, four rollout steps, entropy coefficient 0.005, four update epochs, and linear learning-rate decay.

Measure Stabilized dense reward Dense reward minus 0.01/step
Requested / executed interactions 3,000,000 / 3,014,656 3,000,000 / 3,014,656
Best selected success rate 85% (17/20) 85% (17/20)
Final checkpoint success rate 65% 85%
Best-policy mean episode length 20.8 33.6
Best-policy mean steps to success 15.6 30.7
Best-policy final cube-to-goal distance 0.1026 m 0.1006 m
Training throughput 19,032 steps/s 18,490 steps/s

Checkpoint comparison

Dense-control checkpoint evolution

Every row is a checkpoint evaluated on the same fixed 20 seeds. Return is the unmodified ManiSkill dense return used for diagnosis; it is not the selection criterion. Lower distance is better. Action standard deviation (sigma) is the policy's exploration scale: its gradual decline is expected, while a sudden fall near zero would indicate premature loss of exploration.

Interactions Success Return Final distance Action sigma Learning rate
507,904 75% 7.90 0.1057 m 0.910 2.51e-4
1,015,808 85% 5.47 0.1002 m 0.832 2.01e-4
1,507,328 80% 6.47 0.1131 m 0.762 1.52e-4
2,015,232 35% 9.72 0.2345 m 0.703 1.01e-4
2,506,752 55% 8.05 0.1368 m 0.668 5.22e-5
3,014,656 65% 7.55 0.1114 m 0.655 1.63e-6

Dense-control checkpoint curves

Interpretation

The per-step cost is intended to make reward favour task completion without dithering. In this pilot, it does not improve the peak success rate: both conditions select a policy with 17 successes out of 20. It also does not yet meet the fewer-steps objective: the control policy reaches success faster on this evaluator.

Its useful signal is late-run stability. From 1.5M interactions onward, the time-cost run stays between 75% and 85% success, whereas the control drops to 35% at 2M and finishes at 65%. This is one stochastic training seed, so it is evidence for a follow-up—not a general conclusion. The primary 50M reference should remain the standard dense-reward configuration; repeat this ablation for seeds 9351, 4796, and 1788 before treating it as an improvement.

The evaluated local videos are:

  • artifacts/evaluations/custom-ppo-cuda-stabilized-3M-control-best/videos/0.mp4
  • artifacts/evaluations/custom-ppo-cuda-time-penalty-3M-best/videos/0.mp4

Those are generated artifacts and remain outside Git. Regenerate them with scripts/evaluate.py without --no-video.

5M conservative-update pilot

This dense-reward run changes the update-size controls together: learning rate 3e-4 to 1e-4, target KL 0.10 to 0.03, and update epochs four to two. All other environment, network, rollout, entropy, checkpoint-selection, and evaluation settings are unchanged.

Measure Conservative 5M result
Requested / executed interactions 5,000,000 / 5,013,504
Selected checkpoint 4,505,600 interactions
Selected and independently evaluated success rate 95% (19/20)
Success rate from 3.01M through 5.01M 95% at every checkpoint
Mean episode length 14.95
Mean steps to success 13.11
Final cube-to-goal distance 0.0948 m
Action sigma, first to last checkpoint 0.983 to 0.907
Approximate KL, first to last checkpoint 0.0106 to 0.0023

Conservative PPO checkpoint curves

Representative selected-policy rollout

The success curve rises from 0% at 508k to 55% at 1.02M, 85% at 1.51M, 90% at 2.51M, and 95% at 3.01M. It then stays at 95% through the final 5.01M checkpoint. This is substantially more stable than the previous dense 3M control, which fell from its 85% peak to 35% at 2.02M. The evidence supports the combined conservative package, not any individual parameter: repeat it on seeds 4796 and 1788 before making a general claim or choosing a final 50M setting.