selective-timestep-weighting-and-advantage-based-replay-for-sample-efficient-diffusion-rlhf-af5a35a4·1 events·first seen Aliases: Selective Timestep Weighting and Advantage-Based Replay for Sample-Efficient Diffusion RLHF
A new arXiv preprint proposes two complementary techniques to improve feedback efficiency in diffusion model RLHF: a per-timestep weighting scheme grounded in PPO convergence theory, and a replay mechanism that prioritizes informative trajectories to reduce redundant reward queries. Together, the methods achieve up to 6× improvement in sample efficiency over standard diffusion RLHF baselines under identical hyperparameter settings. The work addresses a practical bottleneck—feedback cost—that limits real-world deployment of RLHF-aligned diffusion models.