ylwu commited on
Commit
f74312f
·
verified ·
1 Parent(s): 0052d73

update bibtex

Browse files
Files changed (1) hide show
  1. README.md +14 -4
README.md CHANGED
@@ -9,7 +9,7 @@ library_name: diffusers
9
  ---
10
 
11
  <h1 align="center">
12
- Ranking-based Preference Optimization </br> for Diffusion Models from Implicit User Feedback
13
  </h1>
14
 
15
  We present a learning framework that aligns text-to-image diffusion models with human preferences through inverse reinforcement learning and a balance of offline and online training.
@@ -18,9 +18,6 @@ We present a learning framework that aligns text-to-image diffusion models with
18
  ```python
19
  import torch
20
  from diffusers import StableDiffusionPipeline, UNet2DConditionModel
21
- from dotenv import load_dotenv
22
-
23
- load_dotenv()
24
 
25
  unet = UNet2DConditionModel.from_pretrained(
26
  "ylwu/diffusion-dro-sd1.5",
@@ -39,5 +36,18 @@ image = pipe(prompt).images[0]
39
  image.save("example.png")
40
  ```
41
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  ## License
43
  The model is licensed under the [CreativeML Open RAIL-M License](https://huggingface.co/spaces/CompVis/stable-diffusion-license).
 
9
  ---
10
 
11
  <h1 align="center">
12
+ [NeurIPS 2025] Ranking-based Preference Optimization </br> for Diffusion Models from Implicit User Feedback
13
  </h1>
14
 
15
  We present a learning framework that aligns text-to-image diffusion models with human preferences through inverse reinforcement learning and a balance of offline and online training.
 
18
  ```python
19
  import torch
20
  from diffusers import StableDiffusionPipeline, UNet2DConditionModel
 
 
 
21
 
22
  unet = UNet2DConditionModel.from_pretrained(
23
  "ylwu/diffusion-dro-sd1.5",
 
36
  image.save("example.png")
37
  ```
38
 
39
+ ## Citation
40
+ ```
41
+ @misc{wu2025rankingbasedpreferenceoptimizationdiffusion,
42
+ title={Ranking-based Preference Optimization for Diffusion Models from Implicit User Feedback},
43
+ author={Yi-Lun Wu and Bo-Kai Ruan and Chiang Tseng and Hong-Han Shuai},
44
+ year={2025},
45
+ eprint={2510.18353},
46
+ archivePrefix={arXiv},
47
+ primaryClass={cs.CV},
48
+ url={https://arxiv.org/abs/2510.18353},
49
+ }
50
+ ```
51
+
52
  ## License
53
  The model is licensed under the [CreativeML Open RAIL-M License](https://huggingface.co/spaces/CompVis/stable-diffusion-license).