Abstract
Inspections on current graph neural networks suggest us to reconsider the computational aspect of the final aggregation. We consider that such aggregations perform a prediction smoothing and impute their potential drawbacks to be the inter-class interference implied by the underlying graphs. We aim at weakening the inter-class connections so that aggregations focus more on intra-class relations and producing smooth predictions according to weakening results. We apply a metric learning module to learn new edge weights and combine entropy losses to ensure the correspondence between the predictions and the learnt distances so that the weights of inter-class edges are reduced and predictions are smoothed according to the modified graph. Experiments on four citation networks and a Wiki network show that in comparison with other state-of-the-art graph neural networks, the proposed algorithm can improve the classification accuracy.
| Original language | English |
|---|---|
| Article number | 108492 |
| Journal | Pattern Recognition |
| Volume | 124 |
| Online published | 11 Dec 2021 |
| DOIs | |
| Publication status | Published - Apr 2022 |
| Externally published | Yes |
Research Keywords
- Adaptive graph smoothing networks
- Graph convolutional networks
- Semi-supervised learning
- Graph node classification