Abstract

Recent studies demonstrate that diffusion planners benefit from sparse-stepplanning over single-step planning. Training models to skip steps in theirtrajectories helps capture long-term dependencies without additional or memorycomputational cost. However, predicting excessively sparse plans degradesperformance. We hypothesize this temporal density threshold is non-uniformacross a temporal horizon and that certain parts of a planned trajectory shouldbe more densely planned. We propose Mixed Density Diffuser (MDD), a diffusionplanner where the densities throughout the horizon are tunable hyperparameters.MDD achieves a new SOTA across the Maze2D, Franka Kitchen, and Antmaze D4RLtask domains.

 

Quick Read (beta)