Sughrue Mion, PLLC
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.
TENCENT AMERICA LLC
G H H H G G H
G06N H04N H04N H04N G06N G06N H04N
inventional inventional inventional inventional inventional inventional inventional
Wei Wei Shan
Jiang Wang Liu