4.7 Article

Mixed Precision Fermi-Operator Expansion on Tensor Cores from a Machine Learning Perspective

Journal

JOURNAL OF CHEMICAL THEORY AND COMPUTATION
Volume 17, Issue 4, Pages 2256-2265

Publisher

AMER CHEMICAL SOC
DOI: 10.1021/acs.jctc.1c00057

Keywords

-

Funding

  1. U.S. Department of Energy Office of Basic Energy Sciences (FWP) [LANLE8AN]
  2. LANL LDRD-ER program
  3. U.S. Department of Energy through the Los Alamos National Laboratory
  4. Computational Systems and Software Environments (CSSE) subprogram of LANL's ASC program (NNSA/DOE)

Ask authors/readers for more resources

A second-order recursive Fermi-operator expansion scheme using mixed precision floating point operations on tensor core units is presented, achieving over 100 teraFLOPS of performance for half-precision floating point operations on Nvidia's A100. The scheme is formulated as a generalized, differentiable deep neural network structure, accelerating convergence by optimizing weight and bias values, and optimizing coefficients of the expansion to accurately represent fractional occupation numbers of electronic states at finite temperatures using a machine learning approach.
We present a second-order recursive Fermi-operator expansion scheme using mixed precision floating point operations to perform electronic structure calculations using tensor core units. A performance of over 100 teraFLOPs is achieved for half-precision floating point operations on Nvidia's A100 tensor core units. The second-order recursive Fermi-operator scheme is formulated in terms of a generalized, differentiable deep neural network structure, which solves the quantum mechanical electronic structure problem. We demonstrate how this network can be accelerated by optimizing the weight and bias values to substantially reduce the number of layers required for convergence. We also show how this machine learning approach can be used to optimize the coefficients of the recursive Fermi-operator expansion to accurately represent the fractional occupation numbers of the electronic states at finite temperatures.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.7
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available