Source author record

Wanrun Jiang

Wanrun Jiang appears in the imported research catalog. Authorship, coauthor and topic links are available while profile ownership is still unclaimed.

ResearcherUnclaimed source record

physics.chem-ph physics.atm-clus Distributed, Parallel, and Cluster Computing physics.comp-ph

Catalog footprint

What is connected

4works

4topics

4close collaborators

Actions

Connect this record

Open graph Browse works

Inspect adjacent papers, topics, institutions and collaborators without losing the researcher page.

Building this map preview

BZPEER is loading the nearby papers, people, topics and institutions for this page.

preprint2022arXiv

DP Compress: a Model Compression Scheme for Generating Efficient Deep Potential Models

Machine-learning-based interatomic potential energy surface (PES) models are revolutionizing the field of molecular modeling. However, although much faster than electronic structure schemes, these models suffer from costly computations via deep neural networks to predict the energy and atomic forces, resulting in lower running efficiency as compared to the typical empirical force fields. Herein, we report a model compression scheme for boosting the performance of the Deep Potential (DP) model, a deep learning based PES model. This scheme, we call DP Compress, is an efficient post-processing step after the training of DP models (DP Train). DP Compress combines several DP-specific compression techniques, which typically speed up DP-based molecular dynamics simulations by an order of magnitude faster, and consume an order of magnitude less memory. We demonstrate that DP Compress is sufficiently accurate by testing a variety of physical properties of Cu, H2O, and Al-Cu-Mg systems. DP Compress applies to both CPU and GPU machines and is publicly available online.

preprint2022arXiv

Extending the limit of molecular dynamics with ab initio accuracy to 10 billion atoms

High-performance computing, together with a neural network model trained from data generated with first-principles methods, has greatly boosted applications of \textit{ab initio} molecular dynamics in terms of spatial and temporal scales on modern supercomputers. Previous state-of-the-art can achieve $1-2$ nanoseconds molecular dynamics simulation per day for 100-million atoms on the entire Summit supercomputer. In this paper, we have significantly reduced the memory footprint and computational time by a comprehensive approach with both algorithmic and system innovations. The neural network model is compressed by model tabulation, kernel fusion, and redundancy removal. Then optimizations such as acceleration of customized kernel, tabulation of activation function, MPI+OpenMP parallelization are implemented on GPU and ARM architectures. Testing results of the copper system show that the optimized code can scale up to the entire machine of both Fugaku and Summit, and the corresponding system size can be extended by a factor of $134$ to an unprecedented $17$ billion atoms. The strong scaling of a $13.5$-million atom copper system shows that the time-to-solution can be 7 times faster, reaching $11.2$ nanoseconds per day. This work opens the door for unprecedentedly large-scale molecular dynamics simulations based on {\it ab initio} accuracy and can be potentially utilized in studying more realistic applications such as mechanical properties of metals, semiconductor devices, batteries, etc. The optimization techniques detailed in this paper also provide insight for relevant high-performance computing applications.

preprint2016arXiv

Electronic structure contribution to hydrogen bonding interaction of a water dimer

Hydrogen bond (H-bond) covalency has recently been observed in ice and liquid water, while the penetrating molecular orbitals (MOs) in the H-bond region of most typical water dimer system, (H2O)2, have also been discovered. However, obtaining the quantitative contribution of these MOs to the H-bond interaction is still problematic. In this work, we introduced the orbital-resolved electron density projected integral (EDPI) along the H-bond to approach this problem. The calculations show that, surprisingly, the electronic occupied orbital (HOMO-4) of (H2O)2 accounts for about 40% of the electron density at the bond critical point. Moreover, the charge transfer analysis visualizes the electron accumulating effect of the orbital interaction within the H-bond between water molecules, supporting its covalent-like character. Our work expands the classical understanding of H-bond with specific contributions from certain MOs, and will also advance further research into such covalency and offer quantitative electronic structure insights into intermolecular systems.

preprint2015arXiv

Covalent features in the hydrogen bond of a water dimer: molecular orbital analysis

The covalent-like characteristics of hydrogen bonds offer a new perspective on intermolecular interactions. Here, using density functional theory and post-Hartree-Fock methods, we reveal that there are two bonding molecular orbitals (MOs) crossing the O and H atoms of the hydrogen-bond in water dimer. Energy decomposition analysis also shows a non-negligible contribution of the induction term. These results illustrate the covalent-like character of the hydrogen bond between water molecules, which contributes to the essential understanding of ice, liquid water, related materials, and life sciences.