Rich Human Feedback for Text-to-Image Generation
Authors: name: Youwei Liang; name: Junfeng He; name: Gang Li; name: Peizhao Li; name: Arseniy Klimovskiy; name: Nicholas Carolan; name: Jiao Sun; name: Jordi Pont-Tuset; name: Sarah Young; name: Feng Yang; name: Junjie Ke; name: Krishnamurthy Dj Dvijotham; name: Katie Collins; name: Yiwen Luo; name: Yang Li; name: Kai J Kohlhoff; name: Deepak Ramachandran; name: Vidhya Navalpakkam
Published: 2023-12-15 22:18:38

Abstract: Recent Text-to-Image (T2I) generation models such as Stable Diffusion and
Imagen have made significant progress in generating high-resolution images
based on text descriptions. However, many generated images still suffer from
issues such as artifacts/implausibility, misalignment with text descriptions,
and low aesthetic quality. Inspired by the success of Reinforcement Learning
with Human Feedback (RLHF) for large language models, prior works collected
human-provided scores as feedback on generated images and trained a reward
model to improve the T2I generation. In this paper, we enrich the feedback
signal by (i) marking image regions that are implausible or misaligned with the
text, and (ii) annotating which words in the text prompt are misrepresented or
missing on the image. We collect such rich human feedback on 18K generated
images and train a multimodal transformer to predict the rich feedback
automatically. We show that the predicted rich human feedback can be leveraged
to improve image generation, for example, by selecting high-quality training
data to finetune and improve the generative models, or by creating masks with
predicted heatmaps to inpaint the problematic regions. Notably, the
improvements generalize to models (Muse) beyond those used to generate the
images on which human feedback data were collected (Stable Diffusion variants).