System and Method for Automated Precision Configuration for Deep Neural Networks
There is provided a system and method of automated precision configuration for deep neural networks. The method includes obtaining an input model and one or more constraints associated with an application and/or target device or process used in the application configured to utilize a deep neural network; learning an optimal low-precision configuration of the architecture using constraints, the training data set, and the validation data set; and deploying the optimal configuration on the target device or process for use in the application.
Latest Deeplite Inc. Patents:
This application claims priority to U.S. Provisional Patent Application No. 62/769,403 filed on Nov. 19, 2018, the contents of which are incorporated herein by reference.
TECHNICAL FIELDThe following relates to systems and methods for automated precision configuration for deep neural networks, for example by enabling low bit-precision weights and activations to be used effectively.
BACKGROUNDIn modern intelligent applications and devices, deep neural networks (DNNs) have become ubiquitous when solving complex computer tasks, such as recognizing objects in images and translating natural language. The success of these networks has been largely dependent on high performance computing machinery, such as Graphics Processing Units (GPUs) and server-class Central Processing Units (CPUs). Consequently, the adoption of DNNs to solve real-world problems is typically limited to scenarios where such computing is available. Recently, many new computer processors specifically designed for artificial intelligence (AI) applications have emerged. These dedicated processors, such as Field Programmable Gate Arrays (FPGAs), Application Specific Integrated Circuits (ASICs) and analog computers offer the promise of more efficient and accessible AI products and services. However, designing DNN models optimized for these new processors remains a significant challenge for AI engineers and application developers. Significant domain expertise and trial-and-error is often required to create an optimized DNN for a specialized hardware. One of the main challenges is how to enable a precision configuration for a given DNN architecture that maintains accuracy and optimizes for memory, energy and latency performance on a given hardware architecture. The task of quantizing individual layers of a DNN, which can contain dozens of layers, often results in sub optimal performance in a real-world environment. Thus, there is significant interest in automating the task of enabling a precision configuration for an entire DNN architecture that considers the properties of the hardware architecture to optimize memory, energy and latency as well as maintain a desired level of accuracy on the given dataset.
To address these problems, there has been a widespread push in academia and industry to make deep learning models more efficient by considering the properties of the hardware architecture in the model optimization process. Many techniques have been proposed for manual quantization of DNNs that show lower bit precision models are feasible for accurate inferencing on new input data.
Prior solutions include a variety of core quantization techniques for various DNN model architectures, as well as having efficient kernels for computation in reduced precision like ARM CMSIS, Intel MKL-DNN and Nvidia TensorRT. The main approach to model quantization is by uniform precision reduction across all layers of a DNN, for example from 32 bit Floating Point to 16 bit, or to 8 bit INT. It has been observed that once a model is trained, a lower bit precision is acceptable for the weights and activations of a DNN model to correctly compute the inference label for a given input. For this reason, many developers and hardware providers are developing in-house or add-on quantization methods that can naively convert the weights and activations of a DNN model to a supported precision for the target hardware (HW). However, when this process is applied and the model is attempted to run on a different HW, the result can often be slower, or the model may be incompatible with the new HW. Additionally, these uniform quantization approaches are often found to sacrifice too much accuracy or limit network performance on complex and large data sets.
At present, two fundamental challenges exist with current quantization techniques, namely: 1) that hand-crafted features and domain expertise is required for automated quantization 2) that time-consuming fine-tuning is often necessary to maintain accuracy.
There exists a need for scalable, automated processes for model quantization on diverse DNN architectures and hardware back-ends. Generally, it is found that the current capacity for model quantization is outpaced by the rapid development of new DNNs and disparate hardware platforms that aim to increase the applicability and efficiency of deep learning workloads.
It is an object of the following to address at least one of the above-mentioned challenges.
SUMMARYIt is recognized that a general approach that is agnostic to both the architecture and target hardware(s) is needed to optimize DNNs, making them faster, smaller and energy-efficient for use in daily life. The following relate to deep learning algorithms, for example, deep neural networks. A method for automated precision configuration, specifically quantization of DNN weights and activations, is described. The following relates to the design of a learning process to leverage trade-offs in different deep neural network precision configurations using computation constraints and hardware properties as inputs. The learning process trains an optimizer agent to adapt large, full precision networks into smaller networks of similar performance that satisfy target constraints in a platform-aware way. By design, the learning process and agent is agnostic to both network architecture and target hardware platform.
In one aspect, there is provided a method of automated precision configuration for deep neural networks, the method comprising: obtaining an input model and one or more constraints associated with an application and/or target device or process used in the application configured to utilize a deep neural network; learning an optimal low-precision configuration of the optimal architecture using the input model, constraints, the training data set, and the validation data set; and deploying the optimal configuration on the target device or process for use in the application.
In another aspect, there is provided a computer readable medium comprising computer executable instructions for automated design space exploration for deep neural networks, the computer executable instructions comprising instructions for performing the above method.
In yet another aspect, there is provided a deep neural network optimization engine configured to perform automated precision configuration for deep neural networks, the engine comprising a processor and memory, the memory comprising computer executable instructions for performing the above method.
One or more embodiments will now be described with reference to the appended drawings wherein:
All should be accessible and beneficial to various applications in everyday life. With the emergence of deep learning on embedded and mobile devices, DNN application designers are faced with stringent power, memory and cost requirements which often leads to inefficient solutions, possibly preventing people from moving to these devices. The system described below can be used to make deep learning applicable, affordable and scalable by bridging the gap between DNNs and hardware back-ends. To do so, a scalable, DNN-agnostic engine is provided, which can enable a platform-aware optimization. The engine targets information inefficiency in the implementation of DNNs, making them applicable for low-end devices. To provide such functionality, the engine:
-
- is configured to be architecture independent, allowing the engine to support different DNN architectures such as convolution neural networks (CNNs), recurrent neural networks (RNNs), etc.;
- is configured to be framework agnostic, enabling developers to readily apply the engine to a project without additional engineering overhead;
- is configured to be hardware agnostic, helping end-users to readily change the back-end hardware or port a model from one hardware to another; and.
One of the core challenges with model optimization for DNN inference is evaluating which precision configuration is best-suited for a given application. The engine described herein uses an AI-driven optimizer to overcome the drawbacks of manual model quantization. Based on computation constraints i.e. a “bit budget”, a software agent selectively changes the bit precision of different layers in the model. Information inefficiencies and novel supported bit-precisions for AI hardware are leveraged to effectively quantize the layers of a network in a platform-aware way.
Turning now to the figures,
For the model converter 22, the engine 10 is configured to support multiple frameworks 20 (e.g. TensorFlow, Pytorch, etc.) and DNN architectures (e.g. CNN, RNN, etc.), to facilitate applying the engine's capabilities on different projects with different AI frameworks 20. To do so, two layers are included, namely: a) the model convertor 22 which contains each AI frameworks' specifications and DNNs' parser to produce the intermediate representation model (IRM) 24 from the original model; and b) the IRM 24 which represents all DNN models in a standard format.
The engine 10 also provides content aware optimization, by providing a two-level intermediate layer composed of: a) the design space exploration module 26, which is an intermediate layer for finding a smaller architecture with similar performance as the given model to reduce memory footprint and computation (described in greater detail below); and b) the quantizer 28, which is a low-level layer for quantizing the network to gain further computation speedup.
Regarding the design space exploration module 26, DNNs are heavily dependent on the design of hyper-parameters like the number of hidden layers, nodes per layer and activation functions, which have traditionally been optimized manually. Moreover, hardware constraints 19 such as memory and power should be considered to optimize the model effectively. Given spaces can easily exceed thousands of solutions, it can be intractable to find a near-optimal solution manually.
Quantizing DNNs has the potential to decrease complexity and memory footprint and facilitate potential deployment on the edge devices. However, precision is typically considered at the design level of an entire model, making it difficult to consider as a tunable hyper parameter. Moreover, exploring efficient precision requires tight integration between the network design, training and implementation, which is not always feasible. Typical implementations of low precision DNNs use uniform precision across all layers of the network while mixed-precision leads to better performance. The engine 10 described herein exploits low precision weights using reinforcement learning to learn an optimal precision configuration across the neural network where each layer may have different precision to get the best out of the target platform 16. Besides mixed-precision, the engine 10 also supports uniform precision, fixed-point, dynamic fixed-point and binary/ternary networks.
It is also recognized that a major challenge lies in enabling support for multiple hardware back-ends while keeping compute, memory and energy footprints at their lowest. Content aware optimization alone is not considered to be enough to solve the challenge of supporting different hardware back ends. The reason being that primitive operations like convolution or matrix multiplication may be mapped and optimized in very different ways for each hardware back-end. These hardware-specific optimizations can vary drastically in terms of memory layout, parallelization threading patterns, caching access patterns and choice of hardware primitives.
The platform aware optimization layer that includes the mapping algorithms 30 is configured to address this challenge. This layer contains standard transformation primitives commonly found in commodity hardware such as CPUs, GPUs, FPGAs, etc. This additional layer provides a toolset to optimize DNNs for FPGAs and automatically map them onto FPGAs for model inference. This automated toolset can save design time significantly. Importantly, many homogeneous and heterogeneous multicore architectures have been introduced currently to continually improve system performance. Compared to homogeneous multicore systems, heterogeneous ones offer more computation power and efficient energy consumption because of the utilization of specialized cores for specific functions and each computational unit provides distinct resource efficiencies when executing different inference phases of deep models (e.g. Binary network on FPGA, full precision part on GPU/DSP, regular arithmetic operations on CPU, etc.). The engine 10 provides optimization primitives targeted at heterogeneous hardware 32, by automatically splitting the DNN's computation on different hardware cores to maximize energy-efficiency and execution time on the target hardware 16.
Using platform aware optimization techniques in combination with content aware optimization techniques achieves significant performance cost reduction across different hardware platforms while delivering the same inference accuracy compared to the state-of-the-art deep learning approaches.
For example, assume an application that desires to run a CNN on a low-end hardware with 60 MB memory. The model size is 450 MB and it needs to meet 10 ms critical response time for each inference operation. The model is 95% accurate, however, 90% accuracy is also acceptable. The CNN designers usually use GPUs to train and run their models, but they would now need to deal with memory and computation power limitations, new hardware architecture and satisfying all constraints (such as memory and accuracy) in the same time. It is considered infeasible to find a solution for the target hardware or may require tremendous engineering effort. In contrast, using the engine 10, and specifying the constraints 19, a user can effectively produce the optimized model by finding a feasible solution, reducing time to market and engineering effort, as illustrated in the chart shown in
Referring now to
The engine 10 provides for automated optimization of deep learning algorithms. The engine 10 also employs an efficient process for design space exploration 26 of DNNs that can satisfy target computation constraints 19 such as speed, model size, accuracy, power consumption, etc. There is provided a learning process for training optimizer agents that automatically explore design trade-offs starting with large, initial DNNs to produce compact DNN designs in a data-driven way. Once an engineer has trained an initial deep neural network on a training data set to achieve a target accuracy for a task, they would then need to satisfy other constraints for the real-world production environment and computing hardware. The proposed process makes this possible by automatically producing an optimized DNN model suitable for the production environment and hardware 16. Referring to
In step 42, shown in
To reuse weights, the engine 10 leverages the class of function-preserving transformations that help to initialize the new network to represent the same function as the given network but use different parameterization to be further trained to improve the performance. Knowledge distillation at step 60 has been employed as a component of the training process to accelerate the training of the student network, especially for large networks.
The transformation actions may lead to defected networks (e.g. not realistic kernel size, number of filters, etc.). It is not worth it to train these networks as they cannot learn properly. To improve the training process, an apparatus has been employed to detect these defected networks earlier and cut off the learning process by using a negative reward for them.
For simplicity and clarity of illustration, where considered appropriate, reference numerals may be repeated among the figures to indicate corresponding or analogous elements. In addition, numerous specific details are set forth in order to provide a thorough understanding of the examples described herein. However, it will be understood by those of ordinary skill in the art that the examples described herein may be practiced without these specific details. In other instances, well-known methods, procedures and components have not been described in detail so as not to obscure the examples described herein. Also, the description is not to be considered as limiting the scope of the examples described herein.
It will be appreciated that the examples and corresponding diagrams used herein are for illustrative purposes only. Different configurations and terminology can be used without departing from the principles expressed herein. For instance, components and modules can be added, deleted, modified, or arranged with differing connections without departing from these principles.
It will also be appreciated that any module or component exemplified herein that executes instructions may include or otherwise have access to computer readable media such as storage media, computer storage media, or data storage devices (removable and/or non-removable) such as, for example, magnetic disks, optical disks, or tape. Computer storage media may include volatile and non-volatile, removable and non-removable media implemented in any method or technology for storage of information, such as computer readable instructions, data structures, program modules, or other data. Examples of computer storage media include RAM, ROM, EEPROM, flash memory or other memory technology, CD-ROM, digital versatile disks (DVD) or other optical storage, magnetic cassettes, magnetic tape, magnetic disk storage or other magnetic storage devices, or any other medium which can be used to store the desired information and which can be accessed by an application, module, or both. Any such computer storage media may be part of the engine 10, any component of or related to the engine, etc., or accessible or connectable thereto. Any application or module herein described may be implemented using computer readable/executable instructions that may be stored or otherwise held by such computer readable media.
The steps or operations in the flow charts and diagrams described herein are just for example. There may be many variations to these steps or operations without departing from the principles discussed above. For instance, the steps may be performed in a differing order, or steps may be added, deleted, or modified.
Although the above principles have been described with reference to certain specific examples, various modifications thereof will be apparent to those skilled in the art as outlined in the appended claims.
Claims
1. A method of automated precision configuration for deep neural networks, the method comprising:
- obtaining an input model and one or more constraints associated with an application and/or target device or process used in the application configured to utilize a deep neural network;
- learning an optimal low-precision configuration of the optimal architecture using the input model, constraints, a training data set, and a validation data set; and
- deploying the optimal configuration on the target device or process for use in the application.
2. The method of claim 1, wherein the optimal configuration is learned using a policy to generate an optimized model from the input model.
3. The method of claim 2, wherein the optimal low-precision configuration of the optimal architecture is learned using the policy to generate a quantized network, the method further comprising:
- fine tuning the quantized network with a knowledge distillation process;
- evaluating the fine-tuned network;
- applying a reward function; and
- iterating for at least one additional quantized network and selecting the optimal low-precision configuration.
4. The method of claim 3, wherein selecting the optimal low-precision configuration comprises selecting a precision configuration that achieves the best reward as determined by the reward function, for the constraints on the target device or process.
5. The method of claim 1, wherein learning the optimal low-precision configuration comprises exploiting low precision weights using reinforcement learning to learn the optimal low-precision configuration across the deep neural network.
6. The method of claim 5, wherein each layer comprises a different precision.
7. The method of claim 1, wherein the constraints comprise at least one of: accuracy, power, cost, supported precision, speed.
8. The method of claim 7, wherein a computation constraint comprises a bit budget.
9. The method of claim 1, wherein the application is an artificial intelligence-based application.
10. A non-transitory computer readable medium comprising computer executable instructions for automated design space exploration for deep neural networks, the computer executable instructions comprising instructions for:
- obtaining an input model and one or more constraints associated with an application and/or target device or process used in the application configured to utilize a deep neural network;
- learning an optimal low-precision configuration of the optimal architecture using the input model, constraints, a training data set, and a validation data set; and
- deploying the optimal configuration on the target device or process for use in the application.
11. A deep neural network optimization engine configured to perform automated design space exploration for deep neural networks, the engine comprising a processor and memory, the memory comprising computer executable instructions for:
- obtaining an input model and one or more constraints associated with an application and/or target device or process used in the application configured to utilize a deep neural network;
- learning an optimal low-precision configuration of the optimal architecture using the input model, constraints, a training data set, and a validation data set; and
- deploying the optimal configuration on the target device or process for use in the application.
12. The engine of claim 11, wherein the optimal configuration is learned using a policy to generate an optimized model from the input model.
13. The engine of claim 2, wherein the optimal low-precision configuration of the optimal architecture is learned using the policy to generate a quantized network, further comprising instructions for:
- fine tuning the quantized network with a knowledge distillation process;
- evaluating the fine-tuned network;
- applying a reward function; and
- iterating for at least one additional quantized network and selecting the optimal low-precision configuration.
14. The engine of claim 13, wherein selecting the optimal low-precision configuration comprises selecting a precision configuration that achieves the best reward as determined by the reward function, for the constraints on the target device or process.
15. The engine of claim 11, wherein learning the optimal low-precision configuration comprises exploiting low precision weights using reinforcement learning to learn the optimal low-precision configuration across the deep neural network.
16. The engine of claim 15, wherein each layer comprises a different precision.
17. The engine of claim 11, wherein the constraints comprise at least one of: accuracy, power, cost, supported precision, speed.
18. The engine of claim 17, wherein a computation constraint comprises a bit budget.
19. The engine of claim 11, wherein the application is an artificial intelligence-based application.
Type: Application
Filed: Nov 18, 2019
Publication Date: Nov 11, 2021
Applicant: Deeplite Inc. (Montréal, QC)
Inventors: Ehsan SABOORI (Montrèal), Davis Mangan SAWYER (Montrèal), MohammadHossein ASKARIHEMMAT (Montrèal), Olivier MASTROPIETRO (Montrèal)
Application Number: 17/250,928