- 专利标题: METHOD AND APPARATUS FOR DETERMINING NETWORK MODEL PRUNING STRATEGY, DEVICE AND STORAGE MEDIUM
-
申请号: EP21180897.7申请日: 2021-06-22
-
公开(公告)号: EP3876166A2公开(公告)日: 2021-09-08
- 发明人: XI, Teng , ZHANG, Gang , WEN, Shengzhao
- 申请人: BEIJING BAIDU NETCOM SCIENCE AND TECHNOLOGY CO. LTD.
- 申请人地址: CN 100085 Beijing 2/F Baidu Campus, No. 10, Shangdi 10th, Haidian District
- 代理机构: Studio Torta S.p.A.
- 优先权: CN202011147374 20201023
- 主分类号: G06N3/08
- IPC分类号: G06N3/08 ; G06N3/04 ; G06N5/00
摘要:
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.
公开/授权文献
信息查询