[發(fā)明專利]一種圖像識別模型的訓(xùn)練方法、圖像識別的方法及裝置

有效

• 申請?zhí)枺?/td>: 202010182180.5
• 申請日: 2020-03-16
• 公開（公告）號: CN111401445B
• 公開（公告）日: 2023-03-10
• 發(fā)明（設(shè)計）人: 李一鳴;吳保元;張勇;樊艷波;李志鋒;劉威;馮巖;江勇;夏樹濤
• 申請（專利權(quán)）人: 騰訊科技（深圳）有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/764
• 代理公司: 深圳市深佳知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44285
• 代理人: 王仲凱

本申請公開了一種圖像識別模型的訓(xùn)練方法，該方法用于人工智能領(lǐng)域，該方法包括：獲取訓(xùn)練數(shù)據(jù)集合；調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第一圖像的類別進(jìn)行預(yù)測，得到第一預(yù)測標(biāo)簽；調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第二圖像的類別進(jìn)行預(yù)測，得到第二預(yù)測標(biāo)簽；對第二圖像進(jìn)行加擾處理得到第三圖像；調(diào)用圖像識別模型對第三圖像的類別進(jìn)行預(yù)測，得到第三預(yù)測標(biāo)簽；根據(jù)第一圖像所對應(yīng)的標(biāo)簽、第一圖像所對應(yīng)的第一預(yù)測標(biāo)簽、第二圖像所對應(yīng)的第二預(yù)測標(biāo)簽以及第三圖像所對應(yīng)的第三預(yù)測標(biāo)簽，對圖像識別模型的第一模型參數(shù)進(jìn)行更新。本申請公開了一種圖像識別的方法和裝置。本申請通過半監(jiān)督學(xué)習(xí)提升模型的識別能力，增強(qiáng)模型的魯棒性。

技術(shù)鄰域

本申請涉及人工智能領(lǐng)域，尤其涉及一種圖像識別模型的訓(xùn)練方法、圖像識別的方法及裝置。

背景技術(shù)

深度學(xué)習(xí)占據(jù)著機(jī)器視覺鄰域的重要地位，在機(jī)器視覺鄰域中，深度學(xué)習(xí)成為了從自動駕駛以及安保等方面的主力。盡管深度網(wǎng)絡(luò)已經(jīng)在處理復(fù)雜問題時所取得了現(xiàn)象級的成功，但是對于帶有輕微擾動的圖像而言，仍然容易出現(xiàn)識別錯誤的情況。

為了能夠抵御這類圖像對模型識別的干擾，目前，提出了一種對抗防御方法，即不斷輸入新類型的樣本并執(zhí)行對抗訓(xùn)練，不斷提升網(wǎng)絡(luò)的魯棒性。為了保證訓(xùn)練的有效性，該方法需要大量帶有標(biāo)簽的訓(xùn)練數(shù)據(jù)，通過對抗訓(xùn)練可以提升模型的魯棒性。

然而，由于對圖像進(jìn)行干擾的方式較多，即使在訓(xùn)練的過程中增加大量的樣本，仍然難以覆蓋所有的樣本，總是會存在新的攻擊樣本來欺騙網(wǎng)絡(luò)，難以提升模型的防御性能，對圖像的識別精度不高。

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

本申請實施例提供了一種圖像識別模型的訓(xùn)練方法、圖像識別的方法及裝置，利用了有標(biāo)簽數(shù)據(jù)和無標(biāo)簽數(shù)據(jù)對模型進(jìn)行半監(jiān)督訓(xùn)練，對于無標(biāo)簽的圖像而言對其進(jìn)行加擾，使得模型在訓(xùn)練過程中對是否加擾過的圖像進(jìn)行識別，從而提升模型的識別能力，增強(qiáng)模型的魯棒性。

有鑒于此，本申請第一方面提供一種圖像識別模型的訓(xùn)練方法，包括：

獲取訓(xùn)練數(shù)據(jù)集合，其中，訓(xùn)練數(shù)據(jù)集合包括至少一個具有標(biāo)簽的圖像樣本對以及至少一個不具有標(biāo)簽的圖像；

調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第一圖像的類別進(jìn)行預(yù)測，得到第一圖像所對應(yīng)的第一預(yù)測標(biāo)簽；

調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第二圖像的類別進(jìn)行預(yù)測，得到第二圖像所對應(yīng)的第二預(yù)測標(biāo)簽；

對第二圖像進(jìn)行加擾處理，得到第三圖像；

調(diào)用圖像識別模型對第三圖像的類別進(jìn)行預(yù)測，得到第三圖像所對應(yīng)的第三預(yù)測標(biāo)簽，其中，第三圖像與第二圖像具有對應(yīng)關(guān)系；

根據(jù)第一圖像所對應(yīng)的標(biāo)簽、第一圖像所對應(yīng)的第一預(yù)測標(biāo)簽、第二圖像所對應(yīng)的第二預(yù)測標(biāo)簽以及第三圖像所對應(yīng)的第三預(yù)測標(biāo)簽，對圖像識別模型的第一模型參數(shù)進(jìn)行更新。

本申請第二方面提供一種圖像識別的方法，包括：

獲取待識別圖像；

調(diào)用圖像識別模型對待識別圖像的類別進(jìn)行預(yù)測，得到圖像類別結(jié)果，其中，圖像識別模型為第一方面所描述的圖像識別模型；

向客戶端發(fā)送圖像類別結(jié)果，以使客戶端展示圖像類別結(jié)果。

本申請第三方面提供一種圖像識別模型訓(xùn)練裝置，包括：

獲取模塊，用于獲取訓(xùn)練數(shù)據(jù)集合，其中，訓(xùn)練數(shù)據(jù)集合包括至少一個具有標(biāo)簽的圖像樣本對以及至少一個不具有標(biāo)簽的圖像；

預(yù)測模塊，用于調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第一圖像的類別進(jìn)行預(yù)測，得到第一圖像所對應(yīng)的第一預(yù)測標(biāo)簽；

預(yù)測模塊，還用于調(diào)用圖像識別模型對訓(xùn)練數(shù)據(jù)集合中的第二圖像的類別進(jìn)行預(yù)測，得到第二圖像所對應(yīng)的第二預(yù)測標(biāo)簽；