Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Human Preference Score: Better Aligning Text-to-Image Models with Human Preference

About

Recent years have witnessed a rapid growth of deep generative models, with text-to-image models gaining significant attention from the public. However, existing models often generate images that do not align well with human preferences, such as awkward combinations of limbs and facial expressions. To address this issue, we collect a dataset of human choices on generated images from the Stable Foundation Discord channel. Our experiments demonstrate that current evaluation metrics for generative models do not correlate well with human choices. Thus, we train a human preference classifier with the collected dataset and derive a Human Preference Score (HPS) based on the classifier. Using HPS, we propose a simple yet effective method to adapt Stable Diffusion to better align with human preferences. Our experiments show that HPS outperforms CLIP in predicting human choices and has good generalization capability toward images generated from other models. By tuning Stable Diffusion with the guidance of HPS, the adapted model is able to generate images that are more preferred by human users. The project page is available here: https://tgxs002.github.io/align_sd_web/ .

Xiaoshi Wu, Keqiang Sun, Feng Zhu, Rui Zhao, Hongsheng Li• 2023

Related benchmarks

TaskDatasetResultRank
Human Preference EvaluationImageReward (test)
Preference Accuracy0.612
18
Human Preference EvaluationHPD v2 (test)
Preference Accuracy73.1
18
Text-to-Image GenerationDiffusionDB Real User Prompts 466 prompts (test)
Win Count1.39e+3
7
Text-to-Image GenerationMT Bench 90 prompts (test)
Total Wins735
7
Human Preference EvaluationMHP Overall (test)
Preference Accuracy65.5
7
Human preference predictionMHP dataset
Overall65.51
7
Human Preference EvaluationHuman Preference Evaluation 371 prompts (test)
Recall @139.89
3
Human Preference EvaluationHuman Preference Evaluation 466 prompts (test)
Preference Accuracy60.79
3
Showing 8 of 8 rows

Other info

Follow for update