本系列博文主要介绍了在文本摘要领域神经网络模型的一些发展,主要基于如下几类模型:
- Basic Encoder-Decoder model
- Encoder-Decoder + Attention
- Deep Reinforced model
- Bert based model
- GAN based model
- consideration
Generative Adversarial Network Based Model
AAAI 2018 《Generative Adversarial Network for Abstractive Text Summarization》
github:textSumGAN
抽象式摘要生成任务目的是从原始输入文本中生成一段较短且精炼的可以表述原文大意的总结性文本,它不仅包含原文中存在的词或是短语,同时还可能生新的的短语和句子。因为如何做好抽象式摘要任务是近些年来自动摘要生成领域一个很重要的任务,不同的研究人员提出了不同的模型,但是以往的模型普遍无法很好的解决如下的这些挑战:
- 基于Seq2Seq的模型生成的简洁及通用的摘要通常包含大量的高频短语,导致模型难以较为精确的表达不同类型文本的大意
- 生成的摘要通常在语法和可读性上乏善可陈
- 以往Seq2Seq的模型通常是使用最大似然估计(MLE)来不断的预测当前文本的下一个词来逐步的生成摘要
传统上基于Seq2Seq的模型不仅无法真正的解决以上的诸多挑战,而且还有一些其他的问题:
- 评价指标不同于训练损失
- 训练阶段每个时间步编码器的输入是真实摘要中的词,而测试时每个时间步编码器的输入是上一时间生成的词,这就可能会导致错误不断的累积,最后生成的文本摘要不能使用
本文是采用GAN和RL的结合来解决离散数据的处理任务,G接受原始输入文本生成一段摘要,然后和真实摘要共同输入到D中,D这里作为一个文本分类器需要判断出输入的摘要是真实的还是G生成的。D会根据判断给出反馈,G使用策略梯度进行更新,经过不断的交替训练,G最终学会生成接近于真实摘要的文本。
Generator
假设原始输入文本记为
,G生成的摘要记为
。这里G采用的仍然是Encoder-Decoder的结构,Encoder使用的是双向LSTM,将
转换为隐状态序列
。Decoder使用基于Attention的单向LSTM,在每个时间步计算隐状态
和上下文向量
,最后将
和
拼接起来输入到全连接层和Softmax层中,得到每个时间步从词汇表中预测下一个词的概率
其中
都是需要学习的参数。
[仅是个人理解所画]
Discriminator
D这里是当作一个文本二分类器,判别输入的摘要文本是人产生的还是机器生成的。D使用CNN来处理输入序列,CNN中使用多个卷积核和变化的窗口大小来捕获不同的特征,并使用了max-pooling-over-time。最后将得到的特征输入到使用softmax的全连接层中得到判别为真实摘要的概率。
在解码阶段,先计算每个时间步的注意力分布
然后计算
最后得到
在训练过程中,G和D仍然采用传统GAN的对抗方式进行训练
D这里作为二分类器,损失函数使用交叉熵即可,G的损失函数包含两项,由策略梯度计算的损失项
和由最大似然估计得到的损失项
。
其中
是真实摘要的词序列。两者通过线性插值的方式得到完整的损失项
,
是缩放参数。
根据策略梯度的计算,参数
的更新为
其中
作为action-value函数,而且有
,同样使用SGD进行参数的更新,这里和常规使用GAN处理文本的方式并没有区别。
实验部分使用的数据集为CNN/Daily Mail Corpus,对比基准模型为:
- abstractive model(ABS)
- pointer-generator coverage networks (PGC)
- abstractive deep reinforced model (DeepRL)
评估指标为ROUGE-1、ROUGE-2、ROUGE-L和human evaluation。实验结果为:
实验证明了本文提出的模型可以产生可读性更好、多样性更好的摘要,相比于基准模型在多个评估指标上都取得了state-of-the-art的效果。
此外在实验中还使用了Trigram Avoidance 避免生成重复的短语,使用Quotation Weight Alleviation 减少直接从源文本摘取短语或句子的比例。
参考文献
[1] Bahdanau, D.; Cho, K.; and Bengio, Y. 2014. Neural machine translation by jointly learning to align and translate. arXiv preprint arXiv:1409.0473.
[2] See, A.; Liu, P. J.; and Manning, C. D. 2017. Get to the point: Summarization with pointer-generator networks. arXiv preprint arXiv:1704.04368.
[3] Kim, Y. 2014. Convolutional neural networks for sentence classification. arXiv preprint arXiv:1408.5882.
[4] Yu, L.; Zhang, W.; Wang, J.; and Seqgan, Y. Y. 2016. Sequence generative adversarial nets with policy gradient. arXiv preprint arXiv:1609.05473 2(3):5.
[5] Paulus, R.; Xiong, C.; and Socher, R. 2017. A deep reinforced model for abstractive summarization. arXiv preprint arXiv:1705.04304.
IEEE ICSC 2019 《GenerativeAdversarial Networks with Policy Gradient for Text Summarization》
这篇文章同样是使用GAN来解决抽象式摘要生成任务,与上一篇的不同之处在于引入了time-decay intra-emporal attention机制,以及在强化学习应用中使用了Stochastic Policy Gradient进行模型的更新。
抛开文章不看,首先来看一下Stochastic Policy Gradient(SPG)和之前的Policy Gradient由什么区别。
先前使用的policy gradient也称为Deterministic Policy Gradient,(DPG)。DPG的policy是从state到action的映射,因此模型的目标函数定义为
它只需要求一次积分,从它的梯度求解公式中也可以看出,DPG因需要考虑状态概率分布。
SPG中的policy是state到action概率分布的映射
,模型的目标函数为
它需要计算两次积分,从它的梯度求解式中可以看出,SPG同时考虑了状态概率分布以及动作概率分布。如果要进行学习训练,就需要大量的样本来覆盖整个二维的状态动作空间。
作者使用了一种更复杂的策略梯度求解方式嘛?为啥呢?
Generator
生成器同样是使用encoder-decoder模型,其中encoder使用双向LSTM,decoder使用单向LSTM,它和之前的模型并没有不同,唯一区别在于使用了time-decay intra-emporal attention,和前面 ICLR 2018 A Deep Reinforced Model for Abstractive Summarization中所使用的第二套注意力机制应该是一样的。
Discriminator
判别器使用CNN模型做二分类,正样本是文档和配套的人写的摘要,负样本为文档和生成的摘要,其他的并没有什么特别。
train
训练过程使用策略梯度进行更新,没什么不同。因为D只接收完整的序列,因此这里同样使用了SeqGAN中的蒙特卡洛搜索,最后的奖励值为N个奖励的平均值。
此外在Training stategies部分给出了详细的参数设置,还是有点借鉴意义。
实验在CNN/Daily Mail上做,评判指标是ROUGE。实验结果如下
从中可以看出GAN的确可以提升模型效果,而且intra-attention也是有用的。
ICCS 2018 《Adversarial Reinforcement Learning for Chinese Text summarization》
中科院在本文中提出了一种用于中文文档的摘要生成模型,由于中文和英文有一定的区别,所以在处理方式上还是有一定的差异。
模型如下所示
生成器同样采用了encoder-decoder模型,两部分都是使用了LSTM,在decoder中同样引入了注意力机制。
其中在decoder中使用的注意力机制称为Text-Attention。
它用于在将输入文本序列
送到RNN之前,计算公式为:
其中
为注意力矩阵,它将文本表示
转换为词嵌入的形式,
是标量。然后就没有介绍了,不是很懂。
训练部分就是使用策略梯度,没有什么不同。
实验
实验所使用的数据集为Large Scale Chinese Short Text Summarization Dataset (LCSTS)和NLPCC Evaluation Task 4。评价指标为ROUGE-1、ROUGE-2和ROUGE-L。
基准模型有
- Abs(basic Seq2Seq model)
- Abs+(Seq2Seq model + Attention)
- Abs + TA (Seq2Seq model + Text-Attention)
- DeepRL(Deep Reinforced model)
本文提出的模型记为ARL。
在数据预处理中,为了减少中文切词的效果优劣对后续结果的影响,这里大部分使用的是字(character)而不是词(word)。
在LCSTS关于word-level的实验中,为了减小词汇表,这里只保留了前150K个频繁词,其余的用UNK标记。此外,在两种实验中都使用了word embedding来表示输入。
在训练过程中还使用了如下的tricks帮助训练:
- teacher-forcing
- gradient clipping
- mini-batch
- learning rate decay
- pre-train
在输入到判别器的负样本中,一般是使用基于互信息的beam search生成,一半是从人写的摘要中采样。
两个数据集上的实验结果如下:
从结果中可以看出char-level的方式得到的效果更好,而且有证明了RL确实有用。
从具体实例中也可以看出ARLc方式得到的结果更好。
最后作者提出了两个思考方向:
- 如何结合汉语本身的特点构建模型
- 如何建立适用于文本摘要任务的无监督或半监督框架,减少对于高质量数据集的依赖
在整个过程中参考了很多网上的资料,由于忘记保存链接,这里就给出参考资料地址了,衷心感谢他人的付出~