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

基本信息

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

摘要

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