AI Model Lightweighting via Pruning and Quantization
Find Innovative SolutionsGenerate Solutions
Solution Overview
Problem
Existing artificial intelligence models for object detection in images require expensive hardware like GPUs and struggle with inference speed while maintaining accuracy, especially in real-time applications like automatic driving.
Innovation Solution
A method of lightweighting an artificial intelligence model through pruning, quantization, and learning by imitation, which involves converting weights to '0' based on sensitivity analysis, converting the model to a signed 8-bit integer type, and learning by imitating another model trained on a larger data set.
Engineering Contradictions & Design Principles
Engineering Contradiction Analysis
1Measurement precision
If an artificial intelligence model is used for object detection in images, then detection accuracy is improved, but inference speed deteriorates and expensive hardware like GPUs is required
Solution Approach 1:
The patent segments the AI model into multiple layers and applies different processing strategies to each layer. Specifically, it divides the model into convolutional layers, pooling layers, and fully connected layers, applying pruning and quantization selectively to different layers based on their sensitivity to these operations. This allows the model to maintain accuracy in critical layers while achieving speedup in less sensitive layers.
Solution Approach 2:
The patent changes the parameter precision of the AI model by applying quantization, which converts high-precision floating-point parameters (32-bit) to low-precision integer parameters (8-bit or 4-bit). This parameter change reduces computational complexity and memory bandwidth requirements, thereby improving inference speed while maintaining detection accuracy through careful selection of quantization strategies and calibration processes.
2Measurement precision
If an artificial intelligence model is used for object detection in images, then detection accuracy is improved, but device complexity and hardware cost increase
Solution Approach 1:
The patent replaces expensive, complex hardware (GPUs with high precision floating-point units) with cheaper, simpler hardware that can execute quantized models. By converting the model to use low-precision integers, the system can run on microcontrollers or simple processors that are orders of magnitude cheaper than GPUs, while maintaining acceptable detection accuracy through the pruning and quantization process.
Solution Approach 2:
The patent substitutes the mechanical/computational system of high-precision floating-point arithmetic with a simpler integer arithmetic system. This substitution replaces complex hardware requirements (GPUs, FPGAs) with simpler processors that can perform integer operations, significantly reducing device complexity and cost while maintaining model functionality through the quantization process.
3Speed
If model pruning is applied to reduce complexity, then inference speed is improved, but model accuracy deteriorates
Solution Approach 1:
The patent applies preliminary training to the pruned model after removing neurons or connections. This retraining process allows the model to recover accuracy lost during pruning by adjusting the remaining parameters, ensuring that the final pruned and retrained model achieves both speed improvement and maintained accuracy. The preliminary action of training before final deployment compensates for the accuracy loss from structural removal.
Solution Approach 2:
The patent uses accuracy feedback to guide the pruning process. By monitoring detection accuracy during and after pruning, the system can iteratively adjust which neurons or connections to remove, ensuring that pruning does not exceed the tolerance threshold for accuracy degradation. This feedback mechanism allows the optimization of inference speed while maintaining acceptable accuracy levels.
4Device complexity
If quantization is applied to convert to 8-bit integers, then hardware cost is reduced, but computational precision deteriorates
Solution Approach 1:
The patent changes the parameter representation from high-precision floating-point (32-bit) to low-precision integer (8-bit or 4-bit) format. This parameter change reduces the memory bandwidth and computational requirements, allowing the model to run on cheaper hardware. The loss in precision is compensated through calibration techniques and careful selection of quantization parameters to maintain acceptable detection accuracy.
Data Source
AI summary
A method of lightweighting an artificial intelligence model can increase inference speed while maintaining accuracy of the artificial intelligence model for detecting objects in an image captured by a camera as much as possible. The method may include the steps of: pruning an artificial intelligence model machine-learned using a first data set, by a data processing device; quantizing the pruned artificial intelligence model, by the data processing device; and learning the artificial intelligence model by imitating another artificial intelligence model previously trained using a second data set including a larger amount of data than the first data set, by the data processing device.


