DINE: Dimensional Interpretability of Node Embeddings
Open Access
- 9 July 2024
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Knowledge and Data Engineering
- Vol. 36 (12) , 7986-7997
Abstract
Graph representation learning methods, such as node embeddings, are powerful approaches to map nodes into a latent vector space, allowing their use for various graph learning tasks. Despite their success, these techniques are inherently black-boxes and few studies have focused on investigating local explanations of node embeddings for specific instances. Moreover, explaining the overall behavior of unsupervised embedding models remains an unexplored problem, limiting global interpretability and debugging potentials. We address this gap by developing human-understandable explanations for latent space dimensions in node embeddings. Towards that, we first develop new metrics that measure the global interpretability of embeddings based on the marginal contribution of the latent dimensions to predicting graph structure. We say an embedding dimension is more interpretable if it can faithfully map to an understandable sub-structure in the input graph - like community structure. Having observed that standard node embeddings have low interpretability, we then introduce Dine (Dimension-based Interpretable Node Embedding). This novel approach can retrofit existing node embeddings by making them more interpretable without sacrificing their task performance. We conduct extensive experiments on synthetic and real-world graphs and show that we can simultaneously learn highly interpretable node embeddings with effective performance in link prediction and node classification.Keywords
All Related Versions
Funding Information
- European Community Program (871042)
- SoBigData++: European Integrated Infrastructure for Social Mining and Big Data Analytics (834756)
- XAI: Science and technology for the eXplanation of AI decision making
This publication has 58 references indexed in Scilit:
- Generating post-hoc explanations for Skip-gram-based node embeddings by identifying important nodes with bridgenessNeural Networks, 2023
- Deep MinCut: Learning Node Embeddings by Detecting CommunitiesPattern Recognition, 2022
- Modularity-aware graph autoencoders for joint community detection and link predictionNeural Networks, 2022
- Explaining individual predictions when features are dependent: More accurate approximations to Shapley valuesArtificial Intelligence, 2021
- Finding Interpretable Concept Spaces in Node Embeddings Using Knowledge Bases2020
- Exploring the Semantic Content of Unsupervised Graph Embeddings: An Empirical StudyData Science and Engineering, 2019
- Explaining prediction models and individual predictions with feature contributionsKnowledge and Information Systems, 2013
- Community detection in graphsPhysics Reports, 2010
- Hierarchical structure and the prediction of missing links in networksNature, 2008
- Stochastic blockmodels: First stepsSocial Networks, 1983