arXiv:2103.06872 · Physical Review A
Tensor networks and efficient descriptions of classical data
Sirui Lu, Márton Kanász-Nagy, Ivan Kukuljan, J. I. Cirac
Abstract
We investigate the potential of tensor-network-based machine-learning methods to scale to large image and text datasets. We study how the mutual information between a subregion and its complement scales with the subsystem size L , similarly to approaches used in quantum many-body physics. Using various mutual information (MI) estimation methods, including an autoregressive network-based estimator, we find that simple image datasets exhibit area law scaling, suggesting efficient representation by two-dimensional tensor networks. More complex image datasets exceed the area law, indicating the need for generalized tensor network states or hybrid models. For Wikipedia text data, we observe power-law scaling I ( L ) ∝ L ν with ν ≈ 0.82 , approaching a volume law. This implies that one-dimensional tensor networks with area law entanglement may not efficiently capture the structure of text. We introduce two models to reproduce this scaling: a quantum-inspired random pair toy model and a linguistically motivated Markovian dependency tree model. In the latter model, matching the observed MI scaling allows us to infer the word-word correlation length distribution in text.