METHOD AND APPARATUS FOR PRUNING NEURAL NETWORKS

    公开(公告)号:US20220284298A1

    公开(公告)日:2022-09-08

    申请号:US17769707

    申请日:2020-10-06

    Abstract: The present invention relates to a method for pruning a neural network comprising a plurality of neurons, said method comprising: an initialization phase, wherein input information is fetched comprising at least parameters ({wni,bni}) related to said neural network and a dataset (D) representative of a task that said neural network has to deal with, wherein said parameters ({wni,bni}) comprising a weights vector (wni) and/or a bias (bni) related to at least one neuron of said plurality of neurons; a regularization phase, wherein said neural network is trained according to a training algorithm by using said dataset (D); a thresholding phase, wherein an element (wnij) of said weights vector (wni) is put at zero when its absolute value is below a given threshold (T).

Patent Agency Ranking