[发明专利]韵律控制的诗词生成方法、装置、设备及存储介质

说明书

本申请涉及自然语言处理，具体提供一种韵律控制的诗词生成方法、装置、设备及存储介质，该方法包括：获取源字符序列，将所述源字符序列的字符信息、句子信息、位置信息的嵌入向量和所述源字符序列的韵律嵌入向量叠加，得到综合嵌入向量；对所述综合嵌入向量进行编码处理，得到输入文本表征向量；获取初始嵌入向量和韵律控制信息，将所述初始嵌入向量和所述韵律控制信息的嵌入向量叠加，得到初始化的输出文本表征向量；对所述输出文本表征向量和所述输入文本表征向量进行解码处理，以得到所述源字符序列对应的生成语句。该方法提高了生成诗句的平仄押韵准确率，而不牺牲诗句的生成质量。

技术领域

本申请涉及自然语言处理技术领域，尤其涉及一种韵律控制的诗词生成方法、装置、设备及存储介质。

背景技术

古人写诗词讲究平仄押韵，是利用汉语声调的平衡交替来造成语言中抑扬顿挫的美，读起来琅琅上口，易于记忆。目前，国内外对于诗歌生成的研究经历了由基于模板的生成方法到基于统计机器翻译的方法再到基于神经网络等方法的演变过程。

随着神经网络的兴起，基于编码器-解码器的框架也逐步应用到诗歌生成当中，该框架利用神经网络强大的拟合能力，能够生成具有一定流畅度的诗句，但在韵律的表现方面与传统的诗歌韵律规则有较大偏差。

发明内容

本申请的主要目的在于提供一种韵律控制的诗词生成方法、装置、设备及存储介质，旨在提高生成诗句的平仄押韵准确性。

第一方面，本申请提供一种韵律控制的诗词生成方法，所述韵律控制的诗词生成方法包括以下步骤：

获取源字符序列，将所述源字符序列的字符信息、句子信息、位置信息的嵌入向量和所述源字符序列的韵律嵌入向量叠加，得到综合嵌入向量；

对所述综合嵌入向量进行编码处理，得到输入文本表征向量；

获取初始嵌入向量和韵律控制信息，将所述初始嵌入向量和所述韵律控制信息的嵌入向量叠加，得到初始化的输出文本表征向量；

对所述输出文本表征向量和所述输入文本表征向量进行解码处理，以得到所述源字符序列对应的生成语句。

第二方面，本申请还提供一种诗词生成模型的训练方法，所述诗词生成模型的训练方法包括：

获取训练样本，所述训练样本包括样本字符序列，以及所述样本字符序列的后一句诗词，所述后一句诗词和所述样本字符序列的韵律相匹配；

将所述样本字符序列的字符信息、句子信息、位置信息的嵌入向量和所述样本字符序列的韵律嵌入向量叠加，得到综合嵌入向量；

对所述综合嵌入向量进行编码处理，得到输入文本表征向量；

对所述后一句诗词的韵律信息进行嵌入处理，得到韵律期望向量；

获取初始嵌入向量，将所述初始嵌入向量和所述韵律期望向量叠加，得到初始化的输出文本表征向量；

对所述输入文本表征向量和所述输入文本表征向量进行解码处理，得到所述样本字符序列对应生成语句；

根据所述生成语句的韵律信息和所述后一句诗词的韵律信息计算损失值，根据所述损失值调整所述诗词生成模型的参数。

第三方面，本申请还提供一种韵律控制的诗词生成装置，所述韵律生成装置包括：

向量叠加模块，用于获取源字符序列，将所述源字符序列的字符信息、句子信息、位置信息的嵌入向量和所述源字符序列的韵律嵌入向量叠加，得到综合嵌入向量。

编码模块，用于对所述综合嵌入向量进行编码处理，得到输入文本表征向量。