一种模型训练方法、文本生成方法、装置、介质及设备

摘要

本发明公开了一种模型训练方法、文本生成方法、装置、介质及设备。其中，模型训练方法包括：获取样本文本，将所述样本文本输入至待训练的文本生成模型，获取所述编码器输出的所述样本文本中各词之间的相关性系数，以及所述解码器输出的自注意力参数和预测结果；基于所述样本文本中各词之间的相关性系数确定各词的重要性指数；基于所述各词的重要性指数、自注意力参数、所述预测结果和所述样本文本对应的标准文本生成损失函数；基于所述损失函数对所述待训练的文本生成模型进行模型参数调节，直到训练过程满足训练停止条件，得到目标文本生成模型。提高文本生成模型从输入文本中提取重要词的能力，进一步提高输出文本的质量。