[發(fā)明專利]韻律控制的詩(shī)詞生成方法、裝置、設(shè)備及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 201911039063.7
• 申請(qǐng)日: 2019-10-29
• 公開(kāi)（公告）號(hào): CN111046648A
• 公開(kāi)（公告）日: 2020-04-21
• 發(fā)明（設(shè)計(jì)）人: 鄧?yán)杳?/a>;莊伯金;王少軍
• 申請(qǐng)（專利權(quán)）人: 平安科技（深圳）有限公司
• 主分類號(hào): G06F40/205
• 分類號(hào): G06F40/205;G06F40/253;G06F16/33
• 代理公司: 深圳市力道知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44507
• 代理人: 何姣

1.一種韻律控制的詩(shī)詞生成方法，其特征在于，包括：

獲取源字符序列，將所述源字符序列的字符信息、句子信息、位置信息的嵌入向量和所述源字符序列的韻律嵌入向量疊加，得到綜合嵌入向量；

對(duì)所述綜合嵌入向量進(jìn)行編碼處理，得到輸入文本表征向量；

獲取初始嵌入向量和韻律控制信息，將所述初始嵌入向量和所述韻律控制信息的嵌入向量疊加，得到初始化的輸出文本表征向量；

對(duì)所述輸出文本表征向量和所述輸入文本表征向量進(jìn)行解碼處理，以得到所述源字符序列對(duì)應(yīng)的生成語(yǔ)句。

2.如權(quán)利要求1所述的韻律控制的詩(shī)詞生成方法，其特征在于，所述韻律嵌入向量包括平仄嵌入向量和/或押韻嵌入向量；

所述詩(shī)詞生成方法還包括：

對(duì)所述源字符序列的平仄信息進(jìn)行嵌入處理，得到平仄嵌入向量；和/或

對(duì)所述源字符序列的押韻信息進(jìn)行嵌入處理，得到押韻嵌入向量；

所述將所述源字符序列的字符信息、句子信息、位置信息的嵌入向量和所述源字符序列的韻律嵌入向量疊加，包括：

將所述源字符序列的字符信息、句子信息、位置信息的嵌入向量以及所述平仄嵌入向量和/或押韻嵌入向量疊加。

3.如權(quán)利要求2所述的韻律控制的詩(shī)詞生成方法，其特征在于，所述獲取初始嵌入向量和韻律控制信息，將所述初始嵌入向量和所述韻律控制信息的嵌入向量疊加包括：

獲取初始化字符序列，對(duì)所述初始化字符序列進(jìn)行嵌入處理，得到所述初始化字符序列的嵌入向量；

獲取平仄控制信息和/或押韻控制信息，對(duì)所述平仄控制信息和/或押韻控制信息進(jìn)行嵌入處理，得到所述平仄控制信息和/或押韻控制信息各自的嵌入向量；

將所述初始化字符序列的嵌入向量，以及所述平仄控制信息和/或押韻控制信息各自的嵌入向量疊加。

4.如權(quán)利要求3所述的韻律控制的詩(shī)詞生成方法，其特征在于，所述獲取平仄控制信息和/或押韻控制信息，包括：

獲取用戶輸入的目標(biāo)語(yǔ)句，根據(jù)所述目標(biāo)語(yǔ)句確定平仄控制信息和/或押韻控制信息，所述目標(biāo)語(yǔ)句和用戶期望的生成語(yǔ)句具有相同的韻律規(guī)則。

5.如權(quán)利要求1所述的韻律控制的詩(shī)詞生成方法，其特征在于，所述對(duì)所述綜合嵌入向量進(jìn)行編碼處理，得到輸入文本表征向量，包括：

基于包括BERT模型的編碼器，對(duì)所述綜合嵌入向量進(jìn)行編碼處理，得到輸入文本表征向量。

6.如權(quán)利要求5所述的韻律控制的詩(shī)詞生成方法，其特征在于，所述對(duì)所述輸出文本表征向量和所述輸入文本表征向量進(jìn)行解碼處理，以得到所述源字符序列對(duì)應(yīng)的生成語(yǔ)句，包括：

基于Transformer解碼器，通過(guò)注意力機(jī)制整合所述輸入文本表征向量對(duì)所述初始化的輸出文本表征向量逐字更新，以得到所述源字符序列對(duì)應(yīng)的生成語(yǔ)句。

7.一種詩(shī)詞生成模型的訓(xùn)練方法，其特征在于，包括：

獲取訓(xùn)練樣本，所述訓(xùn)練樣本包括樣本字符序列，以及所述樣本字符序列的后一句詩(shī)詞，所述后一句詩(shī)詞和所述樣本字符序列的韻律相匹配；

將所述樣本字符序列的字符信息、句子信息、位置信息的嵌入向量和所述樣本字符序列的韻律嵌入向量疊加，得到綜合嵌入向量；

對(duì)所述綜合嵌入向量進(jìn)行編碼處理，得到輸入文本表征向量；

對(duì)所述后一句詩(shī)詞的韻律信息進(jìn)行嵌入處理，得到韻律期望向量；

獲取初始嵌入向量，將所述初始嵌入向量和所述韻律期望向量疊加，得到初始化的輸出文本表征向量；

對(duì)所述輸入文本表征向量和所述輸入文本表征向量進(jìn)行解碼處理，得到所述樣本字符序列對(duì)應(yīng)生成語(yǔ)句；

根據(jù)所述生成語(yǔ)句的韻律信息和所述后一句詩(shī)詞的韻律信息計(jì)算損失值，根據(jù)所述損失值調(diào)整所述詩(shī)詞生成模型的參數(shù)。