错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Graph contrastive learning with high-order feature interactions and adversarial Wasserstein-distance-based alignment

  • Chenxu Wang,
  • Zhizhong Wan,
  • Panpan Meng,
  • Shihao Wang,
  • Zhanggong Wang

摘要

Graph contrastive learning (GCL) has proven to be an effective approach for unsupervised representation learning on graph-structured data. However, existing GCL models face two major limitations. First, existing feature augmentation methods fail to capture the high-order interactions among raw features, which are essential for feature engineering. Second, effective strategies for extracting global information from graphs for contrastive learning remain limited. To address these limitations, we propose a novel GCL model with high-order feature interactions and adversarial Wasserstein-distance-based alignment. Our model employs DNNs to capture complex interactions among raw features and introduce an alignment-based loss function to effectively extract global graph information. While traditional methods for calculating Wasserstein distance between graph views are computationally intensive, we overcome this challenge by training an adversarial Wasserstein-distance discriminator that enables efficient distance computation. We conduct extensive experiments on five benchmark datasets to evaluate the performance of the proposed method. The experimental results demonstrate that our approach achieves superior performance on classification tasks.