Multi-task learning for electronic structure to predict and explore molecular potential energy surfaces
We refine the OrbNet model to accurately predict energy, forces, and other response properties for molecules using a graph neural-network architecture based on features from low-cost approximated quantum operators in the symmetry-adapted atomic orbital basis. The model is end-to-end differentiable due to the derivation of analytic gradients for all electronic structure terms, and is shown to be transferable across chemical space due to the use of domain-specific features. The learning efficiency is improved by incorporating physically motivated constraints on the electronic structure through multi-task learning. The model outperforms existing methods on energy prediction tasks for the QM9 dataset and for molecular geometry optimizations on conformer datasets, at a computational cost that is thousand-fold or more reduced compared to conventional quantum-chemistry calculations (such as density functional theory) that offer similar accuracy.
Code (0)
등록된 구현이 없습니다.
Tasks
Graph Neural NetworkMulti-Task LearningSimilar Papers 제목 키워드 기반
Universal Machine Learning Kohn-Sham Hamiltonian for Materials
While density functional theory (DFT) serves as a prevalent computational approach in electronic structure calculations, its computational demands and scalability limitations persist. Recently, leveraging neural networks…
Neural networks and kernel ridge regression for excited states dynamics of CH$_2$NH$_2^+$: From single-state to multi-state representations and multi-property machine learning models
Excited-state dynamics simulations are a powerful tool to investigate photo-induced reactions of molecules and materials and provide complementary information to experiments. Since the applicability of these simulation t…
BIG-bench Machine Learningmolecular representationregressionEquitable Electronic Health Record Prediction with FAME: Fairness-Aware Multimodal Embedding
Electronic Health Record (EHR) data encompass diverse modalities -- text, images, and medical codes -- that are vital for clinical decision-making. To process these complex data, multimodal AI (MAI) has emerged as a powe…
FairnessAdvancing Molecular Machine Learning Representations with Stereoelectronics-Infused Molecular Graphs
Molecular representation is a critical element in our understanding of the physical world and the foundation for modern molecular machine learning. Previous molecular machine learning models have employed strings, finger…
Graph Neural NetworkMolecular Property Predictionmolecular representationProperty PredictionED-DiT: Physics-Guided Diffusion Pretraining for Transferable Molecular Representations from Electron Density
Pretraining has shown strong potential for learning transferable representations, yet it remains underexplored for electron-density-based molecular learning. Electron density provides a continuous three-dimensional descr…
Point Clouds