menu
techminis

A naukri.com initiative

google-web-stories
source image

Medium

1w

read

290

img
dot

Image Credit: Medium

Prune a Neural Network with Simulated Annealing and Genetic Algorithm (Part 3 — Experiments)

  • The primary data source used for this experiment is the well-known MNIST dataset, a standard in the machine learning community.
  • The LeNet architecture (Lecun et al., 1998) is a classic convolutional neural network (CNN) widely used for image classification tasks.
  • SimulatedAnnealingPruner was applied to prune the trained LeNet with the goal to optimize the pruning mask for each layer to achieve higher sparsity while maintaining the test loss.
  • GeneticAlgorithmPruner was applied using a population-based approach to optimize the pruning masks for each layer, striking a balance between model sparsity and accuracy.

Read Full Article

like

17 Likes

For uninterrupted reading, download the app