Abstract:Cocrystal engineering as an effective way to modify solid-state properties has inspired great interest from diverse material fields while cocrystal density is an important property closely correlated with the material function. In order to accurately predict the cocrystal density, we develop a graph neural network (GNN)-based deep learning framework by considering three key factors of machine learning (data quality, feature presentation, and model architecture). The result shows that different stoichiometric ratios of molecules in cocrystals can significantly influence the prediction performances, highlighting the importance of data quality. In addition, the feature complementary is not suitable for augmenting the molecular graph representation in the cocrystal density prediction, suggesting that the complementary strategy needs to consider whether extra features can sufficiently supplement the lacked information in the original representation. Based on these results, 4144 cocrystals with 1:1 stoichiometry ratio are selected as the dataset, supplemented by the data augmentation of exchanging a pair of coformers. The molecular graph is determined to learn feature representation to train the GNN-based model. Global attention is introduced to further optimize the feature space and identify important atoms to realize the interpretability of the model. Benefited from the advantages, our model significantly outperforms three competitive models and exhibits high prediction accuracy for unseen cocrystals, showcasing its robustness and generality. Overall, our work not only provides a general cocrystal density prediction tool for experimental investigations but also provides useful guidelines for the machine learning application. All source codes are freely available at https://github.com/Xiao-Gua00/CCPGraph.

Comparative study of crystal structure prediction approaches based on a graph network and an optimization algorithm

Crystal structure prediction by combining graph network and optimization algorithm

Enhancing crystal structure prediction by combining computational and experimental data via graph networks

Crystal structure prediction by combining graph network and Bayesian optimization

Crystal structure prediction via combining graph network and Bayesian optimization

CSPBench: a benchmark and critical evaluation of Crystal Structure Prediction

Towards Quantitative Evaluation of Crystal Structure Prediction Performance

Crystal structure prediction using neural network potential and age-fitness Pareto genetic algorithm

CRYSPNet: Crystal Structure Predictions via Neural Network

General Graph Neural Network-Based Model To Accurately Predict Cocrystal Density and Insight from Data Quality and Feature Representation

Enhancing Crystal Structure Prediction by decomposition methods based on graph theory

Graph theory and graph neural network assisted high-throughput crystal structure prediction and screening for energy conversion and storage

CrysToGraph: A Comprehensive Predictive Model for Crystal Materials Properties and the Benchmark

Enhancing Crystal Structure Prediction by Decomposition and Evolution Schemes Based on Graph Theory

Improving Crystal Property Prediction from a Multiplex Graph Perspective

Study of crystal properties based on attention mechanism and crystal graph convolutional neural network

Accelerated Organic Crystal Structure Prediction with Genetic Algorithms and Machine Learning

Connectivity Optimized Nested Graph Networks for Crystal Structures

Predictive crystallography at scale: mapping, validating, and learning from 1,000 crystal energy landscapes

A geometric-information-enhanced crystal graph network for predicting properties of materials