Skip to main content
QUICK REVIEW

[论文解读] Plug and Play Autoencoders for Conditional Text Generation

Florian Mai, Nikolaos Pappas|arXiv (Cornell University)|Oct 6, 2020
Topic Modeling参考文献 51被引用 4
一句话总结

本文提出Emb2Emb,一种即插即用的框架,通过在任意预训练文本自编码器的潜在空间中学习映射,利用对抗性损失和基于偏移的神经映射实现条件文本生成。该方法在风格迁移任务中达到最先进或具有竞争力的性能,训练速度比端到端模型快至四倍,同时仅需极少的标注数据,并在无监督设置下更好地保留内容。

ABSTRACT

Text autoencoders are commonly used for conditional generation tasks such as style transfer. We propose methods which are plug and play, where any pretrained autoencoder can be used, and only require learning a mapping within the autoencoder's embedding space, training embedding-to-embedding (Emb2Emb). This reduces the need for labeled training data for the task and makes the training procedure more efficient. Crucial to the success of this method is a loss term for keeping the mapped embedding on the manifold of the autoencoder and a mapping which is trained to navigate the manifold by learning offset vectors. Evaluations on style transfer tasks both with and without sequence-to-sequence supervision show that our method performs better than or comparable to strong baselines while being up to four times faster.

研究动机与目标

  • 通过利用预训练自编码器,在仅需极少标注数据的情况下实现条件文本生成。
  • 开发一种即插即用的方法,可与任意现有自编码器配合使用,无需重新训练。
  • 在无监督风格迁移任务中提升效率并改善内容保留。
  • 通过确保映射后的嵌入保持在自编码器的潜在流形上,维持生成质量。
  • 降低序列到序列生成中对并行数据的依赖和训练成本。

提出的方法

  • 该方法在自编码器的潜在空间内学习从输入句子嵌入到目标句子嵌入的映射Φ。
  • 它使用对抗性损失项ℒ_adv,以确保输出嵌入位于自编码器的流形上,从而保证生成的有效性。
  • 映射网络被设计为学习并应用偏移向量到输入嵌入,实现细粒度调整。
  • 在监督学习中,该方法最小化预测与真实目标嵌入之间的任务损失ℒ_task。
  • 该框架可兼容任意预训练自编码器,无论其训练时是否考虑了风格属性。
  • 在推理阶段,将学习到的映射与自编码器的编码器和解码器组合,以生成输出文本。

实验结果

研究问题

  • RQ1是否可以不重新训练或微调,直接使用预训练自编码器进行条件文本生成?
  • RQ2潜在空间中的映射如何改善风格迁移中的内容保留和生成质量?
  • RQ3嵌入空间中的对抗性正则化是否能维持有效性并提升性能?
  • RQ4基于偏移的映射架构在速度和准确率方面与端到端训练相比如何?
  • RQ5在低资源条件生成设置下,对无标注数据进行预训练能在多大程度上提升性能?

主要发现

  • Emb2Emb在文本简化(一种监督式风格迁移任务)中优于强基线模型,且训练速度比端到端模型快至四倍。
  • 在无监督情感迁移任务中,Emb2Emb比最先进方法更好地保留了输入内容,同时实现了相当的迁移准确率。
  • 该方法性能接近使用十倍以上标注数据的上界模型,证明了其在低资源设置下的强大泛化能力。
  • 对抗性损失项ℒ_adv对保持嵌入在自编码器潜在流形上至关重要,直接提升了生成质量。
  • 基于偏移的映射架构能够实现有效且稳定的输入嵌入调整,提升任务特定适应性。
  • Emb2Emb的即插即用特性使其可无缝集成到任意预训练自编码器中,实现大规模预训练的复用。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。