用于搜索神經(jīng)網(wǎng)絡(luò)的方法、裝置及存儲(chǔ)介質(zhì)

基本信息

申請(qǐng)?zhí)?/td> CN202210308712.4 申請(qǐng)日 -
公開（公告）號(hào) CN114492767B 公開（公告）日 2022-07-19
申請(qǐng)公布號(hào) CN114492767B 申請(qǐng)公布日 2022-07-19
分類號(hào) G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06N20/10（2019.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 陳志軒;艾國(guó);楊作興;房汝明;向志宏 申請(qǐng)（專利權(quán)）人 深圳比特微電子科技有限公司
代理機(jī)構(gòu) 中國(guó)貿(mào)促會(huì)專利商標(biāo)事務(wù)所有限公司 代理人 -
地址 518000廣東省深圳市南山區(qū)高新南六道航盛科技大廈801
法律狀態(tài) -

摘要

摘要 本公開涉及一種用于搜索神經(jīng)網(wǎng)絡(luò)的方法、裝置及存儲(chǔ)介質(zhì)。提供了一種用于搜索神經(jīng)網(wǎng)絡(luò)的方法，包括：設(shè)置神經(jīng)網(wǎng)絡(luò)的搜索空間，其中，所述搜索空間包括候選的一個(gè)或多個(gè)不同類型的作為神經(jīng)網(wǎng)絡(luò)單元結(jié)構(gòu)的塊以及各個(gè)類型的塊的超參數(shù)的取值范圍；以及通過(guò)對(duì)搜索空間進(jìn)行采樣來(lái)獲取采樣得到的塊的類型和超參數(shù)，根據(jù)所述塊的類型和超參數(shù)確定對(duì)應(yīng)的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)，其中，所述采樣包括以當(dāng)前采樣的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)的性能指標(biāo)為約束并基于貝葉斯優(yōu)化來(lái)獲取下一個(gè)采樣的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)。