[發(fā)明專利]訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法、裝置、設(shè)備及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202010123970.6
• 申請(qǐng)日: 2020-02-28
• 公開（公告）號(hào): CN111325353A
• 公開（公告）日: 2020-06-23
• 發(fā)明（設(shè)計(jì)）人: 卓本剛;黃啟軍;唐興興;林冰垠
• 申請(qǐng)（專利權(quán)）人: 深圳前海微眾銀行股份有限公司
• 主分類號(hào): G06N20/00
• 分類號(hào): G06N20/00
• 代理公司: 深圳市世紀(jì)恒程知識(shí)產(chǎn)權(quán)代理事務(wù)所 44287
• 代理人: 王韜

1.一種訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法包括以下步驟：

獲取訓(xùn)練機(jī)器學(xué)習(xí)模型的各訓(xùn)練數(shù)據(jù)集；

計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的沙普利可加性模型解釋方法SHAP目標(biāo)值；

根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度。

2.如權(quán)利要求1所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值的步驟包括：

計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的SHAP值，計(jì)算各個(gè)特征對(duì)應(yīng)的SHAP值的絕對(duì)值，得到各個(gè)特征對(duì)應(yīng)的SHAP絕對(duì)值；

將所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征確定為目標(biāo)特征，根據(jù)所述目標(biāo)特征在不同訓(xùn)練數(shù)據(jù)集中的SHAP絕對(duì)值得到所述目標(biāo)特征對(duì)應(yīng)的SHAP目標(biāo)值。

3.如權(quán)利要求2所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述根據(jù)所述目標(biāo)特征在不同訓(xùn)練數(shù)據(jù)集中的SHAP絕對(duì)值得到所述目標(biāo)特征對(duì)應(yīng)的SHAP目標(biāo)值的步驟包括：

確定所述目標(biāo)特征在不同訓(xùn)練數(shù)據(jù)集中的SHAP絕對(duì)值，計(jì)算所述目標(biāo)特征在不同訓(xùn)練數(shù)據(jù)集中SHAP絕對(duì)值對(duì)應(yīng)的SHAP平均值；

將所述SHAP平均值確定為所述目標(biāo)特征對(duì)應(yīng)的SHAP目標(biāo)值。

4.如權(quán)利要求2所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的SHAP值的步驟包括：

計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的邊際收益期望；

根據(jù)所述邊際收益期望計(jì)算各所述邊際收益期望對(duì)應(yīng)特征的SHAP值，以得到所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的SHAP值。

5.如權(quán)利要求1所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度的步驟包括：

確定所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值，以及確定各個(gè)特征所在訓(xùn)練數(shù)據(jù)集的數(shù)據(jù)集個(gè)數(shù)；

根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的所述SHAP目標(biāo)值和所述數(shù)據(jù)集個(gè)數(shù)，計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度。

6.如權(quán)利要求5所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的所述SHAP目標(biāo)值和所述數(shù)據(jù)集個(gè)數(shù)，計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度的步驟包括：

計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)的所述SHAP目標(biāo)值與所述數(shù)據(jù)集個(gè)數(shù)之間的商值；

將所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征對(duì)應(yīng)商值相加，得到所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度。

7.如權(quán)利要求1至6任一項(xiàng)所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法，其特征在于，所述根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度的步驟之后，還包括：

根據(jù)各訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度選擇訓(xùn)練所述機(jī)器學(xué)習(xí)模型的目標(biāo)訓(xùn)練數(shù)據(jù)集；

將所述目標(biāo)訓(xùn)練數(shù)據(jù)集輸入所述機(jī)器學(xué)習(xí)模型中，以訓(xùn)練所述機(jī)器學(xué)習(xí)模型。

8.一種訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算裝置，其特征在于，所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算裝置包括：

獲取模塊，用于獲取訓(xùn)練機(jī)器學(xué)習(xí)模型的各訓(xùn)練數(shù)據(jù)集；

計(jì)算模塊，用于計(jì)算所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的沙普利可加性模型解釋方法SHAP目標(biāo)值；根據(jù)所述訓(xùn)練數(shù)據(jù)集中各個(gè)特征的SHAP目標(biāo)值計(jì)算所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度。

9.一種訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算設(shè)備，其特征在于，所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算設(shè)備包括存儲(chǔ)器、處理器和存儲(chǔ)在所述存儲(chǔ)器上并可在所述處理器上運(yùn)行的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算程序，所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算程序被所述處理器執(zhí)行時(shí)實(shí)現(xiàn)如權(quán)利要求1至7任一項(xiàng)中所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法的步驟。

10.一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其特征在于，所述計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)上存儲(chǔ)有訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算程序，所述訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)如權(quán)利要求1至7任一項(xiàng)所述的訓(xùn)練數(shù)據(jù)集的貢獻(xiàn)度計(jì)算方法的步驟。