Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Preference-Diffusion

Preference Diffusion: Optimizing diffusion models with minimal human feedback

Introduction

Diffusion-based image generation models (Stable Diffusion) are good at generating high-quality images but the user has little to no control over the output. To solve this, we employ Direct Preference Optimization (DPO), to align the prompt to the generated output images. We augment this method with real-time minimal human feedback in the form of their output preference, hence narrowing the image space before applying DPO using Gaussian Mixture Modeling.

Method

image

Result

image

Web Interface Video

https://drive.google.com/file/d/1qt0nLLlfGLfcmPBHBcgGhF4dSBFDt3kK/view?usp=sharing

About

Preference Diffusion: Optimizing diffusion models with minimal human feedback

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages