Timezone: »
Modeling the energy and forces of atomic systems is a fundamental problem in computational chemistry with the potential to help address many of the world’s most pressing problems, including those related to energy scarcity and climate change. These calculations are traditionally performed using Density Functional Theory, which is computationally very expensive. Machine learning has the potential to dramatically improve the efficiency of these calculations from days or hours to seconds.We propose the Spherical Channel Network (SCN) to model atomic energies and forces. The SCN is a graph neural network where nodes represent atoms and edges their neighboring atoms. The atom embeddings are a set of spherical functions, called spherical channels, represented using spherical harmonics. We demonstrate, that by rotating the embeddings based on the 3D edge orientation, more information may be utilized while maintaining the rotational equivariance of the messages. While equivariance is a desirable property, we find that by relaxing this constraint in both message passing and aggregation, improved accuracy may be achieved. We demonstrate state-of-the-art results on the large-scale Open Catalyst 2020 dataset in both energy and force prediction for numerous tasks and metrics.
Author Information
Larry Zitnick (Fundamental AI Research at Meta AI)
Abhishek Das (Meta FAIR)
Adeesh Kolluru (Carnegie Mellon University)
Janice Lan (FAIR)
Muhammed Shuaibi (Facebook)
Anuroop Sriram (Facebook AI Research)
Zachary Ulissi (Carnegie Mellon University)
Brandon Wood (FAIR)
More from the Same Authors
-
2022 : Chemistry Insights for Large Pretrained GNNs »
Janice Lan · Katherine Xu -
2022 : Fifteen-minute Competition Overview Video »
Abhishek Das · Zachary Ulissi · Siddharth Goyal · Nima Shoghi · Muhammed Shuaibi · Larry Zitnick · Janice Lan · Brook Wander · Brandon Wood · Anuroop Sriram · Ammar Rizvi · Aini Palizhati · Adeesh Kolluru -
2022 Competition: Open Catalyst Challenge »
Abhishek Das · Muhammed Shuaibi · Aini Palizhati · Siddharth Goyal · Adeesh Kolluru · Janice Lan · Ammar Rizvi · Nima Shoghi · Anuroop Sriram · Brook Wander · Brandon Wood · Zachary Ulissi · Larry Zitnick -
2021 Poster: Compositional Transformers for Scene Generation »
Dor Arad Hudson · Larry Zitnick -
2021 : Open Catalyst Challenge + Q&A »
Abhishek Das · Muhammed Shuaibi · Siddharth Goyal · Adeesh Kolluru · Janice Lan · Aini Palizhati · Anuroop Sriram · Brandon Wood · Aditya Grover · Devi Parikh · Zachary Ulissi · Larry Zitnick -
2020 : Contributed Talk - Biological structure and function emerge from scaling unsupervised learning to 250 million protein sequences »
Alexander Rives · Siddharth Goyal · Joshua Meier · Zeming Lin · Demi Guo · Myle Ott · Larry Zitnick · Rob Fergus -
2020 : Larry Zitnick - fastMRI »
Larry Zitnick -
2019 : fastMRI Challenge Talks »
Nafisa Yakubova · Nicola Pezzotti · Puyang Wang · Larry Zitnick · Dimitrios Karkalousos · Shanhui Sun · Matthan Caan · Tullie Murrell · Patrick Putzky -
2019 Poster: LCA: Loss Change Allocation for Neural Network Training »
Janice Lan · Rosanne Liu · Hattie Zhou · Jason Yosinski -
2019 Poster: Deconstructing Lottery Tickets: Zeros, Signs, and the Supermask »
Hattie Zhou · Janice Lan · Rosanne Liu · Jason Yosinski -
2017 Poster: ELF: An Extensive, Lightweight and Flexible Research Platform for Real-time Strategy Games »
Yuandong Tian · Qucheng Gong · Wendy Shang · Yuxin Wu · Larry Zitnick -
2017 Oral: ELF: An Extensive, Lightweight and Flexible Research Platform for Real-time Strategy Games »
Yuandong Tian · Qucheng Gong · Wendy Shang · Yuxin Wu · Larry Zitnick -
2016 : Gathering common sense knowledge: how to game it? »
Larry Zitnick