Theoretical and Computational Chemistry

A Universal Deep Learning Framework based on Graph Neural Network for Virtual Co-Crystal Screening

Yuanyuan Jiang Sichuan University
,

Abstract

Cocrystal plays an important role in various fields. However, how to choose coformer remains a challenge on experiments. In this work, we develop a novel graph neural network (GNN) based deep learning framework to rapidly predict formation of the cocrystal. A large and reliable data set is first constructed, which contains 7871 samples. A complementary feature representation is proposed by combining molecular graph and molecular descriptors from priori knowledge. A new GNN learning architecture is then explored to effectively embed the priori knowledge into the “endto-end” learning on the molecular graph, in which multi-head attention mechanism is introduced to further optimize the feature space. Consequently, the performance of our model achieves 98.86% accuracy, greatly surpassing some traditional machine learning models and classic GNN models. Furthermore, the out-of-distribution prediction on energetic cocrystals is also high up to 97.11% accuracy, showing strong generalization.


Content

Thumbnail image of manuscripts.pdf
download asset manuscripts.pdf 0.78 MB [opens in a new tab]

Supplementary material

Thumbnail image of Support Information-Table S1.docx
download asset Support Information-Table S1.docx 0.13 MB [opens in a new tab]
Support Information-Table S1
Thumbnail image of Support Information-Table S2.docx
download asset Support Information-Table S2.docx 0.17 MB [opens in a new tab]
Support Information-Table S2
Thumbnail image of Support Information-Figure S1-8, Table S3-5.pdf
download asset Support Information-Figure S1-8, Table S3-5.pdf 0.88 MB [opens in a new tab]
Support Information-Figure S1-8, Table S3-5