SearcharxivSearch

arXiv subjects

Si-Jing Du

Publications and source records attributed to Si-Jing Du.

5 recordsLinked to original sources

Vectorized Symmetric and Fermionic Tensor Network Implementations for GPU-Accelerated Variational Monte Carlo

Variational Monte Carlo (VMC) calculations based on tensor networks (TN) have recently achieved competitive accuracy in ground-state calculations of strongly correlated spin and fermionic systems. However, existing tensor network VMC (TN-VMC) algorithms have not been formulated in a manner that can fully utilize GPU acceleration. We tackle the key missing ingredient, namely, the vectorized evaluation of tensor network amplitudes and tensor network operations. This ensures high GPU utilization by batching over computations with identical structure. In particular, we show how to achieve vectorization for the practically relevant case of abelian symmetric tensor networks (which includes fermionic tensor networks) by developing a ``flat'' tensor network formalism for block-sparse tensor representation and contraction. Using this, we construct a GPU-adapted symmetric TN-VMC workflow with batched tensor network computation. In the two-dimensional Fermi--Hubbard model, we demonstrate a GPU speedup of up to $300 \times$ over single core CPU implementations, for fermionic TN variational wavefunctions.

physics.comp-ph

Statistical mechanics in continuous space with tensor network methods

Tensor network (TN) methods are well established for computing partition functions in statistical mechanics, though this use has traditionally been limited to lattice models. We extend the scope of TN methodology to interacting particle systems in continuous space. Through a real-space discretization combined with a cell-based coarse-graining scheme, we formulate an effective lattice model that explicitly preserves spatial locality. The partition function of this model is represented as a TN, and the thermodynamic quantities are computed via boundary contraction. We apply this framework to the two-dimensional hard-disk problem and demonstrate the strengths of the TN formulation compared to existing Monte Carlo simulations.

cond-mat.stat-mech

Neuralized Fermionic Tensor Networks for Quantum Many-Body Systems

We describe a class of neuralized fermionic tensor network states (NN-fTNS) that introduce non-linearity into fermionic tensor networks through configuration-dependent neural network transformations of the local tensors. The construction uses the fTNS algebra to implement a natural fermionic sign structure and is compatible with standard tensor network algorithms, but gains enhanced expressivity through the neural network parametrization. Using the 1D and 2D Fermi-Hubbard models as benchmarks, we demonstrate that NN-fTNS achieve order of magnitude improvements in the ground-state energy compared to pure fTNS with the same bond dimension, and can be systematically improved through both the tensor network bond dimension and the neural network parametrization. Compared to existing fermionic neural quantum states (NQS) based on Slater determinants and Pfaffians, NN-fTNS offer a physically motivated alternative fermionic structure. Furthermore, compared to such states, NN-fTNS naturally exhibit improved computational scaling and we demonstrate a construction that achieves linear scaling with the lattice size.

cond-mat.dis-nn

Tensor Network Computations That Capture Strict Variationality, Volume Law Behavior, and the Efficient Representation of Neural Network States

We introduce a change of perspective on tensor network states that is defined by the computational graph of the contraction of an amplitude. The resulting class of states, which we refer to as tensor network functions, inherit the conceptual advantages of tensor network states while removing computational restrictions arising from the need to converge approximate contractions. We use tensor network functions to compute strict variational estimates of the energy on loopy graphs, analyze their expressive power for ground-states, show that we can capture aspects of volume law time evolution, and provide a mapping of general feed-forward neural nets onto efficient tensor network functions. Our work expands the realm of computable tensor networks to ones where accurate contraction methods are not available, and opens up new avenues to use tensor networks.

quant-ph

Ground-state properties via machine learning quantum constraints

Ground-state properties are central to our understanding of quantum many-body systems. At first glance, it seems natural and essential to obtain the ground state before analyzing its properties; however, its exponentially large Hilbert space has made such studies costly, if not prohibitive, on sufficiently large system sizes. Here, we propose an alternative strategy based upon the expectation values of an ensemble of operators and the elusive yet vital quantum constraints between them, where the search for ground-state properties simply equates to classical constrained minimization. These quantum constraints are generally obtainable via sampling and then machine learning on a large number of systematically consistent quantum many-body states. We showcase our perspective on 1D fermion chains and spin chains for applicability, effectiveness, caveats, and unique advantages, especially for strongly correlated systems, thermodynamic-limit systems, property designs, etc.

cond-mat.str-el