Methods and systems for tensor network contraction based on local optimization of contraction tree

A system comprising:

• one or more processors, and
• memory communicatively coupled to the one or more processors, the memory storing computer-executable modules executable by the one or more processors that, when executed by the one or more processors, perform actions including:
• obtaining a contraction tree associated with a tensor network, wherein a plurality of vertices and edges of the contraction tree correspond to a set of tensor nodes and indices of the tensor network, respectively;
• iteratively performing operations until a termination condition is satisfied, the operations including:
• selecting a sub-graph of the contraction tree;
• replacing the sub-graph with a local optimal sub-graph;
• and
• obtaining an optimized contraction tree including the local optimal sub-graph;
• and
• outputting the optimized contraction tree.

A computer-readable storage medium storing computer-readable instructions executable by one or more processors of a video compression system, that when executed by the one or more processors, cause the one or more processors to perform actions comprising:

• obtaining a contraction tree associated with a tensor network, wherein a plurality of vertices and edges of the contraction tree correspond to a set of tensor nodes and indices of the tensor network, respectively;
• iteratively performing operations until a termination condition is satisfied, the operations including:
• selecting a sub-graph of the contraction tree;
• replacing the sub-graph with a local optimal sub-graph;
• and
• obtaining an optimized contraction tree including the local optimal sub-graph;
• and
• outputting the optimized contraction tree.