用于對(duì)神經(jīng)網(wǎng)絡(luò)中卷積層進(jìn)行剪枝的方法和裝置

基本信息

申請(qǐng)?zhí)?/td> CN202010171150.4 申請(qǐng)日 -
公開(kāi)（公告）號(hào) CN113392953A 公開(kāi)（公告）日 2021-09-14
申請(qǐng)公布號(hào) CN113392953A 申請(qǐng)公布日 2021-09-14
分類(lèi)號(hào) G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類(lèi) 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 聶遠(yuǎn)飛;董禎;馮歡 申請(qǐng)（專(zhuān)利權(quán)）人 瀾起科技股份有限公司
代理機(jī)構(gòu) 北京市君合律師事務(wù)所 代理人 毛健;杜小鋒
地址 200233上海市徐匯區(qū)宜山路900號(hào)1幢A6
法律狀態(tài) -

摘要

摘要 本申請(qǐng)公開(kāi)了一種用于對(duì)神經(jīng)網(wǎng)絡(luò)中卷積層進(jìn)行剪枝的方法和裝置。其中用于對(duì)神經(jīng)網(wǎng)絡(luò)中卷積層進(jìn)行剪枝的方法包括：獲取目標(biāo)神經(jīng)網(wǎng)絡(luò)，目標(biāo)神經(jīng)網(wǎng)絡(luò)包括待剪枝卷積層，待剪枝卷積層包括C個(gè)濾波器，每個(gè)濾波器中包括K個(gè)卷積核，每個(gè)卷積核包括M行N列個(gè)權(quán)重值，其中C、K、M和N為大于等于1的正整數(shù)；基于卷積核中權(quán)重值的數(shù)量M×N和目標(biāo)壓縮率確定每個(gè)卷積核中待剪枝的權(quán)重值的數(shù)量P，其中P為小于M×N的正整數(shù)；以及將待剪枝卷積層的每個(gè)卷積核中絕對(duì)值最小的P個(gè)權(quán)重值置零，形成剪枝后卷積層。