METHOD AND APPARATUS FOR THREE-DIMENSIONAL (3D)-TREE CODING FOR NEURAL NETWORK MODEL COMPRESSION

摘要

A method of three-dimensional (3D)-Tree coding for neural network model compression, is performed by at least one processor, and includes reshaping a four-dimensional (4D) parameter tensor of a neural network into a 3D parameter tensor of the neural network, the 3D parameter tensor comprising a convolution kernel size, an input feature size, and an output feature size, partitioning the 3D parameter tensor along a plane that is formed by the input feature size and the output feature size into 3D coding tree units (CTU3Ds), partitioning each of the CTU3Ds into a plurality of 3D coding units (CU3Ds) recursively until a predetermined depth, using a quad-tree, constructing a 3D tree for each of the plurality of CU3Ds, and entropy encoding each of a plurality of values of a plurality of nodes of the 3D tree.