知識蒸餾方法及裝置、存儲介質(zhì)、終端

基本信息

申請?zhí)?/td> CN202110443168.X 申請日 -
公開（公告）號 CN113159073A 公開（公告）日 2021-07-23
申請公布號 CN113159073A 申請公布日 2021-07-23
分類號 G06K9/46（2006.01）I;G06K9/62（2006.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 朱政;李路軍;黃冠 申請（專利權(quán)）人 上海芯翌智能科技有限公司
代理機構(gòu) 北京集佳知識產(chǎn)權(quán)代理有限公司 代理人 張振軍
地址 200434上海市虹口區(qū)海寧路137號7層（集中登記地）
法律狀態(tài) -

摘要

摘要 一種知識蒸餾方法及裝置、存儲介質(zhì)、終端，所述方法包括：獲取樣本圖像，將樣本圖像分別輸入至教師網(wǎng)絡(luò)和學(xué)生網(wǎng)絡(luò)，教師網(wǎng)絡(luò)包括第一特征提取模塊，第一特征提取模塊包括一個或多個級聯(lián)的第一特征單元，學(xué)生網(wǎng)絡(luò)包括第二特征提取模塊，第二特征提取模塊包括一個或多個級聯(lián)的第二特征單元，樣本圖像具有標(biāo)簽；將每一第二特征單元計算得到的特征向量聚合至與該第二特征單元對應(yīng)的第一特征單元計算得到的特征向量；根據(jù)第一特征提取模塊輸出的特征向量計算第一預(yù)測結(jié)果，根據(jù)第二特征提取模塊輸出的特征向量計算第二預(yù)測結(jié)果；根據(jù)第一預(yù)測結(jié)果、第二預(yù)測結(jié)果和標(biāo)簽計算蒸餾損失；根據(jù)蒸餾損失更新學(xué)生網(wǎng)絡(luò)。