一種等變網(wǎng)絡(luò)訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)

基本信息

申請(qǐng)?zhí)?/td> CN202210080806.0 申請(qǐng)日 -
公開（公告）號(hào) CN114528976A 公開（公告）日 2022-05-24
申請(qǐng)公布號(hào) CN114528976A 申請(qǐng)公布日 2022-05-24
分類號(hào) G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06V10/82（2022.01）I;G06V10/44（2022.01）I;G06V10/764（2022.01）I;G06K9/62（2022.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 陳智強(qiáng);余山;陳陽 申請(qǐng)（專利權(quán)）人 北京智源人工智能研究院
代理機(jī)構(gòu) 北京辰權(quán)知識(shí)產(chǎn)權(quán)代理有限公司 代理人 -
地址 100083北京市海淀區(qū)中關(guān)村東路1號(hào)院8號(hào)樓三層B201D-1
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種等變網(wǎng)絡(luò)訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)，方法包括：針對(duì)圖像集中的每一圖像，對(duì)圖像進(jìn)行目標(biāo)群變換，將變換后的圖像和圖像輸入等變網(wǎng)絡(luò)進(jìn)行學(xué)習(xí)，獲取等變網(wǎng)絡(luò)的輸出損失；獲取等變網(wǎng)絡(luò)對(duì)圖像學(xué)習(xí)的第一特征圖和對(duì)變換后的圖像學(xué)習(xí)的第二特征圖，根據(jù)第一特征圖和第二特征圖確定等變損失；利用輸出損失和等變損失對(duì)等變網(wǎng)絡(luò)的參數(shù)進(jìn)行優(yōu)化。在不需要改變網(wǎng)絡(luò)結(jié)構(gòu)的基礎(chǔ)上，通過在常規(guī)深度網(wǎng)絡(luò)結(jié)構(gòu)的損失函數(shù)中引入等變損失，并通過使用原始圖像和經(jīng)過目標(biāo)群變換的圖像輸入網(wǎng)絡(luò)，來引導(dǎo)網(wǎng)絡(luò)學(xué)習(xí)目標(biāo)變換群上的等變性，從而獲得目標(biāo)變換群的等變網(wǎng)絡(luò)，避免了為特定的變換群人工設(shè)計(jì)對(duì)應(yīng)的等變網(wǎng)絡(luò)。