Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

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 백본에서 공간 배치 충실도와 구성 일반화를 크게 향상시켜 새로운 최첨단 결과를 달성했습니다.

시사점, 한계점

시사점:
강화학습 기반 리프롬프팅을 통해 짧고 불완전한 프롬프트로도 높은 충실도의 이미지 생성 가능
다양한 T2I 백본에서 공간 배치 충실도 및 구성 일반화 개선
인간 주석 데이터 없이 엔드투엔드 학습 가능
기존 T2I 모델 성능 향상에 기여, 새로운 최첨단 성능 달성
한계점:
보상 모델의 설계가 성능에 큰 영향을 미칠 수 있음. 보상 모델의 최적화가 중요한 과제.
실험은 특정 T2I 백본과 벤치마크에 국한되어, 일반화 성능에 대한 추가적인 검증 필요.
복잡한 시각적 내용이나 추상적인 프롬프트에 대한 성능 평가가 필요.
👍