基于对比学习的摘要生成模型的训练方法和摘要生成方法

摘要

本发明公开了基于对比学习的摘要生成模型的训练方法和摘要生成方法，属于自然语言处理技术领域。模型训练方法包括：获取训练样本集合；利用训练样本集合训练摘要生成模型，得到初步优化的摘要生成模型；利用初步优化的摘要生成模型和训练样本中的文档得到生成摘要，并利用生成摘要和参考摘要构建正负例数据；利用正负例数据和对应的文档，采用对比学习方法对初步优化的摘要生成模型进行训练，得到训练好的摘要生成模型。通过从模型生成摘要和参考摘要中构建正负样例，并且综合考虑忠实性和概括性分数，使得构建的正负样例区分难度大，正样例同时具备忠实性和概括性。解决了现有技术中生成的摘要存在事实性错误、概括性不强的问题。