skip to main content
US FlagAn official website of the United States government
dot gov icon
Official websites use .gov
A .gov website belongs to an official government organization in the United States.
https lock icon
Secure .gov websites use HTTPS
A lock ( lock ) or https:// means you've safely connected to the .gov website. Share sensitive information only on official, secure websites.

Explore Research Products in the PAR
It may take a few hours for recently added research products to appear in PAR search results.


Title: Efficient and Equivariant Graph Networks for Predicting Quantum Hamiltonian
We consider the prediction of the Hamiltonian matrix, which finds use in quantum chemistry and condensed matter physics. Efficiency and equiv- ariance are two important, but conflicting factors. In this work, we propose a SE(3)-equivariant net- work, named QHNet, that achieves efficiency and equivariance. Our key advance lies at the inno- vative design of QHNet architecture, which not only obeys the underlying symmetries, but also en- ables the reduction of number of tensor products by 92%. In addition, QHNet prevents the expo- nential growth of channel dimension when more atom types are involved. We perform experiments on MD17 datasets, including four molecular sys- tems. Experimental results show that our QHNet can achieve comparable performance to the state of the art methods at a significantly faster speed. Besides, our QHNet consumes 50% less mem- ory due to its streamlined architecture. Our code is publicly available as part of the AIRS library (https://github.com/divelab/AIRS).  more » « less
Award ID(s):
2119103
PAR ID:
10460974
Author(s) / Creator(s):
Date Published:
Journal Name:
International Conference on Machine Learning
Format(s):
Medium: X
Sponsoring Org:
National Science Foundation
More Like this
  1. ; ; ; ; (, Proceedings of the 40th International Conference on Machine Learning)
    We consider the prediction of the Hamiltonian matrix, which finds use in quantum chemistry and condensed matter physics. Efficiency and equivariance are two important, but conflicting factors. In this work, we propose a SE(3)-equivariant network, named QHNet, that achieves efficiency and equivariance. Our key advance lies at the innovative design of QHNet architecture, which not only obeys the underlying symmetries, but also enables the reduction of number of tensor products by 92%. In addition, QHNet prevents the exponential growth of channel dimension when more atom types are involved. We perform experiments on MD17 datasets, including four molecular systems. Experimental results show that our QHNet can achieve comparable performance to the state of the art methods at a significantly faster speed. Besides, our QHNet consumes 50% less memory due to its streamlined architecture. Our code is publicly available as part of the AIRS library (https://github.com/divelab/AIRS). 
    more » « less
  2. ; ; ; ; (, Proceedings of Machine Learning Research)
    Andreas Krause, Emma Brunskill (Ed.)
    We consider the prediction of the Hamiltonian matrix, which finds use in quantum chemistry and condensed matter physics. Efficiency and equivariance are two important, but conflicting factors. In this work, we propose a SE(3)-equivariant network, named QHNet, that achieves efficiency and equivariance. Our key advance lies at the innovative design of QHNet architecture, which not only obeys the underlying symmetries, but also enables the reduction of number of tensor products by 92%. In addition, QHNet prevents the exponential growth of channel dimension when more atom types are involved. We perform experiments on MD17 datasets, including four molecular systems. Experimental results show that our QHNet can achieve comparable performance to the state of the art methods at a significantly faster speed. Besides, our QHNet consumes 50% less memory due to its streamlined architecture. Our code is publicly available as part of the AIRS library (https://github.com/divelab/AIRS). 
    more » « less
  3. ; (, Atmospheric Measurement Techniques)
    null (Ed.)
    Abstract. Temperature, H2O, and O3 profiles, as well as CO2, N2O, CH4, chlorofluorocarbon-12 (CFC-12), and sea surface temperature (SST) scalar anomalies are computed using a clear subset of AIRS observations over ocean for the first 16 years of NASA's Earth-Observing Satellite (EOS) Aqua Atmospheric Infrared Sounder (AIRS) operation. The AIRS Level-1c radiances are averaged over 16 d and 40 equal-area zonal bins and then converted to brightness temperature anomalies. Geophysical anomalies are retrieved from the brightness temperature anomalies using a relatively standard optimal estimation approach. The CO2, N2O, CH4, and CFC-12 anomalies are derived by applying a vertically uniform multiplicative shift to each gas in order to obtain an estimate for the gas mixing ratio. The minor-gas anomalies are compared to the National Oceanic and Atmospheric Administration (NOAA) Earth System Research Laboratory (ESRL) in situ values and used to estimate the radiometric stability of the AIRS radiances. Similarly, the retrieved SST anomalies are compared to the SST values used in the ERA-Interim reanalysis and to NOAA's Optimum Interpolation SST (OISST) product. These intercomparisons strongly suggest that many AIRS channels are stable to better than 0.02 to 0.03 K per decade, well below climate trend levels, indicating that the AIRS blackbody is not drifting. However, detailed examination of the anomaly retrieval residuals (observed – computed) shows various small unphysical shifts that correspond to AIRS hardware events (shutdowns, etc.). Some examples are given highlighting how the AIRS radiance stability could be improved, especially for channels sensitive to N2O and CH4. The AIRS shortwave channels exhibit larger drifts that make them unsuitable for climate trending, and they are avoided in this work. The AIRS Level 2 surface temperature retrievals only use shortwave channels. We summarize how these shortwave drifts impacts recently published comparisons of AIRS surface temperature trends to other surface climatologies. 
    more » « less
  4. ; ; ; ; (, arXivorg)
    We consider the prediction of general tensor properties of crystalline materials, including dielectric, piezoelectric, and elastic tensors. A key challenge here is how to make the predictions satisfy the unique tensor equivariance to O(3) group and invariance to crystal space groups. To this end, we propose a General Materials Tensor Network (GMTNet), which is carefully designed to satisfy the required symmetries. To evaluate our method, we curate a dataset and establish evaluation metrics that are tailored to the intricacies of crystal tensor predictions. Experimental results show that our GMTNet not only achieves promising performance on crystal tensors of various orders but also generates predictions fully consistent with the intrinsic crystal symmetries. Our code is publicly available as part of the AIRS library (https://github.com/divelab/AIRS). 
    more » « less
  5. ; ; ; ; (, The Forty-first International Conference on Machine Learning)
    We consider the prediction of general tensor properties of crystalline materials, including dielectric, piezoelectric, and elastic tensors. A key challenge here is how to make the predictions satisfy the unique tensor equivariance to O(3) group and invariance to crystal space groups. To this end, we propose a General Materials Tensor Network (GMTNet), which is carefully designed to satisfy the required symmetries. To evaluate our method, we curate a dataset and establish evaluation metrics that are tailored to the intricacies of crystal tensor predictions. Experimental results show that our GMTNet not only achieves promising performance on crystal tensors of various orders but also generates predictions fully consistent with the intrinsic crystal symmetries. Our code is publicly available as part of the AIRS library (https://github.com/divelab/AIRS). 
    more » « less
  • Citation Formats
  • MLA
  • APA
  • Chicago
  • BibTeX
  • Save / Share this Record
  • Send to Email