Apply Prune function on CNN

1 visualizzazione (ultimi 30 giorni)
frederic feresin
frederic feresin il 21 Gen 2019
Risposto: Maksym Tymchenko il 21 Lug 2023
when pruning is applied to Alexnet, ML returns: "Undefined function 'prune' for input arguments of type 'SeriesNetwork'!" could you please explain how to apply "prune" function to CNN?

Risposte (1)

Maksym Tymchenko
Maksym Tymchenko il 21 Lug 2023
In MATLAB, you can compress deep neural networks by using the Deep Learning Model Quantization Library, which is a free Add-On to the Deep Learning Toolbox.
This library contains a set of different functions that allow you to compress your deep learning network.
You mentioned pruning, this is a type of structural compression where the least important weights in the network are discarded making the network smaller. The "prune" function that you mentioned only accepts "shallow neural networks" as input. That is the reason why it did not work on AlexNet. If you want to perform structural compression on a deep neural network like Alexnet, you can use either:
Both of these functions work only on dlnetwork objects, so you might need to convert your network to that form in order to use them. For an example of how to prune convolutional filters in a CNN see:

Categorie

Scopri di più su Image Data Workflows in Help Center e File Exchange

Prodotti


Release

R2018b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by