Larq is an open-source deep learning library for training neural networks with extremely low precision weights and activations, such as Binarized Neural Networks (BNNs).
Existing deep neural networks use 32 bits, 16 bits or 8 bits to encode each weight and activation, making them large, slow and power-hungry. This prohibits many applications in resource-constrained environments. Larq is the first step towards solving this. It is designed to provide an easy to use, composable way to train BNNs (1 bit) and other types of Quantized Neural Networks (QNNs) and is based on the tf.keras
interface.
To build a QNN, Larq introduces the concept of quantized layers and quantizers. A quantizer defines the way of transforming a full precision input to a quantized output and the pseudo-gradient method used for the backwards pass. Each quantized layer requires an input_quantizer
and a kernel_quantizer
that describe the way of quantizing the incoming activations and weights of the layer respectively. If both input_quantizer
and kernel_quantizer
are None
the layer is equivalent to a full precision layer.
You can define a simple binarized fully-connected Keras model using the Straight-Through Estimator the following way:
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
larq.layers.QuantDense(512,
kernel_quantizer="ste_sign",
kernel_constraint="weight_clip"),
larq.layers.QuantDense(10,
input_quantizer="ste_sign",
kernel_quantizer="ste_sign",
kernel_constraint="weight_clip",
activation="softmax")])
This layer can be used inside a Keras model or with a custom training loop.
Check out our examples on how to train a Binarized Neural Network in just a few lines of code:
Before installing Larq, please install:
- Python version
3.6
or3.7
- Tensorflow version
1.14
,1.15
or2.0.0
:pip install tensorflow # or tensorflow-gpu
You can install Larq with Python's pip package manager:
pip install larq