MolNet: A Chemically Intuitive Graph Neural Network for Prediction of Molecular Properties

MolNet_Equi: A Chemically Intuitive, Rotation-Equivariant Graph Neural Network

Although deep-learning (DL) models suggest unprecedented prediction capabilities in tackling various chemical problems, their demonstrated tasks have so far been limited to the scalar properties including the magnitude of vectorial properties, such as molecular dipole moments. A rotation-equivariant MolNet_Equi model, proposed in this paper, understands and recognizes the molecular rotation in the 3D Euclidean space, and exhibits the ability to predict directional dipole moments in the rotation-sensitive mode, as well as showing superior performance for the prediction of scalar properties. Three consecutive operations of molecular rotationR M ${\left(R\left(M\right)\right)}$ , dipole-moment predictionφ μ R M ${\left({\phi{} }_{\mu }\left(R\left(M\right)\right)\right)}$ , and dipole-moment inverse-rotationR - 1 φ μ R M ${\left({R}^{-1}\left({\phi{} }_{\mu }\left(R\left(M\right)\right)\right)\right)}$ do not alter the original prediction of the total dipole moment of a moleculeφ μ M ${\left({\phi{} }_{\mu }\right(M\left)\right)}$ , assuring the rotational equivariance of MolNet_Equi. Furthermore, MolNet_Equi faithfully predicts the absolute direction of dipole moments given molecular poses, albeit the model has been trained only with the information on dipole-moment magnitudes, not directions. This work highlights the potential of incorporating fundamental yet crucial chemical rules and concepts into DL models, leading to the development of chemically intuitive models.

Identifying Candidate Gene-Disease Associations via Graph Neural Networks

Real-world objects are usually defined in terms of their own relationships or connections. A graph (or network) naturally expresses this model though nodes and edges. In biology, depending on what the nodes and edges represent, we may classify several types of networks, gene-disease associations (GDAs) included. In this paper, we presented a solution based on a graph neural network (GNN) for the identification of candidate GDAs. We trained our model with an initial set of well-known and curated inter- and intra-relationships between genes and diseases. It was based on graph convolutions, making use of multiple convolutional layers and a point-wise non-linearity function following each layer. The embeddings were computed for the input network built on a set of GDAs to map each node into a vector of real numbers in a multidimensional space. Results showed an AUC of 95% for training, validation, and testing, that in the real case translated into a positive response for 93% of the Top-15 (highest dot product) candidate GDAs identified by our solution. The experimentation was conducted on the DisGeNET dataset, while the DiseaseGene Association Miner (DG-AssocMiner) dataset by Stanford's BioSNAP was also processed for performance evaluation only.

Graph neural networks for materials science and chemistry

Machine learning plays an increasingly important role in many areas of chemistry and materials science, being used to predict materials properties, accelerate simulations, design new structures, and predict synthesis routes of new materials. Graph neural networks (GNNs) are one of the fastest growing classes of machine learning models. They are of particular relevance for chemistry and materials science, as they directly work on a graph or structural representation of molecules and materials and therefore have full access to all relevant information required to characterize materials. In this Review, we provide an overview of the basic principles of GNNs, widely used datasets, and state-of-the-art architectures, followed by a discussion of a wide range of recent applications of GNNs in chemistry and materials science, and concluding with a road-map for the further development and application of GNNs.