Revisit Orthogonality in Graph-Regularized MLPs
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.
Authors
- Hengrui Zhang (ORCID: https://orcid.org/0009-0006-1330-0899)
- Soji Adeshina (ORCID: https://orcid.org/0000-0003-3945-3640)
- Vassilis N. Ioannidis (ORCID: https://orcid.org/0000-0002-8367-0733)
- Christos Faloutsos (ORCID: https://orcid.org/0000-0003-2996-9790)
- Philip S. Yu (ORCID: https://orcid.org/0000-0002-3491-5968)
- Jiani Zhang (ORCID: https://orcid.org/0000-0003-0074-6761)
- Da Zheng (ORCID: https://orcid.org/0000-0001-8115-5415)
- George Karypis (ORCID: https://orcid.org/0000-0003-2753-1437)
- Xiao Qin (ORCID: https://orcid.org/0000-0003-3603-3341)
- Shen Wang
Institutions
- Amazon (United States) (US)
- University of Illinois Chicago (US)
Publication Details
- Published
- 2024-10-20
- DOI
- https://doi.org/10.1145/3627673.3679811
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.00