一種基于語義分割的檢測模型壓縮方法

基本信息

申請?zhí)?/td> CN201910259127.8 申請日 -
公開（公告）號 CN109993304B 公開（公告）日 2020-12-15
申請公布號 CN109993304B 申請公布日 2020-12-15
分類號 G06N3/08;G06F40/30;G06F16/35 分類 計算；推算；計數(shù);
發(fā)明人 劉陽;鄭全新;趙英;張磊;董小棟;孟祥松;鄧家勇;江龍;王亞濤 申請（專利權(quán)）人 北京同方軟件有限公司
代理機(jī)構(gòu) - 代理人 -
地址 100083 北京市海淀區(qū)王莊路清華同方科技廣場A座29層
法律狀態(tài) -

摘要

摘要 一種基于語義分割的檢測模型壓縮方法，涉及人工智能領(lǐng)域和計算機(jī)視覺領(lǐng)域。本發(fā)明的方法步驟為：（一）剪枝：1）輸入卷積核權(quán)重。2）對訓(xùn)練完成的網(wǎng)絡(luò)模型進(jìn)行剪枝，獲得稀疏權(quán)重的參數(shù)空間。（二）語義分割：1）對參數(shù)空間進(jìn)行語義分割，獲取超參數(shù)塊和中心詞匯，計算超參數(shù)塊中心位置。2）使用中心詞匯更新原有參數(shù)空間。3）判斷當(dāng)前中心詞匯與上一中心詞匯變化是否小于指定閾值，如果大于則繼續(xù)搜索與中心詞匯相近參數(shù)，并更新中心詞匯返回步驟2）；小于閾值則結(jié)束中心詞匯更新。（三）模型存儲：將訓(xùn)練得到的超參數(shù)塊邊界位置、參數(shù)塊中心位置、和中心詞匯值進(jìn)行保存。本發(fā)明使用超參數(shù)對整個參數(shù)空間進(jìn)行描述，實現(xiàn)參數(shù)空間的整體壓縮，最大化的提高模型整體壓縮比。