Apply Prune function on CNN

조회 수: 4 (최근 30일)
frederic feresin
frederic feresin 2019년 1월 21일
답변: Maksym Tymchenko 2023년 7월 21일
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?
  댓글 수: 2
Shaw
Shaw 2021년 11월 3일
Did you solve this question?
frederic feresin
frederic feresin 2021년 11월 3일

댓글을 달려면 로그인하십시오.

답변 (1개)

Maksym Tymchenko
Maksym Tymchenko 2023년 7월 21일
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:

카테고리

Help CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

제품


릴리스

R2018b

Community Treasure Hunt

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

Start Hunting!

Translated by