Overcoming Shortcut Learning in Graph Neural Networks through Active Explanation Guidance

arXiv:2608.14121v1 Announce Type: cross Abstract: Graph Neural Networks (GNNs) can solve prediction tasks by unintentionally exploiting shortcuts---that is, edges, nodes, and features that correlate with but are not causal for the prediction---which compromise their reliability in out-of-distribution tasks. We introduce XIGL, an architecture-agnostic human-in-the-loop strategy for removing such shortcuts from GNNs. Our key insight is twofold. On the one hand, reliance on shortcuts can be detected by inspecting GNN explanations. On the other hand, once made aware of such shortcuts, sufficiently expert users can provide tailored corrective feedback, which helps deconfound the model. XIGL supports any query strategy; however, since corrective feedback can be expensive to acquire, we develop an active learning strategy for prioritizing explanations that are more likely to display shortcut behavior, lowering annotation and cognitive costs. We showcase the effectiveness of XIGL, including both existing and proposed explanation-based strategies, on several GNN architectures. Our implementation is available online.

Related

• UNR-Explainer: Counterfactual Explanations for Unsupervised Node Representation Learning Models
• ExPath: Targeted Pathway Inference for Biological Knowledge Bases via Graph Learning and Explanation
• Expressive Power of Deep Homomorphism Networks over Relational Databases
• Foundations of Equivariant Deep Learning: Unifying Graph and Sheaf Neural Networks
• TypeBandit: Type-Level Context Allocation and Reweighting for Effective Attribute Completion in Heterogeneous Graph Neural Networks

Source: arXiv cs.AI | 2026-08-17