METHOD AND APPARATUS FOR PRUNING NEURAL NETWORKS

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 o...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: TARTAGLIONE, Enzo, ODIERNA, Francesco, FIANDROTTI, Attilio, BRAGAGNOLO, Andrea, GRANGETTO, Marco
Format: Patent
Sprache:eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung: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).