Saved in:
Bibliographic Details
Main Authors: Ma, Jiangyan, Wang, Yifei, Wang, Yisen
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2310.18716
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929206305226752
author Ma, Jiangyan
Wang, Yifei
Wang, Yisen
author_facet Ma, Jiangyan
Wang, Yifei
Wang, Yisen
contents Spectral embedding is a powerful graph embedding technique that has received a lot of attention recently due to its effectiveness on Graph Transformers. However, from a theoretical perspective, the universal expressive power of spectral embedding comes at the price of losing two important invariance properties of graphs, sign and basis invariance, which also limits its effectiveness on graph data. To remedy this issue, many previous methods developed costly approaches to learn new invariants and suffer from high computation complexity. In this work, we explore a minimal approach that resolves the ambiguity issues by directly finding canonical directions for the eigenvectors, named Laplacian Canonization (LC). As a pure pre-processing method, LC is light-weighted and can be applied to any existing GNNs. We provide a thorough investigation, from theory to algorithm, on this approach, and discover an efficient algorithm named Maximal Axis Projection (MAP) that works for both sign and basis invariance and successfully canonizes more than 90% of all eigenvectors. Experiments on real-world benchmark datasets like ZINC, MOLTOX21, and MOLPCBA show that MAP consistently outperforms existing methods while bringing minimal computation overhead. Code is available at https://github.com/PKU-ML/LaplacianCanonization.
format Preprint
id arxiv_https___arxiv_org_abs_2310_18716
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Laplacian Canonization: A Minimalist Approach to Sign and Basis Invariant Spectral Embedding
Ma, Jiangyan
Wang, Yifei
Wang, Yisen
Machine Learning
Spectral embedding is a powerful graph embedding technique that has received a lot of attention recently due to its effectiveness on Graph Transformers. However, from a theoretical perspective, the universal expressive power of spectral embedding comes at the price of losing two important invariance properties of graphs, sign and basis invariance, which also limits its effectiveness on graph data. To remedy this issue, many previous methods developed costly approaches to learn new invariants and suffer from high computation complexity. In this work, we explore a minimal approach that resolves the ambiguity issues by directly finding canonical directions for the eigenvectors, named Laplacian Canonization (LC). As a pure pre-processing method, LC is light-weighted and can be applied to any existing GNNs. We provide a thorough investigation, from theory to algorithm, on this approach, and discover an efficient algorithm named Maximal Axis Projection (MAP) that works for both sign and basis invariance and successfully canonizes more than 90% of all eigenvectors. Experiments on real-world benchmark datasets like ZINC, MOLTOX21, and MOLPCBA show that MAP consistently outperforms existing methods while bringing minimal computation overhead. Code is available at https://github.com/PKU-ML/LaplacianCanonization.
title Laplacian Canonization: A Minimalist Approach to Sign and Basis Invariant Spectral Embedding
topic Machine Learning
url https://arxiv.org/abs/2310.18716