Quantization Networks (1911.09464v2)

Abstract: Although deep neural networks are highly effective, their high computational and memory costs severely challenge their applications on portable devices. As a consequence, low-bit quantization, which converts a full-precision neural network into a low-bitwidth integer version, has been an active and promising research topic. Existing methods formulate the low-bit quantization of networks as an approximation or optimization problem. Approximation-based methods confront the gradient mismatch problem, while optimization-based methods are only suitable for quantizing weights and could introduce high computational cost in the training stage. In this paper, we propose a novel perspective of interpreting and implementing neural network quantization by formulating low-bit quantization as a differentiable non-linear function (termed quantization function). The proposed quantization function can be learned in a lossless and end-to-end manner and works for any weights and activations of neural networks in a simple and uniform way. Extensive experiments on image classification and object detection tasks show that our quantization networks outperform the state-of-the-art methods. We believe that the proposed method will shed new insights on the interpretation of neural network quantization. Our code is available at https://github.com/aliyun/alibabacloud-quantization-networks.

Overview of "Quantization Networks"

The paper "Quantization Networks" presents a novel methodology for reducing the computational and memory costs of deep neural networks (DNNs) by formulating low-bit quantization as a differentiable non-linear function. This innovative perspective addresses limitations in existing approximation-based and optimization-based quantization methods by allowing for a unified, end-to-end learning approach that circumvents the gradient mismatch problem inherent in traditional quantization techniques.

Key Contributions:

1. Differentiable Quantization Function: The paper introduces a quantization function modeled as a linear combination of Sigmoid functions. This differentiable function can be seamlessly integrated into neural network architectures, facilitating the simultaneous quantization of both weights and activations without additional gradient approximations.
2. Experimentation and Results: Experiments on standard tasks, such as image classification using AlexNet, ResNet-18, and ResNet-50, as well as object detection using SSD on Pascal VOC, demonstrate superior performance over state-of-the-art quantization methods. Notably, the quantization networks achieved lossless performance with only 3-bit quantization on ResNet networks.
3. Layer-wise and Non-uniform Quantization: The layer-specific and non-uniform quantization methodology recognizes the distinct parameter distributions across different network layers, leading to better utilization of available bit budgets compared to uniform quantization.

Implications and Future Directions:

The proposed quantization network framework significantly enhances the efficiency of deploying DNNs on resource-constrained devices by reducing both the temporal and spatial complexity of computations. The flexibility and effectiveness of this approach encourage further exploration into varying network architectures and applications beyond image classification and object detection.

Moreover, the integration of the quantization function as an inherent part of the network's architecture opens up possibilities for more relaxed optimization constraints and potentially more sophisticated learning schemes, such as those incorporating dynamic precision adjustments.

Conclusion:

Quantization Networks provide a robust framework for achieving efficient DNNs, effectively bridging the gap between theoretical rigor and practical applicability. The introduction of a differentiable non-linear quantization function represents a significant step toward more streamlined and adaptive quantization processes, showing promise for widespread adoption across varying machine learning domains. Future research may focus on optimizing the trade-off between precision and efficiency further and exploring the applicability to novel tasks and network architectures.