Abstract
This paper introduces OrthoReg, a simple yet effective Graph-regularized MLP model for semi-supervised node representation learning. We first demonstrate, through empirical observations and theoretical analysis, that node embeddings learned from conventional GR-MLPs suffer from the over-correlation issue. This issue arises when a few dominant singular values overwhelm the embedding space, leading to the limited expressive power of the learned node representations. To mitigate this problem, we propose a novel GR-MLP model called OrthoReg. By incorporating a soft regularization loss on the correlation matrix of node embeddings, OrthoReg explicitly encourages orthogonal node representations, effectively avoiding over-correlated representations. Compared to the currently popular GNN models, our OrthoReg possesses two distinct advantages: 1) Much faster inference speed, particularly for large-scale graphs. 2) Significantly superior performance in inductive cold-start settings. Experiments on semi-supervised node classification tasks, together with the extensive ablation studies, have demonstrated the effectiveness of the proposed designs.
Original language | English (US) |
---|---|
Title of host publication | CIKM 2024 - Proceedings of the 33rd ACM International Conference on Information and Knowledge Management |
Publisher | Association for Computing Machinery |
Pages | 3145-3154 |
Number of pages | 10 |
ISBN (Electronic) | 9798400704369 |
State | Published - Oct 21 2024 |
Externally published | Yes |
Event | 33rd ACM International Conference on Information and Knowledge Management, CIKM 2024 - Boise, United States Duration: Oct 21 2024 → Oct 25 2024 |
Publication series
Name | International Conference on Information and Knowledge Management, Proceedings |
---|---|
ISSN (Print) | 2155-0751 |
Conference
Conference | 33rd ACM International Conference on Information and Knowledge Management, CIKM 2024 |
---|---|
Country/Territory | United States |
City | Boise |
Period | 10/21/24 → 10/25/24 |
Bibliographical note
Publisher Copyright:© 2024 ACM.
Keywords
- graph neural networks
- node representation learning
- semi-supervised learning