English

UNR-Explainer: Counterfactual Explanations for Unsupervised Node Representation Learning Models

Machine Learning 2026-05-19 v1 Artificial Intelligence

Abstract

Node representation learning, such as Graph Neural Networks (GNNs), has emerged as a pivotal method in machine learning. The demand for reliable explanation generation surges, yet unsupervised models remain underexplored. To bridge this gap, we introduce a method for generating counterfactual (CF) explanations in unsupervised node representation learning. We identify the most important subgraphs that cause a significant change in the k-nearest neighbors of a node of interest in the learned embedding space upon perturbation. The k-nearest neighbor-based CF explanation method provides simple, yet pivotal, information for understanding unsupervised downstream tasks, such as top-k link prediction and clustering. Consequently, we introduce UNR-Explainer for generating expressive CF explanations for Unsupervised Node Representation learning methods based on a Monte Carlo Tree Search (MCTS). The proposed method demonstrates superior performance on diverse datasets for unsupervised GraphSAGE and DGI.

Keywords

Cite

@article{arxiv.2605.17285,
  title  = {UNR-Explainer: Counterfactual Explanations for Unsupervised Node Representation Learning Models},
  author = {Hyunju Kang and Geonhee Han and Hogun Park},
  journal= {arXiv preprint arXiv:2605.17285},
  year   = {2026}
}

Comments

Accepted at ICLR 2024