[發明專利]文本轉SQL模型的優化方法及系統

在審

• 申請號: 202110476255.5
• 申請日: 2021-04-29
• 公開（公告）號: CN113177123A
• 公開（公告）日: 2021-07-27
• 發明（設計）人: 俞凱;曹瑞升;陳露
• 申請（專利權）人: 思必馳科技股份有限公司
• 主分類號: G06F16/36
• 分類號: G06F16/36;G06F16/242;G06F40/151;G06F40/211;G06F40/284;G06F40/30;G06N3/04;G06N3/08
• 代理公司: 北京商專永信知識產權代理事務所(普通合伙) 11400
• 代理人: 黃謙;車江華

1.一種文本轉SQL模型的優化方法，包括：

將從文本的異構圖以及所述異構圖對應的線圖中確定的關系特征輸入至隱藏層，其中，所述關系特征至少包括：所述異構圖確定的節點特征以及所述線圖確定的邊特征，以提升所述關系特征的語義；

在所述隱藏層中，通過兩個關系圖注意力網絡分別對所述節點特征和所述邊特征迭代更新，將迭代更新后的節點特征輸入至輸出層；

在所述輸出層中確定對應的SQL。

2.根據權利要求1所述的方法，其中，所述線圖由所述異構圖確定，包括：

當所述異構圖中第一條邊的目標節點是第二條邊的源節點，且所述第一條邊與所述第二條邊不構成閉環時，在對應的線圖中，將與所述第一條邊對應的第一節點向與所述第二條邊對應的第二節點有向連接。

3.根據權利要求1所述的方法，其中，所述通過兩個關系圖注意力網絡分別對所述節點特征和所述邊特征迭代更新包括：

通過第一關系圖注意力網絡，利用所述異構圖確定的節點特征對所述線圖的邊特征迭代，確定包括局部和非局部的第一節點特征；

通過第二關系圖注意力網絡，利用所述線圖確定的邊特征對所述異構圖的節點特征迭代，確定局部節點第二節點特征；

將所述第一節點特征以及所述第二特征進行拼接，作為輸出層的輸入。

4.根據權利要求1所述的方法，其中，所述輸出層包括：文本轉SQL解碼器以及圖修剪模塊；

其中，所述圖修剪模塊用于提高所述文本轉SQL解碼器的判別能力。

5.根據權利要求1所述的方法，其中，所述關系特征在輸入層中確定；

所述輸入層包括上下文相關的特征提取器或GloVe詞向量，用于確定異構圖、線圖各節點的關系特征，其中，所述節點包括：問題節點、表節點、列節點。

6.一種文本轉SQL模型的優化系統，包括：

關系特征輸入程序模塊，用于將從文本的異構圖以及所述異構圖對應的線圖中確定的關系特征輸入至隱藏層，其中，所述關系特征至少包括：所述異構圖確定的節點特征以及所述線圖確定的邊特征，以提升所述關系特征的語義；

關系圖注意力處理程序模塊，用于在所述隱藏層中，通過兩個關系圖注意力網絡分別對所述節點特征和所述邊特征迭代更新，將迭代更新后的節點特征輸入至輸出層；

SQL確定程序模塊，用于在所述輸出層中確定對應的SQL。

7.根據權利要求6所述的系統，其中，所述線圖由所述異構圖確定，當所述異構圖中第一條邊的目標節點是第二條邊的源節點，且所述第一條邊與所述第二條邊不構成閉環時，在對應的線圖中，將與所述第一條邊對應的第一節點向與所述第二條邊對應的第二節點有向連接。

8.根據權利要求6所述的系統，其中，所述關系圖注意力處理程序模塊用于：

通過第一關系圖注意力網絡，利用所述異構圖確定的節點特征對所述線圖的邊特征迭代，確定包括局部和非局部的第一節點特征；

通過第二關系圖注意力網絡，利用所述線圖確定的邊特征對所述異構圖的節點特征迭代，確定局部節點第二節點特征；

將所述第一節點特征以及所述第二特征進行拼接，作為輸出層的輸入。

9.一種電子設備，其包括：至少一個處理器，以及與所述至少一個處理器通信連接的存儲器，其中，所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器能夠執行權利要求1-5中任一項所述方法的步驟。

10.一種存儲介質，其上存儲有計算機程序，其特征在于，該程序被處理器執行時實現權利要求1-5中任一項所述方法的步驟。