Performance assessment of various graph neural network architectures for predicting yield in cross-coupling reactions

Abstract

Machine learning (ML) is revolutionizing various scientific fields, including chemistry. Among different ML techniques, Graph Neural Networks (GNNs) have emerged as a powerful tool for capturing complex relationships in data by representing datapoints as graphs. Applying GNNs for predicting yields in chemical reactions is one of the emerging area of focus. However, handling heterogeneous datasets remains a key challenge. In this study, we utilized a diverse datasets encompassing various transition metal-catalyzed cross-coupling reactions including Suzuki, Sonogashira, Cadiot- Chodkiewicz, Ullmann-type, and Buchwald-Hartwig coupling reactions. To predict reaction yields, we implement multiple GNN architectures, including Message Passing Neural Networks (MPNN), Residual Graph Convolutional Networks (ResGCN), GraphSAGE, Graph Attention Networks (GAT and GATv2), Graph Convolutional Networks (GCN), and Graph Isomorphism Networks (GIN). A comparative analysis of these architectures reveals that MPNN achieves the highest predictive performance, with an R² value of 0.75. Additionally, to enhance the interpretability of our models, we employed the Integrated Gradients method to determine the contribution of each input descriptors to the model’s yield prediction. This study highlights the potential of effective and explainable graphbased models predicting the yields of chemical reactions. This work contributes to the application of machine learning in catalysis, providing valuable insights for optimizing catalytic reactions and contributing to innovations in sustainable chemistry and organic synthesis.