[发明专利]文章标题生成方法、装置、存储介质及电子设备

说明书

本发明提供一种文章标题生成方法、装置、存储介质及电子设备，该生成方法包括：根据文本摘要算法从目标文章中提取目标摘要；基于预训练的标题生成模型和目标摘要，生成第一候选文章标题；基于标题生成模型和目标文章，生成第二候选文章标题；计算第一候选文章标题和第二候选文章标题的标题匹配度，根据标题匹配度从第一候选文章标题中确定目标文章标题。本发明将标题生成模型和目标摘要生成的第一候选文章标题和标题生成模型和目标文章生成的第二候选文章标题进行匹配计算，根据匹配计算结果从第一候选文章标题中得到贴合目标文章内容的标题作为目标文章标题，从而提高文章标题生成的精确度。

技术领域

本发明涉及文本处理技术领域，具体涉及一种文章标题生成方法、装置、存储介质及电子设备。

背景技术

标题和摘要对于文章的创作是非常重要的，但构思一个具有吸引力且贴近文章内容的标题和从文章中抽取或生成符合文章主旨的摘要并非易事。目标摘要生成需要将初始的长文章进行压缩、归纳和总结，从而形成短小精悍且具有概括性含义的短篇文本。文章标题生成需要在摘要生成的基础上，进一步精炼，并搭配适当的风格。

传统的人工阅读并总结的方式效率较低，且受作者主观性影响较大，可能会由于作者的主观因素导致误判，因此生成的文本标题和摘要常常不够准确。在自然语言处理领域内，目前主流的标题自动生成和摘要自动生成方法可分为抽取式和生成式。抽取式是从原文中抽取关键的句子组成摘要，该方法存在丢失信息的风险；生成式则是在理解原文的基础上进行语言的重新组织表达，该方法难以保证生成摘要或标题的通顺度且如果原文章长度过长，生成的效果也会难以保障。

发明内容

本发明的目的在于提出一种文章标题生成方法、装置、存储介质及电子设备，以提高文章标题生成的精确度。

本发明提供一种文章标题生成方法，包括以下步骤：

根据文本摘要算法从目标文章中提取目标摘要；

基于预训练的标题生成模型和所述目标摘要，生成第一候选文章标题；

基于所述标题生成模型和所述目标文章，生成第二候选文章标题；

计算所述第一候选文章标题和所述第二候选文章标题的标题匹配度，根据所述标题匹配度从所述第一候选文章标题中确定目标文章标题；

所述标题生成模型的训练方法包括以下步骤：

获取训练用的原始文本集，所述原始文本集包括原始文章和原始标题；

对所述原始文本集进行预处理，得到格式规范的输入数据，所述预处理为将所述原始文本集的格式统一化处理；

将预处理后的输入数据向改良后的GPT-2模型输入并进行训练，所述改良后的GPT-2模型为在GPT-2模型的下游增加FC层，得到预训练的标题生成模型。

根据本发明提出的文章标题生成方法，具有以下有益效果：本发明采用标题生成模型和目标摘要生成第一候选文章标题，采用标题生成模型和目标文章生成第二候选文章标题，将目标摘要生成的第一候选文章标题和目标文章生成的第二候选文章标题进行匹配计算，根据匹配计算结果从第一候选文章标题中得到贴合目标文章内容的标题作为目标文章标题，从而提高文章标题生成的精确度。

另外，根据本发明提供的文章标题生成方法，还可以具有如下附加的技术特征：

进一步地，计算所述第一候选文章标题和所述第二候选文章标题的标题匹配度，根据所述标题匹配度从所述第一候选文章标题中确定目标文章标题的步骤包括：

计算所述第一候选文章标题和所述第二候选文章标题的标题匹配度，将与所述第二候选文章标题匹配度最高的所述第一候选文章标题作为目标文章标题。