Improving Subgraph Representation Learning via Multi-View Augmentation

Yili Shen,Xiao Liu,Cheng-Wei Ju,Jiaxu Yan,Jun Yi,Zhou Lin,Hui Guan
DOI: https://doi.org/10.48550/arXiv.2205.13038
IF: 5.414
2022-05-25
Machine Learning
Abstract:Subgraph representation learning based on Graph Neural Network (GNN) has exhibited broad applications in scientific advancements, such as predictions of molecular structure-property relationships and collective cellular function. In particular, graph augmentation techniques have shown promising results in improving graph-based and node-based classification tasks. Still, they have rarely been explored in the existing GNN-based subgraph representation learning studies. In this study, we develop a novel multi-view augmentation mechanism to improve subgraph representation learning models and thus the accuracy of downstream prediction tasks. Our augmentation technique creates multiple variants of subgraphs and embeds these variants into the original graph to achieve highly improved training efficiency, scalability, and accuracy. Benchmark experiments on several real-world biological and physiological datasets demonstrate the superiority of our proposed multi-view augmentation techniques in subgraph representation learning.
What problem does this paper attempt to address?