EFFICIENT INFERENCE WITH FAST POINT-SIDED FOLDING

Description

INTRODUCTION

[0001] Aspects of the present disclosure relate to optimizing machine learning model performance, and in particular to improving the efficiency of pointwise convolutions.

[0002] Machine learning may produce a trained model, such as an artificial neural network, classification tree, and the like, which represents a generalized fit to a set of training data that is labeled. Applying the trained model to new data produces inferences, which may be used to gain insights regarding the new data. In some cases, applying the trained model to the new data is referred to as "running an inference" on the new data.

[0003] Creating inferences is computationally intensive, especially as model complexity increases. Accordingly, methods are needed for improving the performance of machine learning models, such as by making inferencing more efficient with neural network models. Attention is drawn to KEIVAN ALIZADEH VAHID ET AL: "Butterfly Transform: An Efficient FFT Based Neural Architecture Design", ARXIV.ORG, CORNELL UNIVERSITY LIBRARY, 201 OLIN LIBRARY CORNELL UNIVERSITY ITHACA, NY 14853, 5 June 2019 (2019-06-05), arXiv: 1906.02256v1, relating to the Butterfly Transform, BFT. BFT is a light weight channel fusion method that reduces the computational complexity of point-wise convolutions from O(n^2) of conventional solutions to O(n log n) with respect to the number of channels while improving the accuracy of the networks under the same range of FLOPs. The BFT generalizes the Discrete Fourier Transform in that its parameters are learned at training time. Replacing channel fusion convolutions with BFT offers 3% absolute top-1 improvement for MobileNetV1-0.25 and 2.5% for ShuffleNet V2-0.5 while maintaining the same number of FLOPS.BRIEF SUMMARY

[0004] The invention is defined by the appended independent claims. Further embodiments are defined by the appended dependent claims. According to an aspect of the invention, there is provided a method according to claim 1.

[0005] According to a further aspect of the invention, there is provided a processing system according to claim 10.

[0006] According to a further aspect of the invention, there is provided a non-transitory computer-readable medium according to claim 11.

[0007] The following description and the related drawings set forth in detail certain illustrative features of one or more embodiments.BRIEF DESCRIPTION OF THE DRAWINGS

[0008] The appended figures depict certain aspects of the one or more embodiments. FIGS. 1A-1D depict examples of various types of neural networks. FIG. 2 depicts an example of a pointwise convolution operation. FIG. 3 depicts an example network with input and output activations. FIG. 4 depicts an example of a fast pointwise convolution method in a bottom-up approach. FIG. 5 depicts an example of a fast pointwise convolution method with a top-down approach. FIG. 6 depicts an example of fast pointwise convolution utilizing a form of variable weight density reduction in which stages are entirely weighted or are entirely not weighted. FIGS. 7A and 7B depict example computations showing the performance improvement in terms of both complexity and number of parameters of fast pointwise convolution over conventional pointwise convolution and butterfly transformation pointwise convolution. FIG. 8 depicts an example method 800 for performing fast pointwise convolution. FIG. 9 depicts an example implementation of a system-on-a-chip (SOC) that may be implemented with embodiments described herein. FIG. 10 depicts an example schematic diagram of a multi-processor processing system that may be implemented with embodiments described herein.

[0009] To facilitate understanding, identical reference numerals have been used, where possible, to designate identical elements that are common to the drawings.DETAILED DESCRIPTION

[0010] Aspects of the present disclosure provide apparatuses, methods, processing systems, and computer-readable mediums for optimizing machine learning model performance, and in particular for improving the efficiency of pointwise convolutions.Deep Neural Networks and Deep Learning

[0011] Deep learning architectures may perform complex tasks, such as object recognition, by learning to represent inputs at successively higher levels of abstraction in each layer, thereby building up a useful feature representation of the input data. In this way, deep learning addresses a major bottleneck of traditional machine learning.

[0012] Prior to the advent of deep learning, a machine learning approach for a task may have relied heavily on human engineered features, perhaps in combination with a shallow classifier. A shallow classifier may be a two-class linear classifier, for example, in which a weighted sum of input values (e.g., input vector components) may be compared with a threshold to predict to which class the input belongs. Human engineered features may be templates or kernels tailored to a specific problem domain by engineers with domain expertise.

[0013] Deep learning architectures, in contrast, may learn to represent features that are similar to what a human engineer might design, but through training. Furthermore, a deep network (e.g., a deep neural network) may learn to represent and recognize new types of features that a human might not have considered.

[0014] A deep learning architecture may be configured to learn a hierarchy of features. For example, if presented with visual data, the first layer may learn to recognize relatively simple features, such as edges, in the input stream. In another example, if presented with auditory data, the first layer may learn to recognize spectral power in specific frequencies. The second layer, taking the output of the first layer as input, may learn to recognize combinations of features, such as simple shapes for visual data or combinations of sounds for auditory data. For instance, higher layers may learn to represent complex shapes in visual data or words in auditory data. Still higher layers may learn to recognize common visual objects or spoken phrases.

[0015] Deep learning architectures may perform especially well when applied to problems that have a natural hierarchical structure. For example, the classification of motorized vehicles may benefit from first learning to recognize wheels, windshields, and other features. These features may be combined at higher layers in different ways to recognize cars, trucks, and airplanes.

[0016] Neural networks may be designed with a variety of connectivity patterns. For example, in feed-forward networks, information is passed from lower to higher layers, with each neuron in a given layer communicating to neurons in higher layers.

[0017] Neural networks may also have recurrent or feedback (also called top-down) connections. A connection form the output from a neuron in a given layer to another neuron in the same layer is called a recurrent connection. A recurrent architecture may be helpful in recognizing patterns that span more than one of the input data chunks that are delivered to the neural network in a sequence. A connection from a neuron in a given layer to a neuron in a lower layer is called a feedback (or top-down) connection. A network with many feedback connections may be helpful when the recognition of a high-level concept may aid in discriminating the particular low-level features of an input.

[0018] The connections between layers of a neural network may be fully connected or locally connected.

[0019] FIG. 1A illustrates an example of a fully connected neural network 102. In a fully connected neural network 102, a neuron in a first layer may communicate its output to every neuron in a second layer, so that each neuron in the second layer will receive input from every neuron in the first layer.

[0020] FIG. 1B illustrates an example of a locally connected neural network 104. In a locally connected neural network 104, a neuron in a first layer may be connected to a limited number of neurons in the second layer. More generally, a locally connected layer of the locally connected neural network 104 may be configured so that each neuron in a layer will have the same or a similar connectivity pattern, but with connections strengths that may have different values (e.g., 110, 112, 114, and 116). The locally connected connectivity pattern may give rise to spatially distinct receptive fields in a higher layer, because the higher layer neurons in a given region may receive inputs that are tuned through training to the properties of a restricted portion of the total input to the network.

[0021] One example of a locally connected neural network is a convolutional neural network. FIG. 1C illustrates an example of a convolutional neural network 106. The convolutional neural network 106 may be configured such that the connection strengths associated with the inputs for each neuron in the second layer are shared (e.g., 108). Convolutional neural networks may be well suited to problems in which the spatial location of inputs is meaningful.

[0022] The processing of each layer of a convolutional network may be considered a spatially invariant template or basis projection. If the input is first decomposed into multiple channels, such as the red, green, and blue channels of a color image, then the convolutional network trained on that input may be considered three-dimensional, with two spatial dimensions along the axes of the image and a third dimension capturing color information. The outputs of the convolutional connections may be considered to form a feature map in the subsequent layer, with each element of the feature map receiving input from a range of neurons in the previous layer and from each of the multiple channels. The values in the feature map may be further processed with a non-linearity, such as a rectification, max(0,x). Values from adjacent neurons may be further pooled, which corresponds to down sampling, and may provide additional local invariance and dimensionality reduction.

[0023] One type of convolutional neural network is a deep convolutional network (DCN). Deep convolutional networks (DCNs) are networks of convolutional layers, configured with additional pooling and normalization layers. DCNs can be trained using supervised learning in which both the input and output targets are known for many exemplars and are used to modify the weights of the network by use of gradient descent methods.

[0024] DCNs may be feed-forward networks. In addition, as described above, the connections from a neuron in a first layer of a DCN to a group of neurons in the next higher layer are shared across the neurons in the first layer. The feed-forward and shared connections of DCNs may be exploited for fast processing. The computational burden of a DCN may be much less, for example, than that of a similarly sized neural network that comprises recurrent or feedback connections.

[0025] FIG. 1D illustrates a detailed example of a DCN 100 designed to recognize visual features from an image 126 input from an image capturing device 130, such as a car-mounted camera. The DCN 100 of the current example may be trained to identify traffic signs and a number provided on the traffic sign. Of course, the DCN 200 may be trained for other tasks, such as identifying lane markings or identifying traffic lights. These are just some example tasks, and many others are possible.

[0026] DCN 100 may be trained with supervised learning. During training, the DCN 100 may be presented with an image, such as the image 126 of a speed limit sign, and a forward pass may then be computed to produce an output 122.

[0027] In this example, DCN 100 includes a feature extraction section and a classification section. Upon receiving the image 126, a convolutional layer 132 applies convolutional kernels (not shown) to the image 126 to generate a first set of feature maps 118. Generally, a kernel comprises a two-dimensional array of weights designed to emphasize different aspects of an input data channel, and a filter comprises a three-dimensional structure comprising multiple kernels stacked together. Three-dimensional filters are frequently used in deep learning.

[0028] In one example, the convolutional kernel for the convolutional layer 132 may be a 5x5 kernel that generates 28x28 feature maps. Further in the present example, because four different feature maps are generated in the first set of feature maps 118, four different convolutional kernels are applied to the image 126 at the convolutional layer 132. The convolutional kernels may also be referred to as filters or convolutional filters.

[0029] The first set of feature maps 118 may be subsampled by a max pooling layer (not shown) to generate a second set of feature maps 120. The max pooling layer reduces the size of the first set of feature maps 118. That is, a size of the second set of feature maps 120, such as 14x14, is less than the size of the first set of feature maps 118, such as 28x28. The reduced size provides similar information to a subsequent layer while reducing memory consumption. The second set of feature maps 120 may be further convolved via one or more subsequent convolutional layers (not shown) to generate one or more subsequent sets of feature maps (not shown).

[0030] In the example of FIG. 1D, the second set of feature maps 120 is convolved to generate a first feature vector 124. Furthermore, the first feature vector 124 is further convolved to generate a second feature vector 128. Each feature of the second feature vector 128 may include a number that corresponds to a possible feature of the image 126, such as "sign," "60," and "100." A softmax function (not shown) may convert the numbers in the second feature vector 128 to a probability. As such, an output 122 of the DCN 100 is a probability of the image 126 including one or more features.

[0031] In the present example, the probabilities in the output 122 for "sign" and "60" are higher than the probabilities of the others of the output 122, such as "30," "40," "50," "70," "80," "90," and "100".

[0032] Before training DCN 100, the output 122 produced by DCN 100 is likely to be incorrect. Thus, an error may be calculated between the output 122 and a target output. The target output is the ground truth of the image 126 (e.g., "sign" and "60"). The weights of DCN 100 may then be adjusted so the output 122 of DCN 100 is more closely aligned with the target output.

[0033] To adjust the weights of DCN 100, a learning algorithm may compute a gradient vector for the weights. The gradient may indicate an amount that an error would increase or decrease if the weight were adjusted. At the top layer, the gradient may correspond directly to the value of a weight connecting an activated neuron in the penultimate layer and a neuron in the output layer. In lower layers, the gradient may depend on the value of the weights and on the computed error gradients of the higher layers. The weights may then be adjusted to reduce the error. This manner of adjusting the weights may be referred to as "back propagation" as it involves a "backward pass" through the neural network.

[0034] In practice, the error gradient of weights may be calculated over a small number of examples, so that the calculated gradient approximates the true error gradient. This approximation method may be referred to as stochastic gradient descent. Stochastic gradient descent may be repeated until the achievable error rate of the entire system has stopped decreasing or until the error rate has reached a target level. After learning, DCN 100 may be presented with new images and a forward pass through the network may yield an output 122 that may be considered an inference or a prediction of the DCN.Convolution in Convolutional Neural Networks

[0035] Convolution is a data analysis technique used in, for example, signal processing, image processing, machine learning, and other technical fields. In deep learning, convolution is used to extract useful features from an input data set. For example, in convolutional neural networks, such as described above, convolution enables the extraction of different features using filters whose weights are automatically learned during training. The extracted features are then combined to make inferences.

[0036] One way to reduce the computational burden (e.g., measured in floating point operations per second (FLOPs)) and the number parameters associated with a convolutional neural network is to factorize the convolutional layers, using a separable depthwise convolution, into two components: (1) spatial fusion, where each spatial channel is convolved independently by a depthwise convolution; and (2) channel fusion, where all the spatial channels are linearly combined by 1x1-convolutions, known as pointwise convolutions.

[0037] During spatial fusion, the network learns features from the spatial planes and during channel fusion the network learns relations between these features across channels. This is sometimes implemented using 3x3 filters for the spatial fusion, and 1x1 filters for the channel fusion. Generally, channel fusion via pointwise convolution is useful for dimensionality reduction for efficient computations, efficient low dimensional embedding or feature pooling, and for applying nonlinearity again after convolution, to name a few things.

[0038] For example, a pointwise convolution may use a 1x1xd kernel that iterates through every single point in an input image of depth d. The depth d of a pointwise convolution kernel generally matches the number of channels of the input data, such as an image with multiple channels for color data.

[0039] Generally, an activation layer, such as ReLU, PReLU, Softmax, or others, can be applied after every layer of a neural network. Applying 1x1xd kernels and adding an activation layer after the kernel may give a network added depth, which may increase its performance.

[0040] FIG. 2 depicts an example of a pointwise convolution operation in which a 1x1x5 kernel 204 is iterated through an 8x8x5 input image 202 to generate an 8x8x1 output image 206. As is depicted in this example, output image 206 has reduced dimensionality (1 channel versus 8), which allows for more efficient computations with output image 206.

[0041] Though not depicted in FIG. 2, multiple (e.g., m) pointwise convolution kernels (e.g., in a filter) can be used to increase the number of output channels of input data. So, for example, m = 256 1x1x5 kernels 204 can be generated, which each output an 8x8x1 output image (e.g., 206), and these output images can be stacked to get a resulting image of 8x8x256, i.e., an image 8 pixels wide by 8 pixels tall with a depth of 256 channels. The resulting increase in output channels provides more parameters for training, which may improve a model's ability to identify features (e.g., in input image 202).

[0042] Inspecting the computational profile of factorized convolutional neural networks at inference time reveals that the computational burden of the spatial fusion is relatively small compared to that of the channel fusion. In fact, the computational complexity of the pointwise convolutions in the channel fusion is quadratic in the number of channels (O(n 2< ) where n is the number of channels.

[0043] FIG. 3 depicts an example of a fully connected network 300 with N = 4 inputs and M = 4 outputs. To determine the complexity of performing a conventional pointwise convolution on network 300, consider a pointwise (1x1) convolution that takes the N activations from input feature map X and produces M activations as output feature map Y. Thus: Y = WX , where X∈R N< , W∈R N< * M< , and Y∈R M< .

[0044] In a conventional pointwise convolution operation, there are N * M multiplications between the activations (X) and weights (W); thus, the order of complexity is O(N 2< ).Fast Pointwise Convolution Methods

[0045] Described herein are fast pointwise convolution methods that use factorization and reduced weight density instead of strictly pointwise multiplication to achieve a similar transformation from feature map X to feature map Y with operation Y = WX at a significantly reduced operational complexity and with reduced number of weights (parameters).

[0046] For example, assume X is of shape [x 0 , x 1 , x 2 , x 3 ] and Y is of shape [y 0 , y 1 , y 2 , y 3 ], along the x 1 and y 1 dimensions of X and Y, respectively, and further assume size(x 1 ) = N and size(y 1 ) = M. Then, a conventional pointwise convolution operates at a complexity of N * M over weights W as defined below: W M × N = w 01 , w 02 , … , w 0 N − 1 , w 11 , w 12 , … , w 1 N − 1 , … , w M − 1 1 , w M − 1 2 , … , w M − 1 N − 1

[0047] By contrast, the methods described herein use factorization and reduced weight density to reduce the number of weights and thus multiplications necessary for pointwise convolution. Weight density may generally refer to the percentage of weighted connections or edges in a network out of all of the connections or edges in the network. In conventional pointwise convolution, the weight density is 100% because all connections in the factorized computation network are weighted. By contrast, the factorization and reduction in weight density described herein results in a required number of parameters and complexity of N * log 2 N + N, which is a significant improvement over N * M for both complexity and the required number of parameters (as depicted in FIGS. 7A and 7B).

[0048] Generally, the higher weight density for a given network, the more trainable weights there are for that network, the more expressive the resulting model is based on that network. However, as described further herein, weight density may be varied to strike the right balance between trained model performance and training complexity.

[0049] The fast pointwise convolution methods described herein result in a variety of improvements over conventional methods. For example, reducing the number of multiplications needed to perform pointwise convolution reduces power usage and latency in processing systems performing the pointwise convolution. Further, reducing the number of trainable weights reduces model size, which beneficially reduces memory usage (e.g., fewer memory transactions); required memory size (e.g., to hold smaller activations); and latency (because fewer memory transactions). Reducing the size of memories (e.g., on-chip memories) further reduces power usage and cost. Thus, fast pointwise convolution is especially beneficial for mobile devices and resource-constrained edge devices.

[0050] Fast pointwise convolution is beneficial in any machine learning context where some accuracy is tradeable for performance. For example, fast pointwise convolution may be used for performing model compression before releasing "run-time" weights in an actual product. The compressed weights may beneficially reduce the memory and processing requirements of the product.

[0051] FIG. 4 depicts an example of a fast pointwise convolution method in a bottom-up approach on network 400. Network 400 may be referred to as a factorized computation network or graph.

[0052] In this example, the following conventions are used: f ij is the jth downward connection weight in the ith stage; g ij is the jth upward connection weight in the ith stage; d ij is the jth direct connection weight in the ith stage; and all other direct connection weights are equal to 1 (alternatively referred to as "identity"). Direct connection weights may alternatively be referred to as horizontal connection weights, and upward and downward connection weights may alternatively be referred to as cross connection weights. Further, connections may alternatively be referred to as edges.

[0053] Generally, the computation network 400 comprises a plurality of stages (402 and 404) with inter-stage connections (e.g., d ij 's, f ij 's, and g ij 's) that are connected between inputs (e.g., [x 0 , x 1 , x 2 , x 3 ]), addition nodes (e.g., 408), which may also be referred to as accumulator nodes, and outputs (e.g., [y 0 , y 1 , y 2 , y 3 ]). The number of stages n may be determined by log 2 N, where N is the number of inputs (e.g., the number of input channels). Thus, in this example, n = log 2 4 = 2.

[0054] In this example, in each stage of the factorization, the inputs are organized into distinct pairs, which results in the inputs [x 0 , x 1 , x 2 , x 3 ] being organized into pairs [x 0 , x 1 ] and [x 2 , x 3 ] in first stage 402 and [x 0 , x 2 ] and [x 1 , x 3 ] in second stage 404.

[0055] Then, all first stage 402 direct connection weights (d ij ) and cross connection weights (f ij and g ij ) are calculated, which in this example includes d 00 , f 00 , g 00 , d 01 , d 02 , f 01 , g 01 , and d 03 . However, in second stage 404, only the cross connection weights are calculated, which in this example includes f 10 , g 10 , f 11 , and g 11 . The second stage direct connection weights in this example are set equal to 1 (or "identity"), which reduces the weight density of the network 400, and therefore reduces its complexity.

[0056] Note that in this example, because of the 0 index origin for the input and output nodes, the stage index number is one less than its ordinal reference. For example, here the "first" stage has a stage index of 0 and the "second stage" has a stage index of 1. These may alternatively be referred to as Stage 0 and Stage 1, consistent with the indexes.

[0057] Notably, network 400 is fully connected because every input x is connected to every output y, as with network 300 in FIG. 3, but the reduced number of trainable weights reduces the number of necessary multiplications when calculating the outputs based on the fully-connected inputs. Specifically, in this example, direct connection weights 406 in second stage 404 are all equal to 1 and thus do not need to be trained or multiplied. Rather, the weights from first stage 402 can just be carried along direct connection weights 404. Thus, the factorization of network 400 in this example results in the following equations: y 0 = d 00 x 0 + g 00 x 1 + d 02 g 10 x 2 + g 01 g 10 x 3 y 1 = f 00 x 0 + d 01 x 1 + f 01 g 11 x 2 + d 03 g 11 x 3 y 2 = d 00 f 10 x 0 + g 00 f 10 x 1 + d 02 x 2 + g 01 x 3 y 3 = f 00 f 11 x 0 + d 01 f 11 x 1 + f 01 x 2 + d 03 x 3

[0058] As depicted in the set of equations above, there are only 12 parameters for this fast pointwise convolution method, as compared to N * M = 16 parameters that would be necessary in a conventional pointwise convolution (e.g., as shown in the example of FIG. 3). Thus, in this example, the weight density of network 400 is 12 / 16 = 3 / 4 or 75%. Despite the significantly reduced complexity, the performance of the resulting model using this fast pointwise convolution method is very similar to that of conventional pointwise convolution. Thus in contexts where performance is prioritized over absolute performance, fast pointwise convolution is beneficial.

[0059] FIG. 5 depicts an example of a fast pointwise convolution method with a top-down approach on network 500. Computationally, this top-down alternative approach is equivalent to the bottom-up approach described and depicted with respect to FIG. 4. However, implementation considerations may lead to a preference for one approach versus the other.

[0060] Here again, in each stage of the factorization, the inputs are organized into distinct pairs, and the number of stages n is again determined by n = log 2 N, where N is the number of inputs (e.g., input channels). Thus, in this example, the number of stages for factorization is determined as n = log 2 4 = 2. As above, the inputs [x 0 , x 1 , x 2 , x 3 ] may again be organized into pairs [x 0 , x 1 ] and [x 2 , x 3 ] in first stage 502 and [x 0 , x 2 ] and [x 1 , x 3 ] in second stage 504.

[0061] In this example, only the first stage 502 cross connection weights (f ij and g ij ) are calculated, which in this example includes f 00 , f 01 , g 00 , and g 01 . Then, in second stage 504, the direct connection weights (d ij ) and cross connection weights (f ij and g ij ) are calculated, which in this example includes d 10 , f 10 , g 10 , d 11 , d 12 , f 11 , g 11 , and d 13 .

[0062] As with network 400 in FIG. 4, here network 500 is fully connected because every input x is connected to every output y. And here again, the use of variable weighting densities across the multiple stages (502 and 504) allows for a reduced number of trainable weights, and thus a reduced number of multiplications. Specifically, in this example, direct connection weights 506 in first stage 504 are all equal to 1 (or identity) and thus do not need to be trained or multiplied. Thus, the factorization of network 400 in this example results in the following equations: y 0 = d 10 x 0 + g 10 x 1 + g 00 d 10 x 2 + g 01 g 10 x 3 y 1 = f 10 x 0 + d 11 x 1 + g 00 f 10 x 2 + g 01 d 11 x 3 y 2 = f 00 d 12 x 0 + f 01 g 11 x 1 + d 12 x 2 + g 11 x 3 y 3 = f 00 f 11 x 0 + f 01 d 13 x 1 + f 11 x 2 + d 13 x 3

[0063] Here again, there are only 12 parameters in the set of equations above for this fast pointwise convolution method, as compared to N * M = 16 parameters that would be necessary in a conventional pointwise convolution (e.g., as shown in the example of FIG. 3). Thus, here again, the weight density of network 500 is 3 / 4 or 75%.Generalized Mathematical Model for Fast Pointwise Convolution

[0064] From the examples described with respect to FIGS. 3 and 4, a generalized mathematical model for fast pointwise convolution can be described as follows.

[0065] Starting with a simple case, the direct connection weights for N = 2 inputs (D N ) are represented as follows: D 2 ≜ d 1 1 d

[0066] Similarly, the downward cross connection weights for N = 2 inputs (F N ) are represented as follows: F 2 ≜ 1 1 f 1

[0067] And finally, the upward cross connection weights for N = 2 inputs (G n ) are represented as follows: G 2 ≜ 1 g 1 1

[0068] Thus, for M = 2 outputs, the following expression is used: Y 2 = D 2 ⊙ F 2 ⊙ G 2 X 2

[0069] In the above equation, ⊙ represents an elementwise multiplication operation.

[0070] The above equations for a simple case of N = 2 can be generalized for any N inputs, where N = 2 n< (or equivalently, n = log 2 N) and n is a positive integer, as follows.

[0071] First, let ⊗ represent a Kronecker multiplication operation. Then, going from N = 2 n< → 2 n+1< , the weight matrices are represented by the following equations: D 2 n + 1 ≜ D 2 ⊗ D 2 n = dD 2 n D 2 n D 2 n dD 2 n

[0072] Similarly, the downward cross connection weights are represented as follows: F 2 n + 1 ≜ F 2 ⊗ F 2 n = F 2 n F 2 n fF 2 n F 2 n

[0073] And finally, the upward cross connection weights are represented as follows: G 2 n + 1 ≜ G 2 ⊗ G 2 n = G 2 n gG 2 n G 2 n G 2 n

[0074] Thus, the following generalized expression: Y N = D N ⊙ F N ⊙ G N X N

[0075] So, for example, in another example where N = 4 inputs and M = 4 outputs, the following expression is used: Y 4 = D 4 ⊙ F 4 ⊙ G 4 X 4

[0076] And, the following equations may be derived where N = 4 inputs and M = 4 outputs: D 4 = d d 1 1 d d 1 1 d d 1 1 d d d 1 1 d = dd d d 1 d dd 1 d d 1 dd d 1 d d dd F 4 = 1 1 1 1 f 1 f 1 f f 1 1 ff f f 1 G 4 = 1 g g gg 1 1 g g 1 g 1 g 1 1 1 1

[0077] Notably, the above equations for N = 4 are just one example, and the same derivation can be made incrementally over stages for networks with various numbers of input nodes N.

[0078] In one example, the weight density of a computation graph may be reduced by setting certain direct connection weights to 1, such as described above with respect to FIGS. 4 and 5. So, for example, a modified version of the direct connection weight matrix for the example of N = 4, D 4 * , may be configured where one or more of the indicated d''s are set to 1: D 4 * = d ′ d d ′ d 1 d ′ d ′ d 1 d d 1 d ′ d d ′ 1 d d ′ d ′ d

[0079] Notably, this is just one example, and many others are possible.Fast Pointwise Convolution Linear Scaling Method

[0080] Because, as above, inputs are organized into pairs when a factorized computation network is created, the number of inputs N is preferably a number of 2 n< , where n is a positive integer. When N is not a number of 2 n< , n may be derived as n = Ceil[log 2 (N)], where Ceil[] is a ceiling function that rounds a real number towards positive infinity. Having such definition for n guarantees that a computation graph is always equal to or larger than the size needed for any arbitrary N as a positive integer.

[0081] Further, if the number of inputs N does not match the number of outputs M, then the factorization scheme described above is not directly applicable. In either case, linear scaling with null-activation skipping may be applied to the fast pointwise convolution computation network so that fast pointwise convolution will work as described above.

[0082] The following is an example algorithm for applying linear scaling with null activation skilling. Initially, assume N and M as the number of channels for input and output features, respectively. Then, the following algorithm may be applied to fast pointwise convolution (FPC): 1: Set numStages = int(ceil(log 2 (max(N, M)))) (sets numStages to the smallest integer value that is bigger than or equal to the log 2 of the larger of N and M); 2: Set N new = M new = 2 numStages< ; (Note that N new ≥ N and M new ≥ M) 3: Append N new - N zeros to the input tensor along the dimension of channel; 4: Perform FPC(N new , M new ) instead of FPC(N, M), but skip all multiplication operations that have a "0" value as the input activation; and 5: At the output, return the first M channels of output in tensor along the dimension of channel.

[0083] Assume the computation complexity (or number of parameters) for FPC(N new , M new ) is C. Then the resultant new complexity (or number of parameters) using this proposed linear scaling method becomes CNM / (N new M new ).

[0084] This linear scaling method with null-activation skipping may be implemented, for example, on an ASIC neural network accelerator or other programmable core with specialized instruction / accelerator extensions.Fast Pointwise Convolution with Skip Connections

[0085] Conventional pointwise convolution does not utilize "skip connections", which are connections that skip a layer and may thus also be referred to as "short cut" connections. A motivation for skipping over layers is to simplify the network, using fewer layers in the initial training stages. This speeds learning by reducing the impact of vanishing gradients, as there are fewer layers to propagate through. The network then gradually restores the skipped layers as it learns the feature space.

[0086] In the examples discussed with respect to FIGS. 4 and 5 there are no "skip connections". However, these examples may be adapted to use skip connections (i.e., in conjunction with fast pointwise convolution). Generic forms of the fast pointwise convolution skip connections can be expressed in the Y = WX form defined as follows.

[0087] With respect to the bottom-up fast pointwise convolution example in FIG. 4: y 0 = d 00 + I 0 x 0 + g 00 x 1 + d 02 g 10 x 2 + g 01 g 10 x 3 y 1 = f 00 x 0 + d 01 + I 1 x 1 + f 01 g 11 x 2 + d 03 g 11 x 3 y 2 = d 00 f 10 x 0 + g 00 f 10 x 1 + d 02 + I 2 x 2 + g 01 x 3 y 3 = f 00 f 11 x 0 + d 01 f 11 x 1 + f 01 x 2 + d 03 + I 3 x 3

[0088] And with respect to the top-down fast pointwise convolution example in FIG. 5: y 0 = d 10 + I 0 x 0 + g 10 x 1 + g 00 d 10 x 2 + g 01 g 10 x 3 y 1 = f 10 x 0 + d 11 + I 1 x 1 + g 00 f 10 x 2 + g 01 d 11 x 3 y 2 = f 00 d 12 x 0 + f 01 g 11 x 1 + d 12 + I 2 x 2 + g 11 x 3 y 3 = f 00 f 11 x 0 + f 01 d 13 x 1 + f 11 x 2 + d 13 + I 3 x 3

[0089] While I 0 = I 1 = I 2 = I 3 are individually configurable, simulation has shown I 0 = I 1 = I 2 = I 3 = 1 to be effective. For example, applying skip connections equal to 1 in conjunction with fast pointwise convolution to achieve equal or better accuracy performance at the reduced computation complexity and using the reduced number of weights. Other choices of these values are possible.Fast Pointwise Convolution with Variable Stage Weighting

[0090] FIG. 6 depicts an example of fast pointwise convolution utilizing a form of variable weight density reduction in which stages are entirely weighted or are entirely not weighted. This stage-based weighting reduction may be referred to as structured variable weight density reduction. With structured variable weight density reduction, the weight density may be calculated more simply as the percentage of stages that are weighted out of all of the stages in the factorized computation network.

[0091] In the example depicted in FIG. 6, N = 8, which results in n = log 2 N = 3 stages. Further, in this example, the weight density is 2 / 3 because the first stage 602 and the third stage 606 are weighted out of the total of three stages (602, 604, and 606), which means that 2 / 3 of the possible connection weights include a weight other than 1 (i.e., identity). By comparison, in FIGS. 4 and 5, the weight density was 12 / 16 = 3 / 4 = 75%.

[0092] Further, in this example: Stage 0 is a "weighting stage" where weights are calculated; Stage 1 is a "non-weighting stage" where no weights are not calculated (i.e., no direct or cross connection weights); and Stage 2 is another weighting stage. Notably, this is another example of a bottom-up approach where the first stage (Stage 0) includes trainable direct connection weights (d ij 's) and cross connection weights (f ij 's and g ij 's) and the final stage (Stage 2) includes only trainable cross connection weights.

[0093] The factorization of network 600 in this example results in the following equations: y 0 = d 00 d 20 x 0 + g 00 d 20 x 1 + d 02 d 20 x 2 + g 01 d 20 x 3 + d 04 g 20 x 4 + g 02 g 20 x 5 + d 06 g 20 x 6 + g 03 g 20 x 7 y 1 = f 00 d 21 x 0 + d 01 d 21 x 1 + f 01 d 21 x 2 + d 03 d 21 x 3 + f 02 g 21 x 4 + d 05 g 21 x 5 + f 03 g 21 x 6 + d 07 g 21 x 7 y 2 = d 00 d 22 x 0 + g 00 d 22 x 1 + d 02 d 22 x 2 + g 01 d 22 x 3 + d 04 g 22 x 4 + g 02 g 22 x 5 + d 06 g 22 x 6 + g 03 g 22 x 7 y 3 = f 00 d 23 x 0 + d 01 d 23 x 1 + f 01 d 23 x 2 + d 03 d 23 x 3 + f 02 g 23 x 4 + d 05 g 23 x 5 + f 03 g 23 x 6 + d 07 g 23 x 7 y 4 = d 00 f 20 x 0 + g 00 f 20 x 1 + d 02 f 20 x 2 + g 01 f 20 x 3 + d 04 d 24 x 4 + g 02 d 24 x 5 + d 06 d 24 x 6 + g 03 d 24 x 7 y 5 = f 00 f 21 x 0 + d 01 f 21 x 1 + f 01 f 21 x 2 + d 03 f 21 x 3 + f 02 d 25 x 4 + d 05 d 25 x 5 + f 03 d 25 x 6 + d 07 d 25 x 7 y 6 = f 00 f 22 x 0 + g 00 f 22 x 1 + d 02 f 22 x 2 + g 01 f 22 x 3 + d 04 d 26 x 4 + g 02 d 26 x 5 + d 06 d 26 x 6 + g 03 d 26 x 7 y 7 = f 00 f 23 x 0 + d 01 f 23 x 1 + f 01 f 23 x 2 + d 03 f 23 x 3 + f 02 d 27 x 4 + d 05 d 27 x 5 + f 03 d 27 x 6 + d 07 d 27 x 7

[0094] In some cases, a sensitivity analysis may be performed in order to determine the impact (e.g., in terms of trained model performance) of making a particular stage a non-weighting stages or a weighting stage.Example Performance Improvements using Fast Pointwise Convolution

[0095] FIGS. 7A and 7B depict example computations showing the performance improvement in terms of both complexity and number of parameters of fast pointwise convolution over conventional pointwise convolution and butterfly transformation pointwise convolution.

[0096] It is clear from the table in FIG. 7A that there is a dramatic reduction in the number of parameters (e.g., weights) for an N → N pointwise convolution when comparing fast pointwise convolution to conventional pointwise convolution and even to improved performance pointwise convolution, such as by butterfly transformation pointwise convolution. As discussed above, this reduction in parameters directly translates to performance improvements of the processing system performing the pointwise convolution. Further as described above, the reduction in parameter count means that the hardware requirements for supporting models relying on pointwise convolution may be beneficially reduced, which further saves chip area, power usage, and cost, to name just a few benefits.

[0097] Similarly, the graph depicted in FIG. 7B depicts the complexity reduction of fast pointwise convolution, as described herein, versus conventional pointwise convolution and butterfly transformation pointwise convolution, in terms of the number of multiply-accumulate operations. The reduction in multiply-accumulate operations beneficially reduces latency and power consumption in a processing system performing the pointwise convolution.Example Method for Performing Fast Pointwise Convolution

[0098] FIG. 8 depicts an example method 800 for performing fast pointwise convolution.

[0099] Method 800 begins at step 802 with receiving input data at a convolutional neural network (CNN) model.

[0100] Method 800 then proceeds to step 804 with generating a factorized computation network comprising a plurality of connections between a first layer of the CNN model and a second layer of the CNN model. The factorized computation network comprises N inputs. The factorized computation network comprises M outputs. The factorized computation network comprises at least one path from every input of the N inputs to every output of the M outputs.

[0101] Method 800 then proceeds to step 806 with setting a connection weight for a plurality of connections in the factorized computation network to 1 so that a weight density for the factorized computation network is < 100%.

[0102] Method 800 then proceeds to step 808 with performing fast pointwise convolution using the factorized computation network to generate fast pointwise convolution output.

[0103] Method 800 then proceeds to step 810 with providing the fast pointwise convolution output to the second layer of the CNN model.

[0104] The factorized computation network comprises a number of stages n = log 2 N.

[0105] In some embodiments of method 800, the plurality of connections comprise a plurality of weighting connections connected to addition nodes between the first layer of the CNN model and the second layer of the CNN model, such as described above with respect to FIGS. 4-6 in one embodiment.

[0106] In some embodiments, setting the connection weight for the plurality of connections in the factorized computation network to 1 comprises setting all connection weights for at least one stage of the number of stages to 1, such as described above with respect to FIG. 6. In some embodiments, setting the connection weight for the plurality of connections in the factorized computation network to 1 comprises setting a subset of connection weights for at least one stage of the number of stages to 1, such as described above with respect to FIGS. 4 and 5.

[0107] In some embodiments, method 800 further comprises: performing linear scaling with null activation skipping to the factorized computation network, such as when N ≠ 2 n< and n is a positive integer, or when or N ≠ M.

[0108] In some embodiments, such as those performing linear scaling with null activation, providing the fast pointwise convolution output to the second layer of the CNN model comprises providing only a first M channels of the fast pointwise convolution output to the second layer of the CNN model.

[0109] In some embodiments, method 800 further comprises: adding a plurality of skip connection to the factorized computation network.

[0110] In some embodiments, method 800 further comprises: generating a model output from the CNN model; and providing the generated model output to a low-power application running on a mobile electronic device.

[0111] In some embodiments, method 800 further comprises performing, by a first processor, a first inference based on the fast pointwise convolution output; performing, by a second processor, a second fast pointwise convolution using the factorized computation network to generate second fast pointwise convolution output; and providing the second fast pointwise convolution output to the second layer of the CNN model. In other words, fast pointwise convolution maybe be parallelized across hardware resources in multi-processor processing systems. For example, the first processor and second processor may be one of the processors described below with respect to FIGS. 9 and / or 10.

[0112] In some embodiments of method 800, the input data is received from a sensor in a mobile electronic device (e.g., one of sensors 914 in FIG. 9). In some embodiments, the input data is associated with an application running on the mobile electronic device and configured for at least one of: biometric feature detection, human presence detection, environmental condition detection, object detection, or object classification.

[0113] The input data is an image. In some embodiments of method 800, not forming part of the invention, the input data is one of image data, video data, audio data, or signal data. In other embodiments, not forming part of the invention, input data may represent other "features" that have been quantified.Example Processing Systems for Optimizing Machine Learning Model Performance

[0114] FIG. 9 illustrates an example implementation of a system-on-a-chip (SOC) 900, which may include a central processing unit (CPU) 902 or a multi-core CPU configured to perform a parallel Monte Carlo dropout function, in accordance with certain aspects of the present disclosure. Variables (e.g., neural signals and synaptic weights), system parameters associated with a computational device (e.g., neural network with weights), delays, frequency bin information, and task information may be stored in a memory block associated with a neural processing unit (NPU) 908, in a memory block associated with a CPU 902, in a memory block associated with a graphics processing unit (GPU) 904, in a memory block associated with a digital signal processor (DSP) 906, in a memory block 918, or may be distributed across multiple blocks. Instructions executed at the CPU 902 may be loaded from a program memory associated with the CPU 902 or may be loaded from a memory block 918.

[0115] The SOC 900 may also include additional processing blocks tailored to specific functions, such as a GPU 904, a DSP 906, a connectivity block 910, which may include fifth generation (5G) connectivity, fourth generation long term evolution (4G LTE) connectivity, Wi-Fi connectivity, USB connectivity, Bluetooth connectivity, and the like, and a multimedia processor 912 that may, for example, detect and recognize gestures. In one implementation, the NPU is implemented in the CPU 902, DSP 906, and / or GPU 904. The SOC 900 may also include one or more sensor processors 914 associated with any manner of sensor, one or more image signal processors (ISPs) 916, and / or a navigation module 920, which may include a global positioning system.

[0116] The SOC 900 may be based on an ARM instruction set. In an aspect of the present disclosure, the instructions loaded into the CPU 902 may comprise code to search for a stored multiplication result in a lookup table (LUT) corresponding to a multiplication product of an input value and a filter weight. The instructions loaded into the CPU 902 may also comprise code to disable a multiplier during a multiplication operation of the multiplication product when a lookup table hit of the multiplication product is detected. In addition, the instructions loaded into the CPU 902 may comprise code to store a computed multiplication product of the input value and the filter weight when a lookup table miss of the multiplication product is detected.

[0117] SOC 900 and / or components thereof may be configured to perform the methods described herein.

[0118] FIG. 10 depicts an example schematic diagram of a multi-processor processing system 1000 that may be implemented with embodiments described herein.

[0119] In this example, system 1000 includes processors 1001, 1003, and 1005, but in other examples, any number of individual processors may be used. Further, though depicted similarly, processors 1001, 1003, and 1005 may be representative of various different kinds of processors in an electronic device, such as CPUs, GPUs, DSPs, NPUs, and the like as described herein.

[0120] Each of processors 1001, 1003, and 1005 includes an instruction scheduler, various hardware sub-components (e.g., hardware X, hardware Y, and hardware Z), and a local memory. The local memory may be a tightly coupled memory (TCM) in some embodiments. Note that the components of each of processors 1001, 1003, and 1005 is shown as the same in this example, each of the processors may have different hardware configurations, different hardware elements, etc.

[0121] Each of processors 1001, 1003, and 1005 is also in data communication with a global memory, such as a DDR memory, or other types of volatile working memory.

[0122] In some implementations, in a multi-processor processing system such as 1000, one of the processors may act as a master processor. For example, processor 1001 may be a master processor in this example. A master processor may include a compiler that, when executed, can determine how a model, such as a neural network model, will be processed by various components of processing system 1000. For example, hardware parallelism may be implemented by mapping portions of the processing of a model to various hardware (e.g., hardware X, hardware Y, and hardware Z) within a given processor (e.g., processor 1001) as well as mapping portions of the processing of the model to other processors (e.g., processors 1003 and 1005) and their associated hardware.

Claims

1. A computer implemented method, comprising: receiving (802) input data at a convolutional neural network, CNN, model, wherein the input data is an image; generating (804) a factorized computation network comprising a plurality of connections between a first layer of the CNN model and a second layer of the CNN model, wherein: the factorized computation network comprises N inputs, the factorized computation network comprises M outputs, and the factorized computation network comprises at least one path from every input of the N inputs to every output of the M outputs; wherein connection weights for the plurality of connections comprise direct connection weight matrices DN, downward cross connection weight matrices FN and upward cross connection weight matrices GN, the connection weight matrices being defined iteratively for N = 2n, n being a positive integer, wherein the weight matrices are defined as a combination of contributions from n stages, with the iteration N = 2n → 2n+1: D 2 n + 1 ≜ dD 2 n D 2 n D 2 n dD 2 n , F 2 n + 1 ≜ F 2 n F 2 n fF 2 n F 2 n , G 2 n + 1 ≜ G 2 n gG 2 n G 2 n G 2 n ; wherein: D 2 ≜ d 1 1 d , F 2 ≜ 1 1 f 1 , G 2 ≜ 1 g 1 1 performing (808) fast pointwise convolution, that is performing pointwise convolution using the factorized computation network, to generate fast pointwise convolution output; wherein the relationship between the M outputs, represented by YN, and the N inputs, represented by XN is represented by the expression: Y N = D N ⊙ F N ⊙ G N X N and providing (810) the fast pointwise convolution output to the second layer of the CNN model, wherein the CNN model is configured to extract features from input images.

2. The method of Claim 1, further comprising: setting (806) a connection weight for a plurality of connections in the factorized computation network to 1 so that a weight density for the factorized computation network is < 100%.

3. The method of Claim 1, wherein: the plurality of connections comprise a plurality of weighting connections connected to addition nodes (408, 508) between the first layer of the CNN model and the second layer of the CNN model.

4. The method of Claim 2, wherein setting the connection weight for the plurality of connections in the factorized computation network to 1 comprises setting all connection weights for at least one stage of the number of stage to 1.

5. The method of Claim 2, wherein setting the connection weight for the plurality of connections in the factorized computation network to 1 comprises setting a subset of connection weights for at least one stage of the number of stages to 1.

6. The method of Claim 1, wherein: prior to any linear scaling N ≠ 2n or N ≠ M, and the method further comprises: performing linear scaling with null activation skipping on the factorized computation network, and providing the fast pointwise convolution output to the second layer of the CNN model comprises providing only a first M channels of the fast pointwise convolution output to the second layer of the CNN model.

7. The method of Claim 1, further comprising: adding a plurality of skip connection to the factorized computation network.

8. The method of Claim 1, further comprising: generating a model output from the CNN model; and providing the model output to an application running on a mobile electronic device.

9. The method of Claim 1, further comprising: performing, by a first processor, a first inference based on the fast pointwise convolution output; performing, by a second processor, a second fast pointwise convolution using the factorized computation network to generate second fast pointwise convolution output; and providing the second fast pointwise convolution output to the second layer of the CNN model.

10. A processing system, comprising: a memory (918) comprising computer-executable instructions; and a first processor (902, 1001, 1003, 1005) configured to execute the computer-executable instructions and cause the processing system to implement the method according to any one of claims 1-8.

11. A non-transitory computer-readable medium comprising instructions that, when executed by a first processor (902, 1001, 1003, 1005) of a processing system, cause the processing system to perform the method according to any one of claims 1-8.