Iterative Pruning - 2.0 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
Release Date
2.0 English

The pruner is designed to reduce the number of model parameters while minimizing the accuracy loss. This is done iteratively as shown in the following figure. Pruning results in accuracy loss while retraining recovers accuracy. Pruning, followed by retraining, forms one iteration. In the first iteration of pruning, the input model is the baseline model, and it is pruned and fine-tuned. In subsequent iterations, the fine-tuned model obtained from the previous iterations is used to prune again. This process is usually repeated several times until a desired sparse model is obtained. A model cannot be pruned to a smaller size at once. Once too many parameters are removed from the model, the performance of the model is reduced and it is challenging to restore the model.

Important: The reduction parameter is gradually increased in every iteration to improve accuracy during the fine-tuning stage.

Following the process of iterative pruning, higher pruning rates can be achieved without any significant loss of model performance.

Figure 1. Iterative Process of Pruning

The four primary tasks in iterative pruner are as follows:

Perform a sensitivity analysis on the model to determine the optimal pruning strategy.
Reduce the number of computations in the input model.
Retrain the pruned model to recover accuracy.
Generate a dense model with reduced weights.

Follow these steps to prune a model. The steps are also shown in the following figure.

  1. Analyze the original baseline model.
  2. Prune the model.
  3. Fine-tune the pruned model.
  4. Repeat steps 2 and 3 several times.
  5. Transform the pruned sparse model to a final dense encrypted model to be used with Vitis AI Quantizer.
Figure 2. Iterative Pruning Workflow

Guidelines for Better Pruning Results

The following is a list of suggestions for better pruning results, higher pruning rate, and smaller accuracy loss.

  • Use as much data as possible to perform a model analysis. Ideally, you should use all the data in the validation dataset but this can be time consuming. You can also use partial validation set data, but you need to make sure at least half of the data set is used.
  • During the fine-tuning stage, experiment with a few parameters, including the initial learning rate, the learning rate decay policy and use the best result as the input to the next round of pruning.
  • The data used in fine-tuning should be the same as the data used to train the baseline.
  • If the accuracy does not improve sufficiently after several fine-tuning experiments, try reducing the pruning rate and then re-run pruning and fine-tuning.