[發明專利]基于類別掩碼的神經網絡剪枝方法在審
| 申請號： | 202111005877.6 | 申請日： | 2021-08-30 |
| 公開（公告）號： | CN113837377A | 公開（公告）日： | 2021-12-24 |
| 發明（設計）人： | 紀榮嶸;張玉鑫 | 申請（專利權）人： | 廈門大學 |
| 主分類號： | G06N3/08 | 分類號： | G06N3/08;G06K9/62 |
| 代理公司： | 廈門南強之路專利事務所(普通合伙) 35200 | 代理人： | 馬應森;曾權 |
| 地址： | 361005 福建*** | 國省代碼： | 福建;35 |
| 權利要求書： | 查看更多 | 說明書： | 查看更多 |
| 摘要： | |||
| 搜索關鍵詞： | 基于 類別 掩碼 神經網絡 剪枝 方法 | ||
【權利要求書】：
1.基于類別掩碼的神經網絡剪枝方法，其特征在于包括以下步驟：
1)為神經網絡中每一個通道分配一個類別掩碼，其維度等于數據集圖像類別數；
2)基于輸入圖像的類別激活相應的掩碼來進行訓練網絡；
3)基于訓練后的類別掩碼進行全局投票，剪枝分數較小的通道；
4)對保留下來的卷積核繼續進行訓練，得到最終的剪枝網絡。
2.如權利要求1所述基于類別掩碼的神經網絡剪枝方法，其特征在于在步驟2)中，所述基于輸入圖像的類別激活相應的掩碼來進行訓練網絡，具體為：當訓練類別A時，即激活通道對應的類別A的掩碼與該通道生成的特征圖進行相乘，進行前向和反向傳播。
3.如權利要求1所述基于類別掩碼的神經網絡剪枝方法，其特征在于在步驟3)中，所述基于訓練后的類別掩碼進行全局投票，即對每個通道的類別掩碼進行取和操作。
下載完整專利技術內容需要扣除積分，VIP會員可以免費下載。
該專利技術資料僅供研究查看技術是否侵權等信息，商用須獲得專利權人授權。該專利全部權利屬于廈門大學，未經廈門大學許可，擅自商用是侵權行為。如果您想購買此專利、獲得商業授權和技術合作，請聯系【客服】
本文鏈接：http://www.szxzyx.cn/pat/books/202111005877.6/1.html，轉載請聲明來源鉆瓜專利網。
