Hierarchical Uncertainty-Aware Graph Neural Network
This work addresses the challenge of robust and interpretable graph learning for applications like social networks or bioinformatics, though it appears incremental by building on existing GNN approaches.
The paper tackles the problem of integrating multi-scale representation learning and uncertainty estimation in graph neural networks to mitigate noise and adversarial perturbations, resulting in a model that achieves state-of-the-art robustness and interpretability on semi-supervised classification tasks.
Recent research on graph neural networks (GNNs) has explored mechanisms for capturing local uncertainty and exploiting graph hierarchies to mitigate data sparsity and leverage structural properties. However, the synergistic integration of these two approaches remains underexplored. This work introduces a novel architecture, the Hierarchical Uncertainty-Aware Graph Neural Network (HU-GNN), which unifies multi-scale representation learning, principled uncertainty estimation, and self-supervised embedding diversity within a single end-to-end framework. Specifically, HU-GNN adaptively forms node clusters and estimates uncertainty at multiple structural scales from individual nodes to higher levels. These uncertainty estimates guide a robust message-passing mechanism and attention weighting, effectively mitigating noise and adversarial perturbations while preserving predictive accuracy on semi-supervised classification tasks. We also offer key theoretical contributions, including a probabilistic formulation, rigorous uncertainty-calibration guarantees, and formal robustness bounds. Extensive experiments on standard benchmarks demonstrate that our model achieves state-of-the-art robustness and interpretability.