Invention Publication
- Patent Title: METHOD AND APPARATUS FOR DETERMINING NETWORK MODEL PRUNING STRATEGY, DEVICE AND STORAGE MEDIUM
-
Application No.: EP21180897.7Application Date: 2021-06-22
-
Publication No.: EP3876166A2Publication Date: 2021-09-08
- Inventor: XI, Teng , ZHANG, Gang , WEN, Shengzhao
- Applicant: BEIJING BAIDU NETCOM SCIENCE AND TECHNOLOGY CO. LTD.
- Applicant Address: CN 100085 Beijing 2/F Baidu Campus, No. 10, Shangdi 10th, Haidian District
- Agency: Studio Torta S.p.A.
- Priority: CN202011147374 20201023
- Main IPC: G06N3/08
- IPC: G06N3/08 ; G06N3/04 ; G06N5/00

Abstract:
Embodiments of the present disclosure disclose a method and apparatus for determining a network model pruning strategy, a device and a storage medium, relate to the field of artificial intelligence technology such as computer vision and deep learning, and may be used in image processing scenarios. The method includes: generating a BN threshold search space using configuration information of the BN threshold search space for a network model for a target hardware; generating a pruning strategy code generator using the BN threshold search space; randomly generating a BN threshold code using the pruning strategy code generator; decoding the BN threshold code to obtain a candidate BN threshold; and determining a target pruning strategy of the network model for the target hardware, based on a pruning accuracy loss of the network model corresponding to the candidate BN threshold. An optimal BN threshold is obtained through automatic search, and the BN threshold is used to replace channel importance to prune the network model, thereby greatly reducing the accuracy loss of the network model after pruning.
Public/Granted literature
- EP3876166A3 METHOD AND APPARATUS FOR DETERMINING NETWORK MODEL PRUNING STRATEGY, DEVICE AND STORAGE MEDIUM Public/Granted day:2022-01-12
Information query