Home > Research > Publications & Outputs > Logical Distillation of Graph Neural Networks

Electronic data

  • 2406.07126v3 (1)

    Accepted author manuscript, 224 KB, PDF document

    Available under license: CC BY-NC-ND: Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License

Links

Text available via DOI:

View graph of relations

Logical Distillation of Graph Neural Networks

Research output: Contribution in Book/Report/Proceedings - With ISBN/ISSNChapter

Published
Close
Publication date2/11/2024
Host publicationProceedings of the 21st International Conference on Principles of Knowledge Representation and Reasoning
EditorsPierre Marquis, Magdalena Ortiz, Maurice Pagnucco
PublisherIJCAI
Pages920-930
Number of pages11
ISBN (electronic)9781956792058
<mark>Original language</mark>English

Abstract

We present a logic based interpretable model for learning on graphs and an algorithm to distill this model from a Graph Neural Network (GNN). Recent results have shown connections between the expressivity of GNNs and the two-variable fragment of first-order logic with counting quantifiers (C2). We introduce a decision-tree based model which leverages an extension of C2 to distill interpretable logical classifiers from GNNs. We test our approach on multiple GNN architectures. The distilled models are interpretable, succinct, and attain similar accuracy to the underlying GNN. Furthermore, when the ground truth is expressible in C2, our approach outperforms the GNN.