Line Graph Vietoris-Rips Persistence Diagram for Topological Graph Representation Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shin, Jaesun, Jeon, Eunjoo, Cho, Taewon, Cho, Namkyeong, Gwon, Youngjune
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916539918188544
author Shin, Jaesun
Jeon, Eunjoo
Cho, Taewon
Cho, Namkyeong
Gwon, Youngjune
author_facet Shin, Jaesun
Jeon, Eunjoo
Cho, Taewon
Cho, Namkyeong
Gwon, Youngjune
contents While message passing graph neural networks result in informative node embeddings, they may suffer from describing the topological properties of graphs. To this end, node filtration has been widely used as an attempt to obtain the topological information of a graph using persistence diagrams. However, these attempts have faced the problem of losing node embedding information, which in turn prevents them from providing a more expressive graph representation. To tackle this issue, we shift our focus to edge filtration and introduce a novel edge filtration-based persistence diagram, named Topological Edge Diagram (TED), which is mathematically proven to preserve node embedding information as well as contain additional topological information. To implement TED, we propose a neural network based algorithm, named Line Graph Vietoris-Rips (LGVR) Persistence Diagram, that extracts edge information by transforming a graph into its line graph. Through LGVR, we propose two model frameworks that can be applied to any message passing GNNs, and prove that they are strictly more powerful than Weisfeiler-Lehman type colorings. Finally we empirically validate superior performance of our models on several graph classification and regression benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17468
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Line Graph Vietoris-Rips Persistence Diagram for Topological Graph Representation Learning
Shin, Jaesun
Jeon, Eunjoo
Cho, Taewon
Cho, Namkyeong
Gwon, Youngjune
Machine Learning
Artificial Intelligence
Algebraic Topology
While message passing graph neural networks result in informative node embeddings, they may suffer from describing the topological properties of graphs. To this end, node filtration has been widely used as an attempt to obtain the topological information of a graph using persistence diagrams. However, these attempts have faced the problem of losing node embedding information, which in turn prevents them from providing a more expressive graph representation. To tackle this issue, we shift our focus to edge filtration and introduce a novel edge filtration-based persistence diagram, named Topological Edge Diagram (TED), which is mathematically proven to preserve node embedding information as well as contain additional topological information. To implement TED, we propose a neural network based algorithm, named Line Graph Vietoris-Rips (LGVR) Persistence Diagram, that extracts edge information by transforming a graph into its line graph. Through LGVR, we propose two model frameworks that can be applied to any message passing GNNs, and prove that they are strictly more powerful than Weisfeiler-Lehman type colorings. Finally we empirically validate superior performance of our models on several graph classification and regression benchmarks.
title Line Graph Vietoris-Rips Persistence Diagram for Topological Graph Representation Learning
topic Machine Learning
Artificial Intelligence
Algebraic Topology
url https://arxiv.org/abs/2412.17468