Highly Accurate Quantum Chemical Property Prediction with Uni-Mol+
- URL: http://arxiv.org/abs/2303.16982v2
- Date: Fri, 7 Jul 2023 06:38:18 GMT
- Title: Highly Accurate Quantum Chemical Property Prediction with Uni-Mol+
- Authors: Shuqi Lu, Zhifeng Gao, Di He, Linfeng Zhang, Guolin Ke
- Abstract summary: We propose a novel approach called Uni-Mol+ to speed up the prediction of quantum chemical (QC) properties.
We introduce a two-track Transformer model backbone and train it with the QC property prediction task.
Our benchmarking results demonstrate that the proposed Uni-Mol+ significantly improves the accuracy of QC property prediction.
- Score: 36.61144345418364
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: Recent developments in deep learning have made remarkable progress in
speeding up the prediction of quantum chemical (QC) properties by removing the
need for expensive electronic structure calculations like density functional
theory. However, previous methods learned from 1D SMILES sequences or 2D
molecular graphs failed to achieve high accuracy as QC properties primarily
depend on the 3D equilibrium conformations optimized by electronic structure
methods, far different from the sequence-type and graph-type data. In this
paper, we propose a novel approach called Uni-Mol+ to tackle this challenge.
Uni-Mol+ first generates a raw 3D molecule conformation from inexpensive
methods such as RDKit. Then, the raw conformation is iteratively updated to its
target DFT equilibrium conformation using neural networks, and the learned
conformation will be used to predict the QC properties. To effectively learn
this update process towards the equilibrium conformation, we introduce a
two-track Transformer model backbone and train it with the QC property
prediction task. We also design a novel approach to guide the model's training
process. Our extensive benchmarking results demonstrate that the proposed
Uni-Mol+ significantly improves the accuracy of QC property prediction in
various datasets. We have made the code and model publicly available at
\url{https://github.com/dptech-corp/Uni-Mol}.
Related papers
- COEFF-KANs: A Paradigm to Address the Electrolyte Field with KANs [5.759388420139191]
We propose a novel method to automatically predict Coulombic Efficiency (CE) based on the composition of liquid electrolytes.
We input the obtained electrolyte features into a Multi-layer Perceptron or Kolmogorov-Arnold Network to predict CE.
Experimental results on a real-world dataset demonstrate that our method achieves SOTA for predicting CE compared to all baselines.
arXiv Detail & Related papers (2024-07-24T14:45:25Z) - Variational Monte Carlo on a Budget -- Fine-tuning pre-trained Neural
Wavefunctions [5.145741425164946]
Deep-learning-based Variational Monte Carlo (DL-VMC) has recently outperformed conventional approaches in terms of accuracy, but only at large computational cost.
We propose a DL-VMC model which has been pre-trained using self-supervised wavefunction optimization on a large and chemically diverse set of molecules.
Applying this model to new molecules without any optimization, yields wavefunctions and absolute energies that outperform established methods such as CCSD(T)-2Z.
arXiv Detail & Related papers (2023-07-15T09:56:22Z) - QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Molecules [69.25826391912368]
We generate a new Quantum Hamiltonian dataset, named as QH9, to provide precise Hamiltonian matrices for 999 or 2998 molecular dynamics trajectories.
We show that current machine learning models have the capacity to predict Hamiltonian matrices for arbitrary molecules.
arXiv Detail & Related papers (2023-06-15T23:39:07Z) - FAENet: Frame Averaging Equivariant GNN for Materials Modeling [123.19473575281357]
We introduce a flexible framework relying on frameaveraging (SFA) to make any model E(3)-equivariant or invariant through data transformations.
We prove the validity of our method theoretically and empirically demonstrate its superior accuracy and computational scalability in materials modeling.
arXiv Detail & Related papers (2023-04-28T21:48:31Z) - Heterogenous Ensemble of Models for Molecular Property Prediction [55.91865861896012]
We propose a method for considering different modalities on molecules.
We ensemble these models with a HuberRegressor.
This yields a winning solution to the 2textsuperscriptnd edition of the OGB Large-Scale Challenge (2022)
arXiv Detail & Related papers (2022-11-20T17:25:26Z) - 3D Equivariant Molecular Graph Pretraining [42.957880677779556]
We tackle 3D molecular pretraining in a complete and novel sense.
We first propose to adopt an equivariant energy-based model as the backbone for pretraining, which enjoys the merit of fulfilling the symmetry of 3D space.
We evaluate our model pretrained from a large-scale 3D dataset GEOM-QM9 on two challenging 3D benchmarks: MD17 and QM9.
arXiv Detail & Related papers (2022-07-18T16:26:24Z) - Equivariant vector field network for many-body system modeling [65.22203086172019]
Equivariant Vector Field Network (EVFN) is built on a novel equivariant basis and the associated scalarization and vectorization layers.
We evaluate our method on predicting trajectories of simulated Newton mechanics systems with both full and partially observed data.
arXiv Detail & Related papers (2021-10-26T14:26:25Z) - SE(3)-equivariant prediction of molecular wavefunctions and electronic
densities [4.2572103161049055]
We introduce general SE(3)-equivariant operations and building blocks for constructing deep learning architectures for geometric point cloud data.
Our model reduces prediction errors by up to two orders of magnitude compared to the previous state-of-the-art.
We demonstrate the potential of our approach in a transfer learning application, where a model trained on low accuracy reference wavefunctions implicitly learns to correct for electronic many-body interactions.
arXiv Detail & Related papers (2021-06-04T08:57:46Z) - Benchmarking adaptive variational quantum eigensolvers [63.277656713454284]
We benchmark the accuracy of VQE and ADAPT-VQE to calculate the electronic ground states and potential energy curves.
We find both methods provide good estimates of the energy and ground state.
gradient-based optimization is more economical and delivers superior performance than analogous simulations carried out with gradient-frees.
arXiv Detail & Related papers (2020-11-02T19:52:04Z) - Automated discovery of a robust interatomic potential for aluminum [4.6028828826414925]
Machine learning (ML) based potentials aim for faithful emulation of quantum mechanics (QM) calculations at drastically reduced computational cost.
We present a highly automated approach to dataset construction using the principles of active learning (AL)
We demonstrate this approach by building an ML potential for aluminum (ANI-Al)
To demonstrate transferability, we perform a 1.3M atom shock simulation, and show that ANI-Al predictions agree very well with DFT calculations on local atomic environments sampled from the nonequilibrium dynamics.
arXiv Detail & Related papers (2020-03-10T19:06:32Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.