Last updated
尽管在语言理解领域模型鲁棒性已经得到了广泛研究，但序列到序列（Seq2Seq）生成模型的鲁棒性研究相对较少。当前最先进的预训练Seq2Seq模型（如BART）在文本生成任务中仍然存在脆弱性，导致生成文本的忠实度和信息量显著下降。这促使研究者提出新的对抗性增强框架，以提高Seq2Seq模型的鲁棒性，从而获得更准确和信息丰富的文本生成。
现有的对抗性学习方法在语言理解任务上取得了一定的成果，但在Seq2Seq生成任务上的效果尚未得到充分验证。此外，现有的对抗性样本生成方法主要关注于模型理解任务，而没有在当前预训练的Seq2Seq模型上进行广泛研究。
本文提出了一种新的对抗性增强框架AdvSeq，它通过在训练过程中自动构建两种类型的对抗性增强样本来提高Seq2Seq模型的鲁棒性：隐式对抗样本（通过扰动词表示）和显式对抗样本（通过词交换）。AdvSeq利用训练过程中的梯度信息来构造这些样本，并采用KL散度为基础的损失函数进行训练，以提高词表示空间中的不变性。
首次对预训练Seq2Seq模型的鲁棒性进行了定量分析，并揭示了其与生成任务中文本的忠实度和信息量之间的紧密联系。
提出了AdvSeq框架，通过增强模型的鲁棒性，有效地提高了各种生成任务中文本的忠实度和信息量。
在三个流行的文本生成任务（文本摘要、表格到文本、对话生成）上进行了自动和人工评估，验证了AdvSeq在忠实度和信息量方面显著优于多个强基线。
在文本摘要、表格到文本和对话生成任务上进行了广泛的实验。实验结果表明，AdvSeq能够有效地提高Seq2Seq模型对对抗性样本的鲁棒性，从而在各种文本生成任务中产生更好的信息量和忠实度。
AdvSeq在提高Seq2Seq模型的鲁棒性方面表现出色，尤其是在面对对抗性样本时，能够生成更忠实和信息丰富的文本。与其他基线方法相比，AdvSeq在自动和人工评估中都取得了显著的优势。
本文首次对预训练Seq2Seq模型的鲁棒性进行了定量分析，并提出了AdvSeq框架来增强模型的鲁棒性，从而改善文本生成的忠实度和信息量。实验结果证明了AdvSeq在多个文本生成任务中的有效性，为未来在这一领域的研究提供了新的视角和方法。
注：
AdvSeq（Adversarial augmentation framework for Sequence-to-Sequence generation）能够增强Seq2Seq模型的鲁棒性，主要通过以下几个关键步骤和机制实现：
对抗性样本的自动构建：
AdvSeq在训练过程中自动构建两种类型的对抗性样本：隐式对抗样本（AdvGrad）和显式对抗样本（AdvSwap）。
这些样本通过在输入数据中引入微小的、对抗性的扰动来模拟潜在的输入变化，从而使模型在面对真实世界中的噪声和变化时更加鲁棒。
隐式对抗样本（AdvGrad）：
AdvGrad利用模型的梯度信息来直接扰动词表示，寻找能够显著影响生成过程的强对抗性扰动。
通过梯度上升方法，AdvSeq在保持输入意义的同时，找到能够最大化损失函数的扰动，从而生成具有挑战性的对抗样本。
显式对抗样本（AdvSwap）：
AdvSwap通过识别输入中对生成目标贡献最大的词（称为显著词），并在这些词的词嵌入空间中寻找语义上保持一致的替代词。
这种方法不仅增加了样本的多样性，而且通过保持原始输入的意义，确保了对抗样本的挑战性。
KL散度为基础的损失函数：
AdvSeq采用基于KL散度的损失函数来训练模型，这种损失函数鼓励模型在词表示空间中寻找更高不变性的表示。
通过最小化输出词分布之间的KL散度，模型学习到的表示更加鲁棒，能够更好地抵抗输入中的微小变化。
FreeLB训练策略：
AdvSeq采用了“Free” Large-Batch Adversarial Training（FreeLB）策略，这是一种高效的对抗性训练方法。
在每次训练步骤中，AdvSeq首先前向传播和反向传播原始损失函数，然后利用保存的梯度信息来构建AdvGrad和AdvSwap样本，并累积它们的梯度。
最后，模型参数通过累积的梯度进行更新，这种策略使得AdvSeq能够有效地利用对抗性样本进行训练。
通过上述机制，AdvSeq能够有效地提高Seq2Seq模型在面对对抗性输入时的鲁棒性，从而在各种文本生成任务中产生更忠实和信息丰富的输出。这种方法不仅提高了模型对输入扰动的抵抗力，还增强了模型在实际应用中的泛化能力。
本文针对Seq2Seq模型在文本生成任务中的脆弱性问题，提出了AdvSeq框架，通过对抗性增强样本的构建和训练，显著提高了模型的鲁棒性。AdvSeq在多个文本生成任务上的表现优于现有方法，显示出在提高生成文本质量方面的潜力。这项工作不仅为Seq2Seq模型的鲁棒性研究提供了新的视角，也为文本生成任务的改进提供了实用的解决方案。
