Search papers, labs, and topics across Lattice.
This paper introduces Graph Diffusion Counterfactual Explanation via Inversion (GDCE-I), a novel method for generating counterfactual explanations in graph neural networks (GNNs) that maintains adherence to domain-specific constraints while exploring the full edit space. By employing a discrete denoising diffusion model with a unique inversion scheme, GDCE-I achieves significant improvements in generating structural modifications that alter model predictions without sacrificing fidelity to the data manifold. The method outperforms existing approaches across four benchmarks and provides interpretable solutions specifically in the molecular domain, addressing critical gaps in the evaluation of graph counterfactuals.
GDCE-I achieves faithful and interpretable counterfactual explanations for graph neural networks without compromising on the integrity of the data structure or the search space.
Graph Neural Networks (GNNs) achieve strong predictive performance on graph-structured data across domains such as chemistry, biology, and network analysis, yet they provide no intrinsic explanation of their predictions. This limits their adoption in high-stakes and safety-critical settings. Counterfactual explanations address this by revealing the minimal structural modifications that would change a model's prediction. On graphs, however, such a modification is hard to produce. The search space is discrete and combinatorial, and a valid answer must respect categorical node and edge types together with domain rules such as chemical valency in the case of molecular graphs. Existing explainers give up one of two things. Either edits are not held on the data manifold, or the search does not span the full edit space. We propose Graph Diffusion Counterfactual Explanation via Inversion (GDCE-I), which gives up neither. A discrete denoising diffusion model with a novel discrete inversion scheme enables distribution-aware edits leveraging the whole domain edit space. We further address the incomplete and inconsistent evaluation of graph counterfactuals by deriving a framework of explanation desiderata and applying it to every method under one shared protocol. Across four benchmarks, GDCE-I outperforms related work by a large margin on the defined framework. For the molecular domain, we further qualitatively show that GDCE-I attains interpretable in-distribution solutions.
