One2Set：将关键短语作为集合生成
计算与语言
2021-05-25 v1 人工智能
摘要
近来，序列到序列模型通过将多个关键短语按预定义顺序拼接为训练时的目标序列，在关键短语生成（KG）任务上取得了显著进展。然而，关键短语本质上是一个无序集合而非有序序列。强加预定义顺序会在训练中引入错误偏置，从而会严重惩罚关键短语之间顺序的偏移。在本工作中，我们提出了一种无需预定义顺序来拼接关键短语的新训练范式 One2Set。为适配该范式，我们提出了一种新颖模型，其利用一组固定的已学习控制码作为条件来并行生成一组关键短语。为解决训练中每次预测与目标之间无对应的问题，我们提出了一种通过二部匹配的 步目标分配机制，其大幅提升了生成关键短语的多样性并降低了重复率。在多个基准上的实验结果表明，我们的方法显著优于最先进（state-of-the-art）方法。
引用
@article{arxiv.2105.11134,
title = {One2Set: Generating Diverse Keyphrases as a Set},
author = {Jiacheng Ye and Tao Gui and Yichao Luo and Yige Xu and Qi Zhang},
journal= {arXiv preprint arXiv:2105.11134},
year = {2021}
}
备注
Accepted by ACL 2021