[發(fā)明專利]模型訓(xùn)練方法、裝置、電子設(shè)備和存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202110208857.2
• 申請(qǐng)日: 2021-02-24
• 公開(kāi)（公告）號(hào): CN112883180A
• 公開(kāi)（公告）日: 2021-06-01
• 發(fā)明（設(shè)計(jì)）人: 毛晨思;任宇翔;吳邊;孟海忠
• 申請(qǐng)（專利權(quán)）人: 掛號(hào)網(wǎng)（杭州）科技有限公司
• 主分類號(hào): G06F16/332
• 分類號(hào): G06F16/332;G06K9/62;G16H10/60;G16H50/20;G16H50/70;G16H80/00
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 孟金喆

本發(fā)明實(shí)施例公開(kāi)了一種模型訓(xùn)練方法、裝置、電子設(shè)備和存儲(chǔ)介質(zhì)。其中，該方法包括：獲取用于訓(xùn)練第一待訓(xùn)練模型的訓(xùn)練樣本集，訓(xùn)練樣本集中包括基于對(duì)話問(wèn)診生成的歷史對(duì)話數(shù)據(jù)子集和/或用于描述病癥的文本數(shù)據(jù)子集；根據(jù)預(yù)設(shè)規(guī)則，提取出訓(xùn)練樣本集中每個(gè)訓(xùn)練樣本數(shù)據(jù)的訓(xùn)練標(biāo)簽；其中，預(yù)設(shè)規(guī)則包括預(yù)設(shè)掩碼集合中的掩碼文字和/或根據(jù)文本數(shù)據(jù)確定的句子位置信息；基于訓(xùn)練樣本數(shù)據(jù)和與訓(xùn)練樣本數(shù)據(jù)對(duì)應(yīng)的訓(xùn)練標(biāo)簽，對(duì)第一待訓(xùn)練模型進(jìn)行訓(xùn)練，得到第一目標(biāo)模型；其中，第一目標(biāo)模型用于對(duì)輸入的目標(biāo)數(shù)據(jù)進(jìn)行處理，得到與目標(biāo)數(shù)據(jù)相對(duì)應(yīng)的數(shù)據(jù)處理結(jié)果。以實(shí)現(xiàn)在不需進(jìn)行人工標(biāo)注的情況下，對(duì)模型進(jìn)行訓(xùn)練的效果，節(jié)省了人力成本。

技術(shù)領(lǐng)域

本發(fā)明實(shí)施例涉及計(jì)算機(jī)技術(shù)，尤其涉及一種模型訓(xùn)練方法、裝置、電子設(shè)備和存儲(chǔ)介質(zhì)。

背景技術(shù)

隨著信息技術(shù)的深入發(fā)展，人工智能技術(shù)與人們的生活、教育和醫(yī)療等產(chǎn)生了緊密的聯(lián)系。

在現(xiàn)階段的模型訓(xùn)練中，在對(duì)模型進(jìn)行訓(xùn)練時(shí)，往往需要首先獲取原始數(shù)據(jù)，然后根據(jù)指定的任務(wù)對(duì)原始數(shù)據(jù)進(jìn)行標(biāo)注，利用標(biāo)注的數(shù)據(jù)作為標(biāo)簽，將原始數(shù)據(jù)和標(biāo)注的數(shù)據(jù)輸入到待訓(xùn)練的模型中對(duì)模型進(jìn)行訓(xùn)練。

上述模型訓(xùn)練方式，需要人工標(biāo)注模型的訓(xùn)練標(biāo)簽，浪費(fèi)人力，成本較高。

發(fā)明內(nèi)容

本發(fā)明實(shí)施例提供一種模型訓(xùn)練方法、裝置、電子設(shè)備和存儲(chǔ)介質(zhì)，以實(shí)現(xiàn)在不需進(jìn)行人工標(biāo)注的情況下，對(duì)模型進(jìn)行訓(xùn)練的效果，節(jié)省了人力成本。

第一方面，本發(fā)明實(shí)施例提供了一種模型訓(xùn)練方法，該方法包括：

獲取用于訓(xùn)練第一待訓(xùn)練模型的訓(xùn)練樣本集，所述訓(xùn)練樣本集中包括基于對(duì)話問(wèn)診生成的歷史對(duì)話數(shù)據(jù)子集和/或用于描述病癥的文本數(shù)據(jù)子集；

根據(jù)預(yù)設(shè)規(guī)則，提取出所述訓(xùn)練樣本集中每個(gè)訓(xùn)練樣本數(shù)據(jù)的訓(xùn)練標(biāo)簽；其中，所述預(yù)設(shè)規(guī)則包括預(yù)設(shè)掩碼集合中的掩碼文字和/或根據(jù)文本數(shù)據(jù)確定的句子位置信息；

基于所述訓(xùn)練樣本數(shù)據(jù)和與所述訓(xùn)練樣本數(shù)據(jù)對(duì)應(yīng)的訓(xùn)練標(biāo)簽，對(duì)所述第一待訓(xùn)練模型進(jìn)行訓(xùn)練，得到第一目標(biāo)模型；

其中，所述第一目標(biāo)模型用于對(duì)輸入的目標(biāo)數(shù)據(jù)進(jìn)行處理，得到與所述目標(biāo)數(shù)據(jù)相對(duì)應(yīng)的數(shù)據(jù)處理結(jié)果。

第二方面，本發(fā)明實(shí)施例還提供了一種模型訓(xùn)練裝置，該裝置包括：

訓(xùn)練樣本集獲取模塊，用于獲取用于訓(xùn)練第一待訓(xùn)練模型的訓(xùn)練樣本集，所述訓(xùn)練樣本集中包括基于對(duì)話問(wèn)診生成的歷史對(duì)話數(shù)據(jù)子集和/或用于描述病癥的文本數(shù)據(jù)子集；

訓(xùn)練標(biāo)簽確定模塊，用于根據(jù)預(yù)設(shè)規(guī)則，提取出所述訓(xùn)練樣本集中每個(gè)訓(xùn)練樣本數(shù)據(jù)的訓(xùn)練標(biāo)簽；其中，所述預(yù)設(shè)規(guī)則包括預(yù)設(shè)掩碼集合中的掩碼文字和/或根據(jù)文本數(shù)據(jù)確定的句子位置信息；

第一目標(biāo)模型確定模塊，用于基于所述訓(xùn)練樣本數(shù)據(jù)和與所述訓(xùn)練樣本數(shù)據(jù)對(duì)應(yīng)的訓(xùn)練標(biāo)簽，對(duì)所述第一待訓(xùn)練模型進(jìn)行訓(xùn)練，得到第一目標(biāo)模型；其中，所述第一目標(biāo)模型用于對(duì)輸入的目標(biāo)數(shù)據(jù)進(jìn)行處理，得到與所述目標(biāo)數(shù)據(jù)相對(duì)應(yīng)的數(shù)據(jù)處理結(jié)果。

第三方面，本發(fā)明實(shí)施例還提供了一種電子設(shè)備，該電子設(shè)備包括：

一個(gè)或多個(gè)處理器；

存儲(chǔ)裝置，用于存儲(chǔ)一個(gè)或多個(gè)程序；

當(dāng)所述一個(gè)或多個(gè)程序被所述一個(gè)或多個(gè)處理器執(zhí)行，使得所述一個(gè)或多個(gè)處理器實(shí)現(xiàn)本發(fā)明實(shí)施例中任一所述的模型訓(xùn)練方法。

第四方面，本發(fā)明實(shí)施例還提供了一種包含計(jì)算機(jī)可執(zhí)行指令的存儲(chǔ)介質(zhì)，所述計(jì)算機(jī)可執(zhí)行指令在由計(jì)算機(jī)處理器執(zhí)行時(shí)用于執(zhí)行本發(fā)明實(shí)施例中任一所述的模型訓(xùn)練方法。