Skip to main content

A Model Compression Toolkit for neural networks

Project description

Model Compression Toolkit (MCT)

Model Compression Toolkit (MCT) is an open-source project for neural network model optimization under efficient, constrained hardware.

This project provides researchers, developers, and engineers tools for optimizing and deploying state-of-the-art neural networks on efficient hardware.

Specifically, this project aims to apply quantization to compress neural networks.

MCT is developed by researchers and engineers working at Sony Semiconductor Israel.

Table of Contents

Getting Started

This section provides an installation and a quick starting guide.

Installation

To install the latest stable release of MCT, run the following command:

pip install model-compression-toolkit

For installing the nightly version or installing from source, refer to the installation guide.

Quick start & tutorials

For an example of how to use MCT with TensorFlow or PyTorch on various models and tasks, check out the quick-start page and the results CSV.

In addition, a set of notebooks are provided for an easy start. For example:

Supported Python Versions

Currently, MCT is being tested on various Python versions:

Python Version
Run Tests - Python 3.10
Run Tests - Python 3.9
Run Tests - Python 3.8
Run Tests - Python 3.7

Supported NN-Frameworks Versions

MCT supports compressing models built with the TensorFlow or PyTorch frameworks, and is tested on various python versions:

TensorFlow Version PyTorch Version
tests tests
tests tests
tests tests

Supported Features

MCT supports different quantization methods:

Quantization Method Complexity Computational Cost
PTQ Low Low (order of minutes)
GPTQ (parameters fine-tuning using gradients) Mild Mild (order of 2-3 hours)
QAT High High (order of 12-36 hours)

In addition, MCT supports different quantization schemes for quantizing weights and activations:

  • Power-Of-Two (hardware-friendly quantization [1])
  • Symmetric
  • Uniform

Main features:

  • Graph optimizations: Transforming the model to an equivalent (yet, more efficient) model (for example, batch-normalization layer folding to its preceding linear layer).
  • Quantization parameter search: Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Error, but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
  • Advanced quantization algorithms: To prevent a performance degradation some algorithms are applied such as:
    • Shift negative correction: Symmetric activation quantization can hurt the model's performance when some layers output both negative and positive activations, but their range is asymmetric. For more details please visit [1].
    • Outliers filtering: Computing z-score for activation statistics to detect and remove outliers.
  • Clustering: Using non-uniform quantization grid to quantize the weights and activations to match their distributions.*
  • Mixed-precision search: Assigning quantization bit-width per layer (for weights/activations), based on the layer's sensitivity to different bit-widths.
  • Visualization: You can use TensorBoard to observe useful information for troubleshooting the quantized model's performance (for example, the model in different phases of the quantization, collected statistics, similarity between layers of the float and quantized model and bit-width configuration for mixed-precision quantization). For more details, please read the visualization documentation.
  • Target Platform Capabilities: The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the TPC README.

Experimental features

Some features are experimental and subject to future changes.

For more details, we highly recommend visiting our project website where experimental features are mentioned as experimental.

Results

Keras

Graph of MobileNetV2 accuracy on ImageNet vs average bit-width of weights, using single-precision quantization, mixed-precision quantization, and mixed-precision quantization with GPTQ.

For more results, please see [1]

Pytorch

We quantized classification networks from the torchvision library. In the following table we present the ImageNet validation results for these models:

Network Name Float Accuracy 8Bit Accuracy
MobileNet V2 [3] 71.886 71.444
ResNet-18 [3] 69.86 69.63
SqueezeNet 1.1 [3] 58.128 57.678

For more results, please refer to quick start.

Contributions

MCT aims at keeping a more up-to-date fork and welcomes contributions from anyone.

*You will find more information about contributions in the Contribution guide.

License

Apache License 2.0.

References

[1] Habi, H.V., Peretz, R., Cohen, E., Dikstein, L., Dror, O., Diamant, I., Jennings, R.H. and Netzer, A., 2021. HPTQ: Hardware-Friendly Post Training Quantization. arXiv preprint.

[2] MobilNet from Keras applications.

[3] TORCHVISION.MODELS

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mct-nightly-1.9.0.20230702.post453.tar.gz (353.6 kB view details)

Uploaded Source

Built Distribution

File details

Details for the file mct-nightly-1.9.0.20230702.post453.tar.gz.

File metadata

File hashes

Hashes for mct-nightly-1.9.0.20230702.post453.tar.gz
Algorithm Hash digest
SHA256 57a6c3f41c7582cf2cc9737dc8cc78a7be05020f662ba4e7a3a47c19a51f09c4
MD5 749fefb5b67d6e8075f978f7ab4db4ce
BLAKE2b-256 6403f4ab64c8f51583dca2b0c3a5e6fa27b8c0e41117d0bced7e5affc3e29551

See more details on using hashes here.

File details

Details for the file mct_nightly-1.9.0.20230702.post453-py3-none-any.whl.

File metadata

File hashes

Hashes for mct_nightly-1.9.0.20230702.post453-py3-none-any.whl
Algorithm Hash digest
SHA256 88026d55973b0cc3d6b527614a1e03920cb3f642a82eb59721078d2fda295fdd
MD5 18bc06d8f9bec71cf9f61b3190348692
BLAKE2b-256 56a08cf6ea5afe16c0cc6ac12d6227de7485c5defc7118091d6a838f808cbfe4

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page