分類模型的訓練方法及裝置、電子設備、存儲介質(zhì)

基本信息

申請?zhí)?/td> CN202210021820.3 申請日 -
公開(公告)號 CN114358197A 公開(公告)日 2022-04-15
申請公布號 CN114358197A 申請公布日 2022-04-15
分類號 G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06V10/764(2022.01)I;G06V10/774(2022.01)I 分類 計算;推算;計數(shù);
發(fā)明人 單齊齊;周濤;史治國 申請(專利權)人 上海億保健康管理有限公司
代理機構(gòu) 北京超凡志成知識產(chǎn)權代理事務所(普通合伙) 代理人 李丙林
地址 201800上海市嘉定區(qū)滬宜公路1185號1幢17層1718室、1720室
法律狀態(tài) -

摘要

摘要 本申請?zhí)峁┮环N分類模型的訓練方法及裝置、電子設備、計算機可讀存儲介質(zhì),方法包括:構(gòu)建與已訓練的標準分類模型對應的輕量模型;將樣本數(shù)據(jù)集中的樣本發(fā)票圖像分別輸入標準分類模型和輕量模型,獲得標準分類模型在預測過程的第一損失參數(shù)和輕量模型在預測過程中的第二損失參數(shù);對第一損失參數(shù)和第二損失參數(shù)加權求和,得到目標損失參數(shù);依據(jù)目標損失參數(shù)對輕量模型的網(wǎng)絡參數(shù)進行調(diào)整,并迭代至輕量模型收斂,得到輕量分類模型。本申請方案,訓練過程中參考了標準分類模型的第一憑據(jù)參數(shù),可將標準分類模型從樣本發(fā)票圖像中學習到的知識傳遞至輕量模型,從而訓練出預測效果更佳、模型復雜度較低的輕量分類模型。