GeoT: A Geometry-aware Transformer for Reliable Molecular Property Prediction and Chemically Interpretable Representation Learning
arXiv:2106.15516
Abstract
In recent years, molecular representation learning has emerged as a key area of focus in various chemical tasks. However, many existing models fail to fully consider the geometric information of molecular structures, resulting in less intuitive representations. Moreover, the widely used message-passing mechanism is limited to provide the interpretation of experimental results from a chemical perspective. To address these challenges, we introduce a novel Transformer-based framework for molecular representation learning, named the Geometry-aware Transformer (GeoT). GeoT learns molecular graph structures through attention-based mechanisms specifically designed to offer reliable interpretability, as well as molecular property prediction. Consequently, GeoT can generate attention maps of interatomic relationships associated with training objectives. In addition, GeoT demonstrates comparable performance to MPNN-based models while achieving reduced computational complexity. Our comprehensive experiments, including an empirical simulation, reveal that GeoT effectively learns the chemical insights into molecular structures, bridging the gap between artificial intelligence and molecular sciences.
arXiv commentThis paper is currently under review. The code is available at this URL: https://github.com/oleneyl/geometry-aware-transformer
References in corpus (16)
- Adam: A Method for Stochastic Optimization
- Neural Message Passing for Quantum Chemistry
- Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
- Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
- Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)
- Machine Learning of Accurate Energy-Conserving Molecular Force Fields
- PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
- Convolutional Networks on Graphs for Learning Molecular Fingerprints
- Searching for Activation Functions
- Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
- SchNet: A continuous-filter convolutional neural network for modeling quantum interactions
- SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networks
- sGDML: Constructing Accurate and Data Efficient Molecular Force Fields Using Machine Learning
- Cormorant: Covariant Molecular Neural Networks
- Molecule Attention Transformer
- Graph-Aware Transformer: Is Attention All Graphs Need?