MATLAB Answers

Feature extraction using CNN and classification with SVM

30 views (last 30 days)
Hi,
I have a question on feature extraction from 2D CNN and classifying features with SVM. First let me introduce what I am trying to do;
1) I use pretrained network AlexNet which is trained with ImageNet.
2) I have a small dataset and use transfer learning for the classification problem. First, I trained my database with AlexNet by retraining all the parameters inside the network (no freezing layers) and observed the accuracy.
3) Now I want to classify the extracted features from the network with SVM. Should I use the initial AlexNet network's layer for feature extraction (default AlexNet) or the retrained network's layer on step 2?
I actually tried both of them and acquired higher accuracy on retrained network with almost %20 difference compared with the initial AlexNet. Is it just because the retrained parameters performed well on SVM? That's why a higher accuracy observed? Which one should be used?

Accepted Answer

Raynier Suresh
Raynier Suresh on 17 Mar 2021
Hi, It usually depends on the problem and the approach would change for different problems, there is no single answer for it. Based on the data and the problem you are trying to solve choose the most suitable method.

More Answers (0)

Community Treasure Hunt

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

Start Hunting!