Tag: align-prop
- Aligning Text-to-Image Diffusion Models with Reward Backpropagation (10 Jul 2023)
This is my reading note for Aligning Text-to-Image Diffusion Models with Reward Backpropagation. This paper proposes a method how to train diffusion model for a given reward function in a memory efficient way, especially it utilities Lora and checkpoints . To avoid model collapse, it also proposes to randomly truncate number of steps.