arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2603.18528cs.AI
收藏

基于相关性的多奖励优化用于组合生成

Correlation-Weighted Multi-Reward Optimization for Compositional Generation

Korea University(韩国大学);Korea Institute of Science and Technology(韩国科学技术院)

Jungmyung Wi, Hyunsoo Kim, Donghyun Kim

AI总结:

本文提出相关性加权多奖励优化方法,通过调整概念奖励权重提升多概念生成效果,改进了SD3.5和FLUX.1-dev模型在多概念基准测试中的表现。

摘要:

文本到图像模型生成的图像与自然语言提示高度匹配,但组合生成一直是核心挑战。模型难以在单一提示中满足多个概念,常遗漏某些概念导致部分成功。此类失败凸显了在奖励优化中联合优化多个概念的难度,其中竞争性概念可能相互干扰。为此,我们提出相关性加权多奖励优化(\ours),通过利用概念奖励之间的相关结构,自适应地加权每个属性概念。通过考虑概念间的相互作用,\ours平衡了竞争性的奖励信号,并强调那些在样本间部分满足但不一致生成的概念,从而提升组合生成能力。具体而言,我们将多概念提示分解为预定义的概念组(例如,对象、属性和关系),并通过专用奖励模型获取每个概念的奖励信号。然后我们自适应地重新加权这些奖励,利用基于相关性的难度估计,为冲突或难以满足的概念分配更高的权重。通过将优化聚焦于每个组中最困难的概念,\ours鼓励模型同时一致地满足所有请求的属性。我们应用该方法训练了最先进的扩散模型SD3.5和FLUX.1-dev,并在具有挑战性的多概念基准测试中(包括ConceptMix、GenEval 2和T2I-CompBench)展示了持续改进。

英文摘要:

Text-to-image models produce images that align well with natural language prompts, but compositional generation has long been a central challenge. Models often struggle to satisfy multiple concepts within a single prompt, frequently omitting some concepts and resulting in partial success. Such failures highlight the difficulty of jointly optimizing multiple concepts during reward optimization, where competing concepts can interfere with one another. To address this limitation, we propose Correlation-Weighted Multi-Reward Optimization (CMO), a framework that leverages the correlation structure among concept rewards to adaptively weight each attribute concept in optimization. By accounting for interactions among concepts, CMO balances competing reward signals and emphasizes concepts that are partially satisfied yet inconsistently generated across samples, improving compositional generation. Specifically, we decompose multi-concept prompts into pre-defined concept groups (e.g., objects, attributes, and relations) and obtain reward signals from dedicated reward models for each concept. We then adaptively reweight these rewards, assigning higher weights to conflicting or hard-to-satisfy concepts using correlation-based difficulty estimation. By focusing optimization on the most challenging concepts within each group, CMO encourages the model to consistently satisfy all requested attributes simultaneously. We apply our approach to train state-of-the-art diffusion models, SD3.5 and FLUX.1-dev, and demonstrate consistent improvements on challenging multi-concept benchmarks, including ConceptMix, GenEval 2, and T2I-CompBench. The code is available at https://github.com/TheDarkKnight-21th/CMO.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑