Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Kolmogorov-Arnold graph neural networks for chemically informed prediction tasks on inorganic nanomaterials

About

The recent development of Kolmogorov-Arnold Networks (KANs) has found its application in the field of Graph Neural Networks (GNNs) particularly in molecular data modeling and potential drug discovery. Kolmogorov-Arnold Graph Neural Networks (KAGNNs) expand on the existing set of GNN models with KAN-based counterparts. KAGNNs have been demonstrably successful in surpassing the accuracy of MultiLayer Perceptron (MLP)-based GNNs in the task of molecular property prediction. These models were widely tested on the graph datasets consisting of organic molecules. In this study, we explore the application of KAGNNs towards inorganic nanomaterials. In 2024, a large scale inorganic nanomaterials dataset was published under the title CHILI (Chemically-Informed Large-scale Inorganic Nanomaterials Dataset), and various MLP-based GNNs have been tested on this dataset. We adapt and test our own KAGNNs appropriate for eight defined tasks. Our experiments reveal that, KAGNNs frequently surpass the performance of their counterpart GNNs. Most notably, on crystal system and space group classification tasks in CHILI-3K, KAGNNs achieve the new state-of-the-art results of 99.5 percent and 96.6 percent accuracy, respectively, compared to the previous 65.7 and 73.3 percent each.

Nikita Volzhin, Soowhan Yoon• 2025

Related benchmarks

TaskDatasetResultRank
Absolute Position PredictionCHILI-100K
MAE16.182
6
Atom ClassificationCHILI-3K
F1 Score65.7
6
Crystal System ClassificationCHILI-3K
F1 Score99.5
6
Crystal System ClassificationCHILI-100K
F1 Score0.477
6
Space Group ClassificationCHILI-3K
F1 Score96.6
6
Space Group ClassificationCHILI-100K
F1 Score21.6
6
xPDF PredictionCHILI-100K
MSE0.011
6
Atom ClassificationCHILI-100K
F1 Score44.9
6
Edge Attribute PredictionCHILI-3K
MSE (Edge Attribute)0.031
6
Edge Attribute PredictionCHILI-100K
MSE0.04
6
Showing 10 of 16 rows

Other info

Follow for update