arXiv subjects

Ankit Mahajan

Publications and source records attributed to Ankit Mahajan.

26 records · Page 2Linked to original sources

arXiv2022

ipie: A Python-based Auxiliary-Field Quantum Monte Carlo Program with Flexibility and Efficiency on CPUs and GPUs

We report the development of a python-based auxiliary-field quantum Monte Carlo (AFQMC) program, ipie, with preliminary timing benchmarks and new AFQMC results on the isomerization of [Cu$_2$O$_2$$]^{2+}$. We demonstrate how implementations for both central and graphical processing units (CPUs and GPUs) are achieved in ipie. We show an interface of ipie with PySCF as well as a straightforward template for adding new estimators to ipie. Our timing benchmarks against other C++ codes, QMCPACK and Dice, suggest that ipie is faster or similarly performing for all chemical systems considered on both CPUs and GPUs. Our results on [Cu$_2$O$_2$$]^{2+}$ using selected configuration interaction trials show that it is possible to converge the ph-AFQMC isomerization energy between bis($μ$-oxo) and $μ$-$η^2$:$η^2$ peroxo configurations to the exact known results for small basis sets with $10^5$ to $10^6$ determinants. We also report the isomerization energy with a quadruple-zeta basis set with an estimated error less than a kcal/mol, which involved 52 electrons and 290 orbitals with $10^6$ determinants in the trial wavefunction. These results highlight the utility of ph-AFQMC and ipie for systems with modest strong correlation and large-scale dynamic correlation.

physics.chem-ph↗

arXiv2022

Selected configuration interaction wave functions in phaseless auxiliary field quantum Monte Carlo

We present efficient algorithms for using selected configuration interaction (sCI) trial wave functions in phaseless auxiliary field quantum Monte Carlo (ph-AFQMC). These advancements, geared towards optimizing computational performance for longer CI expansions, allow us to use up to a million configurations feasibly in the trial state for ph-AFQMC. In one example, we found the cost of ph-AFQMC per sample to increase only by a factor of about $3$ for a calculation with $10^4$ configurations compared to that with a single one, demonstrating the tiny computational overhead due to a longer expansion. This favorable scaling allows us to study the systematic convergence of the phaseless bias in AFQMC calculations with an increasing number of configurations and provides a means to gauge the accuracy of ph-AFQMC with other trial states. We also show how the scalability issues of sCI trial states for large system sizes could be mitigated by restricting them to a moderately sized orbital active space and leveraging the near-cancellation of out of active space phaseless errors.

physics.chem-ph↗

arXiv2021

Taming the sign problem in auxiliary field quantum Monte Carlo using accurate trial wave functions

We explore different ways of incorporating accurate trial wave functions into free projection auxiliary field quantum Monte Carlo (fp-AFQMC). Trial states employed include coupled cluster singles and doubles, multi-Slater, and symmetry projected mean-field wave functions. We adapt a recently proposed fast multi-Slater local energy evaluation algorithm for fp-AFQMC, making the use of long expansions from selected configuration interaction methods feasible. We demonstrate how these trial wave functions serve to mitigate the sign problem and accelerate convergence in quantum chemical problems, allowing the application of fp-AFQMC to systems of substantial sizes. Our calculations on the widely studied $\text{Cu}_2\text{O}_2^{2+}$ model system show that many previously reported isomerization energies differ substantially from the near-exact fp-AFQMC value.

physics.chem-ph↗

arXiv2020

Efficient local energy evaluation for multi-Slater wave functions in orbital space quantum Monte Carlo

Recent developments in selected configuration interaction methods have led to increased interest in using multi-Slater trial wave functions in various quantum Monte Carlo (QMC) methods. Here we present an algorithm for calculating the local energy of a multi-Slater wave function in orbital space QMC. For an ab initio Hamiltonian, our algorithm has a cost scaling of $O(n^5 + n_c)$, as opposed to the $O(n^4n_c)$ scaling of existing orbital space algorithms, where $n$ is the system size, and $n_c$ is the number of configurations in the wave function. We present our method using variational Monte Carlo calculations with the Jastrow multi-Slater wave function, although the formalism should be applicable for auxiliary field quantum Monte Carlo. We apply it to polyacetylene and demonstrate the possibility of using a much larger number of configurations than possible using existing methods.

physics.chem-ph↗

arXiv2020

Efficient multireference perturbation theory without high-order reduced density matrices

We present a stochastic approach to perform strongly contracted n-electron valence state perturbation theory (SC-NEVPT), which only requires one- and two-body reduced density matrices, without introducing approximations. We use this method to perform SC-NEVPT2 for CASSCF wave functions obtained from selected configuration interaction, although the approach is applicable to a larger class of wave functions, including those from variational Monte Carlo (VMC). The accuracy of this approach is demonstrated for small test systems, and the scaling is investigated with the number of virtual orbitals and the molecule size. We also find the SC-NEVPT2 energy to be relatively insensitive to the quality of the reference wave function. Finally, the method is applied to the Fe(II)-Porphyrin system with a (32e, 29o) active space, and to the isomerization of Cu2O2 in a (28e, 32o) active space.

physics.chem-ph↗

arXiv2019

Multireference configuration interaction and perturbation theory without reduced density matrices

The computationally expensive evaluation and storage of high-rank reduced density matrices (RDMs) has been the bottleneck in the calculation of dynamic correlation for multireference wave functions in large active spaces. We present a stochastic formulation of multireference configuration interaction (MRCI) and perturbation theory (MRPT) that avoids the need for these expensive RDMs. The algorithm presented here is flexible enough to incorporate a wide variety of active space reference wave functions, including selected configuration interaction, matrix product states, and symmetry-projected Jastrow mean field wave functions. It enjoys the usual attractive features of Monte Carlo methods, such as embarrassing parallelizability and low memory costs. We find that the stochastic algorithm is already competitive with the deterministic algorithm for small active spaces, containing as few as 14 orbitals. We illustrate the utility of our stochastic formulation using benchmark applications.

physics.chem-ph↗

arXiv2019

An accelerated linear method for optimizing non-linear wavefunctions in variational Monte Carlo

Although the linear method is one of the most robust algorithms for optimizing non-linearly parametrized wavefunctions in variational Monte Carlo, it suffers from a memory bottleneck due to the fact at each optimization step a generalized eigenvalue problem is solved in which the Hamiltonian and overlap matrices are stored in memory. Here we demonstrate that by applying the Jacobi-Davidson algorithm, one can solve the generalized eigenvalue problem iteratively without having to build and store the matrices in question. The resulting direct linear method greatly lowers the cost and improves the scaling of the algorithm with respect to the number of parameters. To further improve the efficiency of optimization for wavefunctions with a large number of parameters, we use the first order method AMSGrad far from the minimum as it is very inexpensive, and only switch to the direct linear method near the end of the optimization where methods such as AMSGrad have long convergence tails. We apply this improved optimizer to various wavefunctions with both real and orbital space Jastrow factors for atomic systems such as Beryllium and Neon, molecular systems such as the Carbon dimer and Iron(II) Porphyrin, and model systems such as the Hubbard model and Hydrogen chains.

cond-mat.str-el↗

arXiv2019

Symmetry projected Jastrow mean field wavefunction in variational Monte Carlo

We extend our low-scaling variational Monte Carlo (VMC) algorithm to optimize the symmetry projected Jastrow mean field (SJMF) wavefunctions. These wavefunctions consist of a symmetry-projected product of a Jastrow and a general broken-symmetry mean field reference. Examples include Jastrow antisymmetrized geminal power (JAGP), Jastrow-Pfaffians, and resonating valence bond (RVB) states among others, all of which can be treated with our algorithm. We will demonstrate using benchmark systems including the nitrogen molecule, a chain of hydrogen atoms, and the 2-D Hubbard model that a significant amount of correlation can be obtained by optimizing the energy of the SJMF wavefunction. This can be achieved at a relatively small cost when multiple symmetries including spin, particle number, and complex conjugation are simultaneously broken and projected. We also show that reduced density matrices can be calculated using the optimized wavefunctions, which allows us to calculate other observables such as correlation functions and will enable us to embed the VMC algorithm in a complete active space self-consistent field (CASSCF) calculation.

cond-mat.str-el↗