Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

FragNet: A Graph Neural Network for Molecular Property Prediction with Four Levels of Interpretability

About

Molecular property prediction is essential in a variety of contemporary scientific fields, such as drug development and designing energy storage materials. Although there are many machine learning models available for this purpose, those that achieve high accuracy while also offering interpretability of predictions are uncommon. We present a graph neural network that not only matches the prediction accuracies of leading models but also provides insights on four levels of molecular substructures. This model helps identify which atoms, bonds, molecular fragments, and connections between fragments are significant for predicting a specific molecular property. Understanding the importance of connections between fragments is particularly valuable for molecules with substructures that do not connect through standard bonds. The model additionally can quantify the impact of specific fragments on the prediction, allowing the identification of fragments that may improve or degrade a property value. These interpretable features are essential for deriving scientific insights from the model's learned relationships between molecular structures and properties.

Gihan Panapitiya, Peiyuan Gao, C Mark Maupin, Emily G Saldanha• 2024

Related benchmarks

TaskDatasetResultRank
Molecular property predictionQM9
Cv1.884
70
Molecular property predictionPCQM4M V2
MAE0.107
10
Molecular property predictionMolecule3D (random)
MAE0.0319
9
Molecular property predictionMolecule3D (scaffold)
MAE0.145
9
Molecular property predictionOLED 100–500 atoms (random)
S1 MAE0.51
7
Showing 5 of 5 rows

Other info

Follow for update