InstructEngine: Instruction-driven Text-to-Image Alignment

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Lu, Xingyu, Hu, Yuhang, Zhang, YiFan, Jiang, Kaiyu, Liu, Changyi, Zhang, Tianke, Wang, Jinpeng, Yuan, Chun, Wen, Bin, Yang, Fan, Gao, Tingting, Zhang, Di
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912336921493504
author Lu, Xingyu
Hu, Yuhang
Zhang, YiFan
Jiang, Kaiyu
Liu, Changyi
Zhang, Tianke
Wang, Jinpeng
Yuan, Chun
Wen, Bin
Yang, Fan
Gao, Tingting
Zhang, Di
author_facet Lu, Xingyu
Hu, Yuhang
Zhang, YiFan
Jiang, Kaiyu
Liu, Changyi
Zhang, Tianke
Wang, Jinpeng
Yuan, Chun
Wen, Bin
Yang, Fan
Gao, Tingting
Zhang, Di
contents Reinforcement Learning from Human/AI Feedback (RLHF/RLAIF) has been extensively utilized for preference alignment of text-to-image models. Existing methods face certain limitations in terms of both data and algorithm. For training data, most approaches rely on manual annotated preference data, either by directly fine-tuning the generators or by training reward models to provide training signals. However, the high annotation cost makes them difficult to scale up, the reward model consumes extra computation and cannot guarantee accuracy. From an algorithmic perspective, most methods neglect the value of text and only take the image feedback as a comparative signal, which is inefficient and sparse. To alleviate these drawbacks, we propose the InstructEngine framework. Regarding annotation cost, we first construct a taxonomy for text-to-image generation, then develop an automated data construction pipeline based on it. Leveraging advanced large multimodal models and human-defined rules, we generate 25K text-image preference pairs. Finally, we introduce cross-validation alignment method, which refines data efficiency by organizing semantically analogous samples into mutually comparable pairs. Evaluations on DrawBench demonstrate that InstructEngine improves SD v1.5 and SDXL's performance by 10.53% and 5.30%, outperforming state-of-the-art baselines, with ablation study confirming the benefits of InstructEngine's all components. A win rate of over 50% in human reviews also proves that InstructEngine better aligns with human preferences.
format Preprint
id arxiv_https___arxiv_org_abs_2504_10329
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle InstructEngine: Instruction-driven Text-to-Image Alignment
Lu, Xingyu
Hu, Yuhang
Zhang, YiFan
Jiang, Kaiyu
Liu, Changyi
Zhang, Tianke
Wang, Jinpeng
Yuan, Chun
Wen, Bin
Yang, Fan
Gao, Tingting
Zhang, Di
Computer Vision and Pattern Recognition
Reinforcement Learning from Human/AI Feedback (RLHF/RLAIF) has been extensively utilized for preference alignment of text-to-image models. Existing methods face certain limitations in terms of both data and algorithm. For training data, most approaches rely on manual annotated preference data, either by directly fine-tuning the generators or by training reward models to provide training signals. However, the high annotation cost makes them difficult to scale up, the reward model consumes extra computation and cannot guarantee accuracy. From an algorithmic perspective, most methods neglect the value of text and only take the image feedback as a comparative signal, which is inefficient and sparse. To alleviate these drawbacks, we propose the InstructEngine framework. Regarding annotation cost, we first construct a taxonomy for text-to-image generation, then develop an automated data construction pipeline based on it. Leveraging advanced large multimodal models and human-defined rules, we generate 25K text-image preference pairs. Finally, we introduce cross-validation alignment method, which refines data efficiency by organizing semantically analogous samples into mutually comparable pairs. Evaluations on DrawBench demonstrate that InstructEngine improves SD v1.5 and SDXL's performance by 10.53% and 5.30%, outperforming state-of-the-art baselines, with ablation study confirming the benefits of InstructEngine's all components. A win rate of over 50% in human reviews also proves that InstructEngine better aligns with human preferences.
title InstructEngine: Instruction-driven Text-to-Image Alignment
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2504.10329