arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.02205cs.CL
收藏

Attention Head Masking for Inference Time Content Selection in Abstractive Summarization

University of Michigan(密歇根大学)

Shuyang Cao, Lu Wang

英文摘要:

How can we effectively inform content selection in Transformer-based abstractive summarization models? In this work, we present a simple-yet-effective attention head masking technique, which is applied on encoder-decoder attentions to pinpoint salient content at inference time. Using attention head masking, we are able to reveal the relation between encoder-decoder attentions and content selection behaviors of summarization models. We then demonstrate its effectiveness on three document summarization datasets based on both in-domain and cross-domain settings. Importantly, our models outperform prior state-of-the-art models on CNN/Daily Mail and New York Times datasets. Moreover, our inference-time masking technique is also data-efficient, requiring only 20% of the training samples to outperform BART fine-tuned on the full CNN/DailyMail dataset.

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

↑