TY - JOUR
T1 - Graph contrastive learning with high-order feature interactions and adversarial Wasserstein-distance-based alignment
AU - Wang, Chenxu
AU - Wan, Zhizhong
AU - Meng, Panpan
AU - Wang, Shihao
AU - Wang, Zhanggong
N1 - Publisher Copyright:
© The Author(s), under exclusive licence to Springer-Verlag GmbH Germany, part of Springer Nature 2024.
PY - 2025/6
Y1 - 2025/6
N2 - 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.
AB - 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.
KW - Adversarial Wasserstein-distance-based alignment
KW - Feature interaction
KW - Graph contrastive learning
UR - https://www.scopus.com/pages/publications/85209722893
U2 - 10.1007/s13042-024-02461-4
DO - 10.1007/s13042-024-02461-4
M3 - 文章
AN - SCOPUS:85209722893
SN - 1868-8071
VL - 16
SP - 3449
EP - 3460
JO - International Journal of Machine Learning and Cybernetics
JF - International Journal of Machine Learning and Cybernetics
IS - 5
ER -