DualHeads: Reward-tilted few-step diffusion made simple

Yuanzhi Zhu, Eleftherios Tsonis, Simo Alami Chehboune +3

Distilled diffusion models generate high-quality images in a few sampling steps, but aligning them with task requirements and human preferences is difficult. Methods developed for…