RePrompt: Reasoning-Augmented Reprompting for Text-to-Image Generation via Reinforcement Learning
Created by
Haebom
저자
Mingrui Wu, Lu Wang, Pu Zhao, Fangkai Yang, Jianjin Zhang, Jianfeng Liu, Yuefeng Zhan, Weihao Han, Hao Sun, Jiayi Ji, Xiaoshuai Sun, Qingwei Lin, Weiwei Deng, Dongmei Zhang, Feng Sun, Qi Zhang, Rongrong Ji
개요
본 논문은 짧고 불완전한 프롬프트로부터 사용자 의도를 충실히 반영하는 이미지 생성에 어려움을 겪는 기존 텍스트-이미지(T2I) 생성 모델의 한계를 해결하기 위해, 강화 학습을 통해 명시적인 추론을 프롬프트 개선 과정에 도입하는 새로운 리프롬프팅 프레임워크인 RePrompt를 제안합니다. RePrompt는 수작업 규칙이나 스타일 재작성에 의존하는 대신, 이미지 수준 결과를 최적화하여 구조적이고 자기 반성적인 프롬프트를 생성하도록 언어 모델을 훈련합니다. 사용자 선호도, 의미적 정렬, 시각적 구성 측면에서 생성된 이미지를 평가하는 맞춤형 보상 모델은 프롬프트 생성을 개선하기 위한 간접적인 감독을 제공하며, 인간이 주석을 단 데이터 없이도 엔드투엔드 훈련을 가능하게 합니다. GenEval과 T2I-Compbench 실험 결과, RePrompt는 다양한 T2I 백본에서 공간 배치 충실도와 구성 일반화를 크게 향상시켜 새로운 최첨단 결과를 달성했습니다.
시사점, 한계점
•
시사점:
◦
강화학습 기반 리프롬프팅을 통해 짧고 불완전한 프롬프트로도 높은 충실도의 이미지 생성 가능