TensorNetwork on TensorFlow: A Spin Chain Application Using Tree Tensor Networks
TensorNetwork is an open source library for implementing tensor network algorithms in TensorFlow. We describe a tree tensor network (TTN) algorithm for approximating the ground state of either a periodic quantum spin chain (1D) or a lattice model on a thin torus (2D), and implement the algorithm using TensorNetwork. We use a standard energy minimization procedure over a TTN ansatz with bond dimension $\chi$, with a computational cost that scales as $O(\chi^4)$. Using bond dimension $\chi \in [32,256]$ we compare the use of CPUs with GPUs and observe significant computational speed-ups, up to a factor of $100$, using a GPU and the TensorNetwork library.
Code (3)
Tasks
GPUTensor NetworksSimilar Papers 제목 키워드 기반
TensorNetwork: A Library for Physics and Machine Learning
TensorNetwork is an open source library for implementing tensor network algorithms. Tensor networks are sparse data structures originally designed for simulating quantum many-body physics, but are currently also applied …
BIG-bench Machine LearningTensor NetworksTensorNetwork for Machine Learning
We demonstrate the use of tensor networks for image classification with the TensorNetwork open source library. We explain in detail the encoding of image data into a matrix product state form, and describe how to contrac…
BIG-bench Machine LearningCPUGeneral ClassificationGPU+3Ising Models with Hidden Markov Structure: Applications to Probabilistic Inference in Machine Learning
In this paper, we investigate tree-indexed Markov chains (Gibbs measures) defined by a Hamiltonian that couples two Ising layers: hidden spins \(s(x) \in \{\pm 1\}\) and observed spins \(\sigma(x) \in \{\pm 1\}\) on a Ca…
Anomaly DetectionDenoisingWeakly-supervised LearningSpineNet: Learning Scale-Permuted Backbone for Recognition and Localization
Convolutional neural networks typically encode an input image into a series of intermediate features with decreasing resolutions. While this structure is suited to classification tasks, it does not perform well for tasks…
Image ClassificationInstance SegmentationNeural Architecture Searchobject-detection+3Estimation of the reduced density matrix and entanglement entropies using autoregressive networks
We present an application of autoregressive neural networks to Monte Carlo simulations of quantum spin chains using the correspondence with classical two-dimensional spin systems. We use a hierarchy of neural networks ca…