- Patent Title: Method and system for balanced-weight sparse convolution processing
-
Application No.: US17391432Application Date: 2021-08-02
-
Publication No.: US11763150B2Publication Date: 2023-09-19
- Inventor: Zhibin Xiao , Enxu Yan , Wei Wang , Yong Lu
- Applicant: MOFFETT INTERNATIONAL CO., LIMITED
- Assignee: Moffett International Co., Limited
- Current Assignee: Moffett International Co., Limited
- Current Assignee Address: HK Hong Kong
- Agency: Sheppard Mullin Richter & Hampton LLP
- Main IPC: G06N3/08
- IPC: G06N3/08 ; G06N3/04

Abstract:
Methods, systems, and apparatus, including computer programs encoded on computer storage media, for balanced-weight sparse convolution processing. An exemplary method comprises: obtaining an input tensor and a plurality of filters at a layer within a neural network; segmenting the input tensor into a plurality of sub-tensors; dividing a channel dimension of each of the plurality of filters into a plurality of channel groups; pruning each of the plurality of filters so that each of the plurality of channel groups of each filter comprises a same number of non-zero weights; segmenting each of the plurality of filters into a plurality of the sub-filters according to the plurality of channel groups; and assigning the plurality of sub-tensors and the plurality of sub-filters to a plurality of processors for parallel convolution processing.
Public/Granted literature
- US20210406686A1 METHOD AND SYSTEM FOR BALANCED-WEIGHT SPARSE CONVOLUTION PROCESSING Public/Granted day:2021-12-30
Information query