本文介绍了BertSum模型,通过在预训练语言模型BERT上进行编码器端到端微调,应用于文本摘要生成任务,实现高效的摘要提取。
BertSum 代码是针对纸质《Fine-tune BERT for Extractive Summarization》。新:请查看我们的更新结果(25/3/2019):
楷模ROUGE-1 ROUGE-2 ROUGE-L
变压器基线40.9 18.02 37.17
BERTSUM +分类器43.23 20.22 39.60
BERTSUM +变压器43.25 20.24 39.63
BERTSUM + LSTM 43.22 20.17 39.59
Python版本:此代码在Python3.6中运行。
软件包要求:pytorch, pytorch_pretrained_bert, tensorboardX, 多进程pyrouge
一些代码是从ONMT借来的。