Daily Arxiv

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

DRAGON: Distributional Rewards Optimize Diffusion Generative Models

Created by
  • Haebom
Category
Empty

저자

Yatong Bai, Jonah Casebeer, Somayeh Sojoudi, Nicholas J. Bryan

DRAGON: Distributional RewArds for Generative OptimizatioN

개요

DRAGON은 원하는 결과에 맞춰 미디어 생성 모델을 미세 조정하기 위한 유연한 프레임워크입니다. 전통적인 강화 학습(RLHF)이나 DPO와 같은 쌍별 선호도 접근 방식에 비해 더 유연하며, 개별 예제 또는 예제 분포를 평가하는 보상 함수를 최적화할 수 있습니다. DRAGON은 온라인 생성물을 수집하고, 이를 보상 함수로 평가하여 긍정 및 부정 샘플 세트를 구축한 다음, 이들 유한 세트 간의 대비를 활용하여 분포형 보상 최적화를 근사합니다. 오디오 도메인 텍스트-음악 확산 모델을 20개의 보상 함수(음악 미학 모델, CLAP 점수, Vendi 다양성, FAD 등)로 미세 조정하여 평가한 결과, 81.45%의 평균 승률을 달성했습니다. 예제 세트를 기반으로 한 보상 함수는 생성을 향상시키고 모델 기반 보상과 유사한 성능을 보였습니다. 적절한 예제 세트를 사용하면 인간 선호도 주석에 대한 학습 없이도 60.95%의 인간 투표 음악 품질 승률을 달성했습니다.

시사점, 한계점

다양한 보상 함수(개별, 분포)를 사용하여 미디어 생성 모델을 최적화하는 유연한 프레임워크 제공
교차 모달 인코더(CLAP)를 사용하여 텍스트-오디오 등 서로 다른 모달리티 간의 보상 함수 구성 가능
인간 선호도 주석 없이도 인간이 인지하는 품질 향상 가능
20개의 보상 함수에 대해 평균 81.45%의 높은 승률 달성
예제 세트를 사용한 보상 함수가 모델 기반 보상과 유사한 성능을 보임
인간 투표 음악 품질에서 60.95% 승률 달성
제공된 정보 내에서 한계점은 명시되지 않음
👍