Latent Reward Registers for Diffusion Preference Alignment

Yuanshen Guan, Zipeng Feng, Chengru Song +2

Aligning diffusion models with human preferences usually relies on a sparse terminal reward evaluated on the final generated samples, which creates a severe temporal credit-assignm…