[發明專利]文本風格遷移方法、裝置、設備及存儲介質

在審

• 申請號: 202210363906.4
• 申請日: 2022-04-08
• 公開（公告）號: CN114818695A
• 公開（公告）日: 2022-07-29
• 發明（設計）人: 劉弘一;李金龍;楊一梟;蒲珂宇
• 申請（專利權）人: 招商銀行股份有限公司
• 主分類號: G06F40/289
• 分類號: G06F40/289;G06F40/253;G06N3/04;G06N3/08
• 代理公司: 深圳市世紀恒程知識產權代理事務所 44287
• 代理人: 高川

1.一種文本風格遷移方法，其特征在于，所述文本風格遷移方法包括：

獲取待遷移文本信息；

將所述待遷移文本信息輸入至已訓練的文本風格遷移模型，得到目標風格遷移文本，其中，所述文本風格遷移模型包括編碼器和解碼器，所述文本風格遷移模型為基于待訓練文本信息以及預設風格要求的特征向量進行訓練獲得。

2.如權利要求1所述的文本風格遷移方法，其特征在于，所述將所述待遷移文本信息輸入至已訓練的文本風格遷移模型，得到目標風格遷移文本的步驟包括：

通過所述編碼器對所述待遷移文本進行編碼處理，獲得編碼向量；

將所述編碼向量輸入至所述解碼器做解碼處理，獲得目標特征向量；

基于所述目標特征向量，生成所述目標風格遷移文本。

3.如權利要求1所述的文本風格遷移方法，其特征在于，在所述將所述待遷移文本信息輸入至已訓練的文本風格遷移模型，得到目標風格遷移文本的步驟之前，所述文本風格遷移方法還包括：

獲取待訓練文本信息，并對所述待訓練文本信息進行分詞處理；

基于分詞后的各個分詞信息和預設風格要求的特征向量，對待訓練遷移模型的編碼器和解碼器進行迭代訓練，獲得所述文本風格遷移模型。

4.如權利要求3所述的文本風格遷移方法，其特征在于，所述基于分詞后的各個分詞信息和預設風格要求的特征向量，對待訓練遷移模型的編碼器和解碼器進行迭代訓練，獲得所述文本風格遷移模型的步驟包括：

基于分詞后的各個分詞信息，通過所述編碼器進行編碼處理，獲得所述待訓練文本信息對應的編碼向量，其中，所述編碼向量包括均值向量和方差向量；

基于所述均值向量和方差向量，選取滿足預設概率分布的采樣向量；

基于所述采樣向量和所述預設風格要求的特征向量，確定所述解碼器的目標輸入向量；

將所述目標輸入向量輸入所述解碼器進行解碼，獲得解碼結果；

基于所述解碼結果和所述待訓練文本信息的真實標簽之間的差異，對所述編碼器和所述解碼器進行優化訓練，獲得所述文本風格遷移模型。

5.如權利要求4所述的文本風格遷移方法，其特征在于，所述基于所述解碼結果和所述待訓練文本信息的真實標簽之間的差異，對所述編碼器和所述解碼器進行優化訓練，獲得所述文本風格遷移模型的步驟包括：

基于預設損失函數，計算所述解碼結果與待訓練文本的真實標簽之間的模型損失；

計算所述模型損失的梯度，并基于梯度迭代優化所述編碼器和所述解碼器的模型參數，得到訓練后的文本風格遷移模型。

6.如權利要求1所述的文本風格遷移方法，其特征在于，所述基于所述采樣向量和所述預設風格要求的特征向量，確定所述解碼器的目標輸入向量的步驟包括：

將所述采樣向量和所述特征向量進行加權處理，獲得加權特征向量，并將所述加權特征向量作為所述解碼器的目標輸入向量。

7.如權利要求1所述的文本風格遷移方法，其特征在于，所述獲取待訓練文本信息的步驟包括：

獲取歷史樣本信息；

對所述歷史樣本信息進行預處理，并通過預設數據增強方式構建所述待訓練文本信息。

8.一種文本風格遷移裝置，其特征在于，所述文本風格遷移裝置包括：

獲取模塊，用于獲取待遷移文本信息；

風格遷移模塊，用于將所述待遷移文本信息輸入至已訓練的文本風格遷移模型，得到目標風格遷移文本，其中，所述文本風格遷移模型包括編碼器和解碼器，所述文本風格遷移模型為基于待訓練文本信息以及預設風格要求的特征向量進行訓練獲得。

9.一種文本風格遷移設備，其特征在于，所述文本風格遷移設備包括：存儲器、處理器以及存儲在存儲器上的文本風格遷移程序，

所述文本風格遷移程序被所述處理器執行實現如權利要求1至7中任一項所述文本風格遷移方法的步驟。

10.一種存儲介質，所述存儲介質為計算機可讀存儲介質，其特征在于，所述計算機可讀存儲介質上存儲有文本風格遷移程序，所述文本風格遷移程序被處理器執行實現如權利要求1至7中任一項所述文本風格遷移方法的步驟。