A deep neural network for molecular wave functions in quasi-atomic minimal basis representation

Abstract

The emergence of machine learning methods in quantum chemistry provides new methods to revisit an old problem: Can the predictive accuracy of electronic structure calculations be decoupled from their numerical bottlenecks? Previous attempts to answer this question have, among other methods, given rise to semi-empirical quantum chemistry in minimal basis representation. We present an adaptation of the recently proposed SchNet for Orbitals (SchNOrb) deep convolutional neural network model [Nature Commun. 10, 5024 (2019)] for electronic wave functions in an optimised quasi-atomic minimal basis representation. For five organic molecules ranging from 5 to 13 heavy atoms, the model accurately predicts molecular orbital energies and wavefunctions and provides access to derived properties for chemical bonding analysis. Particularly for larger molecules, the model outperforms the original atomic-orbital-based SchNOrb method in terms of accuracy and scaling. We conclude by discussing the future potential of this approach in quantum chemical workflows.

arXiv comment15 pages, 9 figures

References in corpus (21)

• Adam: A Method for Stochastic Optimization
• Maximally localized Wannier functions: Theory and applications
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Machine learning and the physical sciences
• SchNet - a deep learning architecture for molecules and materials
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Machine Learning Unifies the Modelling of Materials and Molecules
• By-passing the Kohn-Sham equations with machine learning
• Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
• Deep neural network solution of the electronic Schrödinger equation
• SchNet: A continuous-filter convolutional neural network for modeling quantum interactions
• SchNetPack: A Deep Learning Toolbox For Atomistic Systems
• Density-functional tight-binding for beginners
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
• Transferability in Machine Learning for Electronic Structure via the Molecular Orbital Basis
• Representing molecule-surface interactions with symmetry-adapted neural networks
• Operators in Machine Learning: Response Properties in Chemical Space
• A Universal Density Matrix Functional from Molecular Orbital-Based Machine Learning: Transferability across Organic Molecules
• Accurate tight-binding Hamiltonian matrices from ab-initio calculations: Minimal basis sets

Cited by in corpus (16)

• Machine Learning Force Fields
• Machine learning for electronically excited states of molecules
• Perspective on integrating machine learning into computational chemistry and materials science
• Machine learning and excited-state molecular dynamics
• Equivariant representations for molecular Hamiltonians and N-center atomic-scale properties
• Equivariant analytical mapping of first principles Hamiltonians to accurate and transferable materials models
• Universal Machine Learning Kohn-Sham Hamiltonian for Materials
• Machine Learning and Data-Driven Methods in Computational Surface and Interface Science
• Convergence to the fixed-node limit in deep variational Monte Carlo
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• Mean-Field Density Matrix Decompositions
• Low-data deep quantum chemical learning for accurate MP2 and coupled-cluster correlations
• Integrated workflows and interfaces for data-driven semi-empirical electronic structure calculations
• Machine Learning Wavefunction
• Orbital Mixer: Using Atomic Orbital Features for Basis Dependent Prediction of Molecular Wavefunctions
• Fault-tolerant Quantum Chemical Calculations with Improved Machine-Learning Models