Enhancing Atomic Systems Energy Prediction with Graph Neural Networks
Alrashdi, Baketah
Alrashdi, Baketah
Author
Supervisor
Department
Machine Learning
Embargo End Date
2024-01-01
Type
Thesis
Date
2024
License
Language
English
Collections
Research Projects
Organizational Units
Journal Issue
Abstract
This thesis investigates the application of Graph Neural Networks (GNNs) with inte-grated attention mechanisms to enhance predictions of energy and forces in atomic sys- tems, focusing on the Polarizable Atom Interaction Neural Network (PaiNN) model [20]. By incorporating novel parameters that account for atomic distances and bond types, our approach aims to refine the model s accuracy and interpretability in predicting molecular dynamics and properties. Utilizing the Open Catalyst Dataset (OC20), we explore several PaiNN configurations that include Principal Neighborhood Aggregation (PNA) [2] for dynamic attention-based interaction processing. Our results show that the attention- augmented PaiNN model outperforms baseline GNN models in predicting scalar and tensorial properties of molecules, resulting in a significant increase in data efficiency and predictive accuracy. This study combines computational chemistry and machine learning to present a novel way for modeling molecule energy estimates. The incorporation of attention processes into GNNs marks a significant advancement in molecular simulations, paving the path for novel applications in material science, drug development, and beyond.
Citation
B. Alrashdi, "Enhancing Atomic Systems Energy Prediction with Graph Neural Networks", MS. Thesis, Machine Learning, MBZUAI, Abu Dhabi, UAE, 2024