Latent Data Augmentation for Node Classification with Graph Neural Networks - Archive ouverte HAL
Communication Dans Un Congrès Année : 2024

Latent Data Augmentation for Node Classification with Graph Neural Networks

Abderaouf Gacem
Mohammed Haddad
Hamida Seba

Résumé

We consider the problem of data augmentation for graph neural networks. While other areas of deep learning, especially computer vision, has benefited greatly from data augmentation, graph deep learning is limited by the complex structure and nodes dependency of graph datasets. The works studying data augmentation for graphs focus on ways to train the model on generated graphs or disturb the original one by masking attributes or dropping nodes/edges. Our work studies data augmentation for graphs in the latent space. We propose an architecture to train a graph neural network including a data augmentation component which also has a denoising and structure enhancer function. Our architecture leverages the supervised feedback to introduce the graph neural network component to a new latent representation of the graph with lesser irrelevant connections on the graph domain. We perform experiments on datasets from several domains with different sizes and show that our architecture improves the performance over a variety of plain graph neural networks on node classification.
Fichier non déposé

Dates et versions

hal-04820461 , version 1 (05-12-2024)

Identifiants

  • HAL Id : hal-04820461 , version 1

Citer

Abderaouf Gacem, Mohammed Haddad, Hamida Seba. Latent Data Augmentation for Node Classification with Graph Neural Networks. ACS/IEEE 21st International Conference on Computer Systems and Applications, Oct 2024, Sousse, Tunisia. ⟨hal-04820461⟩
0 Consultations
0 Téléchargements

Partager

More