[發明專利]一種對話生成方法、裝置及計算機設備

在審

• 申請號: 202010127591.4
• 申請日: 2020-02-28
• 公開（公告）號: CN111368051A
• 公開（公告）日: 2020-07-03
• 發明（設計）人: 于鳳英;王健宗
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06F16/332
• 分類號: G06F16/332;G06F16/33
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 任敏

本申請適用數據處理技術領域，提供了一種對話生成方法、裝置及計算機設備，所述方法包括：S1、獲取對話生成模型和輸入語句；S2、將輸入語句輸入對話生成模型，得到答復語句并記錄在預設存儲空間；S3、判斷預設存儲空間中記錄的答復語句是否符合預設中止條件，若是，則執行步驟S5，若否，則執行步驟S4；S4、根據答復語句計算獎勵值，根據獎勵值以及強化學習的損失函數計算對話生成模型的損失值，根據損失值對對話生成模型進行更新，并將答復語句作為新的輸入語句，返回步驟S2；S5、將預設存儲空間中記錄的答復語句按照記錄時間的先后順序進行排列，得到會話文檔。本申請可以解決現有技術中難以合理得到用戶會話信息的問題。

技術領域

本申請屬于數據處理技術領域，尤其涉及一種對話生成方法、裝置及計算機設備。

背景技術

隨著大數據產業的發展，越來越多的企業需要收集用戶會話信息，將這些用戶會話信息應用于各種下游業務中，例如客服機器人等業務。

但是，由于政策的嚴格管控，不允許各個企業隨意交易用戶會話信息，導致有的企業采集用戶會話信息的渠道少，無法獲取到大量的用戶會話信息以支撐各種下游業務中，例如，如果只有少量的用戶會話信息作為訓練樣本，會導致客服機器人的回答準確性低。

因此，如何合理地得到更多的用戶會話信息成為了本領域技術人員亟需解決的技術問題。

發明內容

有鑒于此，本申請實施例提供了一種對話生成方法、裝置及計算機設備，以解決現有技術中難以合理得到用戶會話信息的問題。

本申請實施例的第一方面提供了一種對話生成方法，包括：

S1、獲取對話生成模型和輸入語句，所述對話生成模型為根據輸入的輸入語句生成相應的答復語句的神經網絡模型；

S2、將所述輸入語句輸入所述對話生成模型，得到答復語句并記錄在預設存儲空間；

S3、判斷所述預設存儲空間中記錄的答復語句是否符合預設中止條件，若是，則執行步驟S5，若否，則執行步驟S4；

S4、根據所述答復語句計算獎勵值，根據所述獎勵值以及強化學習的損失函數計算所述對話生成模型的損失值，根據所述損失值對所述對話生成模型進行更新，并將所述答復語句作為新的輸入語句，返回步驟S2；

S5、將所述預設存儲空間中記錄的答復語句按照記錄時間的先后順序進行排列，得到會話文檔。

本申請實施例的第二方面提供了一種對話生成裝置，包括：

初始獲取模塊，用于獲取對話生成模型和輸入語句，所述對話生成模型為根據輸入的輸入語句生成相應的答復語句的神經網絡模型；

語句生成模塊，用于將所述輸入語句輸入所述對話生成模型，得到答復語句并記錄在預設存儲空間；

條件判斷模塊，用于判斷所述預設存儲空間中記錄的答復語句是否符合預設中止條件，若是，則觸發語句輸出模塊，若否，則觸發更新返回模塊；

更新返回模塊，用于根據所述答復語句計算獎勵值，根據所述獎勵值以及強化學習的損失函數計算所述對話生成模型的損失值，根據所述損失值對所述對話生成模型進行更新，并將所述答復語句作為新的輸入語句輸入更新后的對話生成模型，觸發語句生成模塊；

語句輸出模塊，用于將所述預設存儲空間中記錄的答復語句按照記錄時間的先后順序進行排列，得到會話文檔。

本申請實施例的第三方面提供了一種計算機設備，包括存儲器、處理器以及存儲在所述存儲器中并可在所述處理器上運行的計算機程序，所述處理器執行所述計算機程序時實現如上述方法的步驟。