Alibaba Group Holding Limited
OPTIMIZATION PROCESSING FOR NEURAL NETWORK MODEL

Abstract:

The present disclosure provides a method and a device for optimization processing of neural network models. The method includes the following: determining one or more target layers of the neural network model based on the number of neurons at each layer of the neural network model; for each of the one or more target layers, adding a virtual layer between the target layer and a preceding layer of the target layer, where neurons at the virtual layer are separately connected to neurons at the target layer and neurons at the preceding layer of the target layer, and addition of the virtual layer reduces the number of connections between the target layer and the preceding layer of the target layer; and training the neural network model after having added the virtual layers, to obtain an optimized neural network model.