[發明專利]一種搜索重排模型的確定方法及裝置

在審

• 申請號: 202210367936.2
• 申請日: 2022-04-08
• 公開（公告）號: CN114722086A
• 公開（公告）日: 2022-07-08
• 發明（設計）人: 張志鋼
• 申請（專利權）人: 北京奇藝世紀科技有限公司
• 主分類號: G06F16/2457
• 分類號: G06F16/2457;G06N3/04;G06N3/08;G06K9/62
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 王寶筠

本申請公開了一種搜索重排模型的確定方法及裝置，通過初始打分排序模型對目標搜索詞的多個搜索結果條目進行初始排序打分獲得各自的初始排序分數，通過打分評估模型根據期望排序對初始排序分數進行評估，并確定各自的獎勵分數；進一步根據初始排序分數和獎勵分數確定多個搜索結果條目的損失函數，并根據該損失函數對初始打分模型進行排序模型訓練，獲得用于對多個搜索結果條目進行搜索重排的目標打分排序模型。由于獎勵分數是基于期望排序確定的，因此利用根據初始排序分數和獎勵分數確定的損失函數對初始打分排序模型進行排序模型訓練，使得初始打分排序模型能夠向輸出期望排序的方向進行優化，獲得目標打分排序模型，提高排序結果的準確度。

技術領域

本申請涉及數據處理領域，特別是涉及一種搜索重排模型的確定方法及裝置。

背景技術

隨著信息技術的快速發展，在線搜索成為人們獲取信息的重要途徑之一。具體的，用戶在搜索系統中輸入搜索詞后，系統將召回大量的搜索結果條目，并對其進行初排、精排，最終將部分展示給用戶。

在搜索系統中，搜索排序的質量好壞很大程度地影響了頁面質量，具體是指，向用戶展示搜索結果的頁面中所顯示的搜索結果與用戶的搜索期望之間的匹配程度。當頁面質量較差時，將帶給用戶大量冗余信息的呈現，降低用戶獲取信息的效率；當頁面質量較好時，將為用戶展示一個更加符合用戶預期內容的頁面，提高用戶獲取信息的效率。

可見，如何提高針對待排序的搜索結果條目的排序結果的準確度，對于提高用戶的搜索體驗具有重要的意義。

發明內容

為了解決上述技術問題，本申請提供了一種搜索重排模型的確定方法及裝置，提高了對目標搜索詞的搜索結果排序的準確度。

本申請實施例公開了如下技術方案：

一方面，本申請實施例提供了一種搜索重排模型的確定方法，所述方法包括：

獲取目標搜索詞對應的訓練樣本集；所述訓練樣本集包括所述目標搜索詞和所述目標搜索詞的多個搜索結果條目；

將所述訓練樣本集輸入初始打分排序模型，通過所述初始打分排序模型確定所述多個搜索結果條目各自的初始排序分數；

將包括目標排序標簽的所述訓練樣本集和所述初始排序分數輸入打分評估模型，通過所述打分評估模型基于所述目標排序標簽和所述初始排序分數，確定所述初始排序分數各自的獎勵分數；其中，所述目標排序標簽用于標識所述訓練樣本集中的所述多個搜索結果條目的期望排序；

基于所述初始排序分數和所述獎勵分數，確定所述多個搜索結果條目的損失函數；

根據所述損失函數對所述初始打分排序模型進行排序模型訓練，得到用于對所述多個搜索結果條目進行搜索重排的目標打分排序模型。

另一方面，本申請實施例提供了一種搜索重排模型的確定裝置，所述裝置包括獲取單元、確定單元和訓練單元：

所述獲取單元，用于獲取目標搜索詞對應的訓練樣本集；所述訓練樣本集包括所述目標搜索詞和所述目標搜索詞的多個搜索結果條目；

所述確定單元，用于將所述訓練樣本集輸入初始打分排序模型，通過所述初始打分排序模型確定所述多個搜索結果條目各自的初始排序分數；

所述確定單元，還用于將包括目標排序標簽的所述訓練樣本集和所述初始排序分數輸入打分評估模型，通過所述打分評估模型基于所述目標排序標簽和所述初始排序分數，確定所述初始排序分數各自的獎勵分數；其中，所述目標排序標簽用于標識所述訓練樣本集中的所述多個搜索結果條目的期望排序；

所述確定單元，還用于基于所述初始排序分數和所述獎勵分數，確定所述多個搜索結果條目的損失函數；