arXiv:2410.06084cs.LG2024-10ICLR被引 9

让生成模型在不增加计算量的情况下,同时提升音乐生成的质量与多样性。

Diversity-Rewarded CFG Distillation

  • 用蒸馏+强化学习奖励多样性,让模型无需依赖CFG也能生成优质内容。
  • 在MusicLM上实现质量-多样性权衡优于传统CFG方法,人类评测更认可。
  • 支持模型权重合并,可灵活控制生成结果的风格与创新程度。

生成模型正深刻改变音乐等创造性领域,推理时的无分类器引导(CFG)虽关键,但使推理成本翻倍,并限制生成内容的原创性与多样性。本文提出一种多样性奖励的CFG蒸馏方法,通过两个训练目标优化:(1)蒸馏目标,使模型独立生成接近CFG增强预测的结果;(2)带多样性奖励的强化学习目标,鼓励对同一提示生成多样化输出。微调后模型可在无推理开销下生成高质量、多样化的结果,并支持权重融合策略:通过插值两个模型(一个重质量,一个重多样性),部署时可动态调节质量-多样性平衡,甚至进一步提升性能。我们在MusicLM(Agostinelli et al., 2023)上进行大量实验,结果表明本方法在质量-多样性帕累托最优上超越了CFG。人类评估显示,微调后合并的模型生成样本在质量和多样性上均优于基线模型加CFG。更多生成示例见 https://google-research.github.io/seanet/musiclm/diverse_music/。

原文摘要 · Abstract (English)

Generative models are transforming creative domains such as music generation, with inference-time strategies like Classifier-Free Guidance (CFG) playing a crucial role. However, CFG doubles inference cost while limiting originality and diversity across generated contents. In this paper, we introduce diversity-rewarded CFG distillation, a novel finetuning procedure that distills the strengths of CFG while addressing its limitations. Our approach optimises two training objectives: (1) a distillation objective, encouraging the model alone (without CFG) to imitate the CFG-augmented predictions, and (2) an RL objective with a diversity reward, promoting the generation of diverse outputs for a given prompt. By finetuning, we learn model weights with the ability to generate high-quality and diverse outputs, without any inference overhead. This also unlocks the potential of weight-based model merging strategies: by interpolating between the weights of two models (the first focusing on quality, the second on diversity), we can control the quality-diversity trade-off at deployment time, and even further boost performance. We conduct extensive experiments on the MusicLM (Agostinelli et al., 2023) text-to-music generative model, where our approach surpasses CFG in terms of quality-diversity Pareto optimality. According to human evaluators, our finetuned-then-merged model generates samples with higher quality-diversity than the base model augmented with CFG. Explore our generations at https://google-research.github.io/seanet/musiclm/diverse_music/.

生成模型音乐生成多样性蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。