arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.13299cs.CLcs.LG
收藏

Mix and Match:基于能量语言模型的无学习可控文本生成

Mix and Match: Learning-free Controllable Text Generation using Energy Language Models

University of California San Diego(加州大学圣迭戈分校);Toyota Technological Institute at Chicago (TTIC)(芝加哥丰田技术研究院(TTIC))

Fatemehsadat Mireshghallah, Kartik Goyal, Taylor Berg-Kirkpatrick

摘要:

控制文本生成的最新工作要么需要基于属性的微调的基础语言模型(LM),或限制了参数化的属性映射,以兼容的基础自回归LM。在这项工作中,我们提出了Mix and Match LM,这是一种基于全局分数的可控文本生成方法,它结合了任意预训练的黑盒模型,用于在生成的文本中实现所需的属性,而不涉及任何关于黑盒模型的微调或结构假设。我们将可控生成的任务解释为从基于能量的模型中抽取样本,该模型的能量值是来自黑盒模型的分数的线性组合,黑盒模型分别负责流畅性、控制属性和对任何条件上下文的忠诚度。我们使用Metropolis-Hastings抽样方案从这个基于能量的模型中使用双向上下文和全局属性特征进行抽样。我们验证了我们的方法在各种控制生成和基于风格的文本修订任务的有效性,优于最近提出的方法,涉及额外的训练,微调,或限制性的假设模型的形式。

英文摘要:

Recent work on controlled text generation has either required attribute-based fine-tuning of the base language model (LM), or has restricted the parameterization of the attribute discriminator to be compatible with the base autoregressive LM. In this work, we propose Mix and Match LM, a global score-based alternative for controllable text generation that combines arbitrary pre-trained black-box models for achieving the desired attributes in the generated text without involving any fine-tuning or structural assumptions about the black-box models. We interpret the task of controllable generation as drawing samples from an energy-based model whose energy values are a linear combination of scores from black-box models that are separately responsible for fluency, the control attribute, and faithfulness to any conditioning context. We use a Metropolis-Hastings sampling scheme to sample from this energy-based model using bidirectional context and global attribute features. We validate the effectiveness of our approach on various controlled generation and style-based text revision tasks by outperforming recently proposed methods that involve extra training, fine-tuning, or restrictive assumptions over the form of models.

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

↑